Update server.py

server.py

@@ -1,13 +1,28 @@
 from fastapi import FastAPI
+import gradio as gr
 
 app = FastAPI()
 
-
+# --- API endpoints (required) ---
 @app.post("/reset")
 async def reset():
     return {"status": "ok"}
 
+@app.get("/status")
+async def status():
+    return {"status": "online"}
+
+
+# --- Gradio UI ---
+def demo_fn(task):
+    return f"Running task: {task} (UI restored 🚀)"
+
+demo = gr.Interface(
+    fn=demo_fn,
+    inputs=gr.Dropdown(["easy", "medium", "hard"]),
+    outputs="text",
+    title="Email Gatekeeper"
+)
 
-@app.get("/")
-async def root():
-    return {"message": "email triage Rl environment is running"}
+# ✅ Mount UI at root
+app = gr.mount_gradio_app(app, demo, path="/")