Fix HF Spaces runtime

app.py

@@ -1,109 +1,132 @@
 import os
-
-import os
-
-os.environ["GRADIO_SERVER_NAME"] = "0.0.0.0"
-os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"
-os.environ["GRADIO_ALLOW_FLAGGING"] = "never"
-
-
-
 import gradio as gr
 import requests
 
+# ✅ HF Spaces safe settings
+os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"
+
 BACKEND_URL = "https://adarshds-thesisbackend.hf.space"
 
 
-# -----------------------------
+# =====================================================
 # 💬 RAG CHAT
-# -----------------------------
+# =====================================================
 def ask_question(message, history):
-    response = requests.post(
-        f"{BACKEND_URL}/rag/ask",
-        json={"question": message},
-        timeout=120,
-    )
+    try:
+        response = requests.post(
+            f"{BACKEND_URL}/rag/ask",
+            json={"question": message},
+            timeout=120,
+        )
+
+        data = response.json()
 
-    data = response.json()
+        images_md = ""
+        for img in data.get("images", []):
+            images_md += f"\n\n![img]({BACKEND_URL}/{img})"
 
-    images = ""
-    for img in data.get("images", []):
-        images += f"\n\n![img]({BACKEND_URL}/{img})"
+        answer = data["answer"] + images_md
 
-    answer = data["answer"] + images
+        history.append((message, answer))
+        return "", history
 
-    history.append((message, answer))
-    return "", history
+    except Exception as e:
+        history.append((message, f"❌ Error: {str(e)}"))
+        return "", history
 
 
-# -----------------------------
+# =====================================================
 # 📄 UPLOAD PDF
-# -----------------------------
+# =====================================================
 def upload_pdf(file):
-    response = requests.post(
-        f"{BACKEND_URL}/upload/pdf",
-        files={"file": file},
-        timeout=300,
-    )
+    try:
+        response = requests.post(
+            f"{BACKEND_URL}/upload/pdf",
+            files={"file": file},
+            timeout=300,
+        )
+        return response.json()["message"]
 
-    return response.json()["message"]
+    except Exception as e:
+        return f"❌ Upload failed: {str(e)}"
 
 
-# -----------------------------
+# =====================================================
 # 🧠 VISUALIZE
-# -----------------------------
+# =====================================================
 def visualize(image, question):
-    response = requests.post(
-        f"{BACKEND_URL}/visualize",
-        files={"image": image},
-        data={"question": question},
-        timeout=120,
-    )
+    try:
+        response = requests.post(
+            f"{BACKEND_URL}/visualize",
+            files={"image": open(image, "rb")},
+            data={"question": question},
+            timeout=120,
+        )
+
+        data = response.json()
 
-    data = response.json()
+        return (
+            data["answer"],
+            f"{BACKEND_URL}/{data['output_image']}",
+        )
 
-    return data["answer"], f"{BACKEND_URL}/{data['output_image']}"
+    except Exception as e:
+        return f"❌ Error: {str(e)}", None
 
 
-# -----------------------------
-# 📝 GRADE
-# -----------------------------
+# =====================================================
+# 📝 GRADE ANNOTATION
+# =====================================================
 def grade(image):
-    response = requests.post(
-        f"{BACKEND_URL}/grade-annotation/",
-        files={"file": image},
-        timeout=120,
-    )
+    try:
+        response = requests.post(
+            f"{BACKEND_URL}/grade-annotation/",
+            files={"file": open(image, "rb")},
+            timeout=120,
+        )
 
-    return response.json()["result"]
+        return response.json()["result"]
 
+    except Exception as e:
+        return f"❌ Grading failed: {str(e)}"
 
-# ==================================================
-# 🎨 UI
-# ==================================================
 
-with gr.Blocks(analytics_enabled=False) as demo:
+# =====================================================
+# 🎨 UI
+# =====================================================
+with gr.Blocks(
+    title="Multimodal RAG Anatomy Tutor",
+    analytics_enabled=False,
+    show_api=False,   # 💥 CRITICAL FIX
+) as demo:
 
     gr.Markdown("# 🧠 Multimodal RAG Anatomy Tutor")
 
-    # 💬 CHAT TAB
-    with gr.Tab("Chat"):
+    # ================= CHAT =================
+    with gr.Tab("💬 Chat"):
         chatbot = gr.Chatbot(height=500)
         msg = gr.Textbox(placeholder="Ask anatomy question...")
-        msg.submit(ask_question, [msg, chatbot], [msg, chatbot])
 
-    # 📄 UPLOAD TAB
-    with gr.Tab("Upload PDF"):
+        msg.submit(
+            ask_question,
+            [msg, chatbot],
+            [msg, chatbot],
+        )
+
+    # ================= UPLOAD =================
+    with gr.Tab("📄 Upload PDF"):
         file = gr.File(file_types=[".pdf"])
         upload_btn = gr.Button("Upload & Ingest")
         upload_output = gr.Textbox()
+
         upload_btn.click(upload_pdf, file, upload_output)
 
-    # 🧠 VISUAL TAB
-    with gr.Tab("Visualize Anatomy"):
+    # ================= VISUALIZE =================
+    with gr.Tab("🧠 Visualize Anatomy"):
         vis_image = gr.Image(type="filepath")
         vis_question = gr.Textbox(label="Ask about the structure")
         vis_btn = gr.Button("Visualize")
+
         vis_answer = gr.Textbox()
         vis_output = gr.Image()
 
@@ -113,12 +136,15 @@ with gr.Blocks(analytics_enabled=False) as demo:
             [vis_answer, vis_output],
         )
 
-    # 📝 GRADING TAB
-    with gr.Tab("Annotation Grading"):
+    # ================= GRADING =================
+    with gr.Tab("📝 Annotation Grading"):
         grade_image = gr.Image(type="filepath")
         grade_btn = gr.Button("Grade")
+
         grade_result = gr.Textbox(lines=15)
+
         grade_btn.click(grade, grade_image, grade_result)
 
 
-demo.queue().launch(server_name="0.0.0.0", server_port=7860)
+# ✅ HF SPACES LAUNCH MODE
+demo.queue(api_open=False).launch()