Spaces:
Running
Running
File size: 2,248 Bytes
2d75ddd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
from fastapi import FastAPI, UploadFile, Form, Request
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import HTMLResponse, JSONResponse, FileResponse
from fastapi.templating import Jinja2Templates
import shutil, os
from tempfile import gettempdir
# ✅ Create app
app = FastAPI()
# ✅ CORS middleware
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
# ✅ Templates
templates = Jinja2Templates(directory="templates")
# ✅ Serve Homepage
@app.get("/", response_class=HTMLResponse)
async def serve_home(request: Request):
return templates.TemplateResponse("home.html", {"request": request})
# ✅ Predict endpoint (handles image + document)
@app.post("/predict")
async def predict(question: str = Form(...), file: UploadFile = Form(...)):
try:
temp_path = f"temp_{file.filename}"
with open(temp_path, "wb") as f:
shutil.copyfileobj(file.file, f)
is_image = file.content_type.startswith("image/")
if is_image:
from .appImage import answer_question_from_image
from PIL import Image
image = Image.open(temp_path).convert("RGB")
answer, audio_path = answer_question_from_image(image, question)
else:
from .app import answer_question_from_doc
class NamedFile:
def __init__(self, name): self.filename = name
def read(self): return open(self.filename, "rb").read()
answer, audio_path = answer_question_from_doc(NamedFile(temp_path), question)
os.remove(temp_path)
if audio_path and os.path.exists(audio_path):
return JSONResponse({
"answer": answer,
"audio": f"/qtAnswering/audio/{os.path.basename(audio_path)}"
})
else:
return JSONResponse({"answer": answer})
except Exception as e:
return JSONResponse({"error": str(e)}, status_code=500)
# ✅ Serve audio files
@app.get("/audio/{filename}")
async def get_audio(filename: str):
filepath = os.path.join(gettempdir(), filename)
return FileResponse(filepath, media_type="audio/mpeg")
|