from fastapi import FastAPI, Request
from fastapi.middleware.cors import CORSMiddleware
import google.generativeai as genai
import os
from dotenv import load_dotenv

load_dotenv()
app = FastAPI()

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_methods=["*"],
    allow_headers=["*"]
)

genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
model = genai.GenerativeModel("models/gemini-1.5-flash")

@app.get("/")
def home():
    return {"message": "MBZ-Voice-SDK backend is running ✅"}

@app.post("/ask")
async def ask(request: Request):
    data = await request.json()
    query = data.get("query")
    if not query:
        return {"answer": "Please say something."}
    try:
        response = model.generate_content(query)
        return {"answer": response.text}
    except Exception as e:
        return {"answer": f"Error: {str(e)}"}