from fastapi import FastAPI
import uvicorn
from models.model_loader import generate_response

app = FastAPI()

@app.get("/")
def home():
    return {"message": "DeepSeek AI is running on Hugging Face!"}

@app.post("/generate/")
def generate_task(task: str):
    response = generate_response(task)
    return {"response": response}

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