Spaces:
Running
Running
File size: 2,517 Bytes
4ab997d |
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 69 70 71 72 73 |
from fastapi import FastAPI, UploadFile, Form, Request
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import HTMLResponse, JSONResponse, FileResponse
from fastapi.staticfiles import StaticFiles
from fastapi.templating import Jinja2Templates
import shutil, os
from tempfile import gettempdir
app = FastAPI()
# ✅ CORS to allow frontend access
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
# ✅ Static assets
app.mount("/resources", StaticFiles(directory="resources"), name="resources")
app.mount("/static", StaticFiles(directory="static"), name="static")
# ✅ Jinja2 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"/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
@app.get("/audio/{filename}")
async def get_audio(filename: str):
filepath = os.path.join(gettempdir(), filename)
return FileResponse(filepath, media_type="audio/mpeg")
|