# ai.py

import requests
import os
from dotenv import load_dotenv

load_dotenv()
AI_URL = os.getenv("OLLAMA_API_URL")

def get_ai_response(prompt):
    payload = {
        "model": "mistral:7b",  # Adjust to match your model
        "prompt": prompt,
        "stream": False
    }

    try:
        response = requests.post(AI_URL, json=payload)
        if response.status_code == 200:
            result = response.json()
            return result.get("response", "[No response]")
        else:
            return f"[Error {response.status_code}] {response.text}"
    except Exception as e:
        return f"[Request failed] {str(e)}"