from fastapi import FastAPI, Request
from fastapi.responses import HTMLResponse
from fastapi.templating import Jinja2Templates
from transformers import pipeline

pipe = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1", trust_remote_code=True)

app = FastAPI()
templates = Jinja2Templates(directory="templates")

@app.get('/')
def home():
    return {"hello": "Bitfumes"}

@app.get('/ask', response_class=HTMLResponse)
async def ask_form(request: Request):
    return templates.TemplateResponse("ask.html", {"request": request})

@app.post('/ask')
async def generate(prompt: str):
    result = pipe(prompt, max_length=100, num_return_sequences=1)
    return {"generated_text": result[0]['generated_text']}