from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel
from fastapi.staticfiles import StaticFiles
from services.generate_questions_service import GenerateQuestionsService

generate_questions_service = GenerateQuestionsService()

class Body(BaseModel):
    matter: str
    subject: str
    difficultie: str

app = FastAPI()

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

@app.post("/generate_questions")
async def generate_questions(body: Body):
    print(body)
    matter = body.matter
    subject = body.subject
    difficultie = body.difficultie
    query = f"Quero que você gere questões de {matter}, sendo do assunto: {subject} e sendo da dificuldade: {difficultie}."
    res = generate_questions_service.handle(f"""{query}""")
    return res

app.mount("/", StaticFiles(directory="static", html=True), name="static")

if __name__ == "__main__":
    import uvicorn
    uvicorn.run("app:app", host="0.0.0.0", port=8000)