Spaces:
Running
Running
File size: 2,445 Bytes
8b84edc 84ef14b 8b84edc 2918218 d92713e 8b84edc 9f09ae7 8b84edc c649b34 8b84edc 84ef14b 8b84edc 84ef14b 8b84edc c649b34 8b84edc |
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")
|