Update app.py

app.py

@@ -4,6 +4,8 @@ import torch
 import os
 from pydantic import BaseModel
 from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import HTMLResponse
+from fastapi.staticfiles import StaticFiles
 
 
 # ✅ Force Hugging Face cache to /tmp (writable in Spaces)
@@ -46,6 +48,15 @@ def ask(question: str, max_new_tokens: int = 50):
     return {"question": question, "answer": answer}
 
 
+# Mount static folder
+app.mount("/static", StaticFiles(directory="static"), name="static")
+
+@app.get("/ui", response_class=HTMLResponse)
+def serve_ui():
+    html_path = os.path.join("static", "index.html")
+    with open(html_path, "r", encoding="utf-8") as f:
+        return HTMLResponse(f.read())
+
 
 # Health check endpoint
 @app.get("/health")