Fix HF Spaces runtime 3

app.py

@@ -12,21 +12,18 @@ BACKEND_URL = "https://adarshds-thesisbackend.hf.space"
 # -----------------------------
 def ask_question(message, history):
     try:
-        response = requests.post(
+        r = requests.post(
             f"{BACKEND_URL}/rag/ask",
             json={"question": message},
             timeout=120,
         )
+        data = r.json()
 
-        data = response.json()
-
-        images_md = ""
+        images = ""
         for img in data.get("images", []):
-            images_md += f"\n\n![img]({BACKEND_URL}/{img})"
-
-        answer = data["answer"] + images_md
+            images += f"\n\n![img]({BACKEND_URL}/{img})"
 
-        history.append((message, answer))
+        history.append((message, data["answer"] + images))
         return "", history
 
     except Exception as e:
@@ -35,19 +32,18 @@ def ask_question(message, history):
 
 
 # -----------------------------
-# 📄 UPLOAD PDF
+# 📄 UPLOAD
 # -----------------------------
 def upload_pdf(file):
     try:
-        response = requests.post(
+        r = requests.post(
             f"{BACKEND_URL}/upload/pdf",
             files={"file": file},
             timeout=300,
         )
-        return response.json()["message"]
-
+        return r.json()["message"]
     except Exception as e:
-        return f"❌ {str(e)}"
+        return str(e)
 
 
 # -----------------------------
@@ -55,19 +51,17 @@ def upload_pdf(file):
 # -----------------------------
 def visualize(image, question):
     try:
-        response = requests.post(
+        r = requests.post(
             f"{BACKEND_URL}/visualize",
             files={"image": open(image, "rb")},
             data={"question": question},
             timeout=120,
         )
-
-        data = response.json()
+        data = r.json()
 
         return data["answer"], f"{BACKEND_URL}/{data['output_image']}"
-
     except Exception as e:
-        return f"❌ {str(e)}", None
+        return str(e), None
 
 
 # -----------------------------
@@ -75,60 +69,48 @@ def visualize(image, question):
 # -----------------------------
 def grade(image):
     try:
-        response = requests.post(
+        r = requests.post(
             f"{BACKEND_URL}/grade-annotation/",
             files={"file": open(image, "rb")},
             timeout=120,
         )
-
-        return response.json()["result"]
-
+        return r.json()["result"]
     except Exception as e:
-        return f"❌ {str(e)}"
+        return str(e)
 
 
 # ==================================================
-# 🎨 UI
+# UI
 # ==================================================
 
-with gr.Blocks(analytics_enabled=False) as demo:
+with gr.Blocks() as demo:
 
     gr.Markdown("# 🧠 Multimodal RAG Anatomy Tutor")
 
-    # 💬 CHAT
     with gr.Tab("Chat"):
         chatbot = gr.Chatbot(height=500)
-        msg = gr.Textbox(placeholder="Ask anatomy question...")
+        msg = gr.Textbox()
         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)
+        btn = gr.Button("Upload & Ingest")
+        out = gr.Textbox()
+        btn.click(upload_pdf, file, out)
 
-    # 🧠 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()
-
-        vis_btn.click(
-            visualize,
-            [vis_image, vis_question],
-            [vis_answer, vis_output],
-        )
+        img = gr.Image(type="filepath")
+        q = gr.Textbox()
+        btn = gr.Button("Visualize")
+        ans = gr.Textbox()
+        out_img = gr.Image()
+        btn.click(visualize, [img, q], [ans, out_img])
 
-    # 📝 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)
+        img = gr.Image(type="filepath")
+        btn = gr.Button("Grade")
+        res = gr.Textbox(lines=15)
+        btn.click(grade, img, res)
 
 
-# ✅ THIS IS THE ONLY SAFE LAUNCH FOR HF SPACES
 demo.queue().launch()