Update app.py

app.py

@@ -1,100 +1,105 @@
-import gradio as gr
-import tempfile
-import uuid
 import os
-import base64
-from nivra_agent import nivra_chat, nivra_vision
+import uuid
+import gradio as gr
+from PIL import Image
 
-# ===============================
-# Chat endpoint
-# ===============================
-def chat_fn(message, history):
-    history = history or []
-    history.append({"role": "user", "content": message})
-    response = nivra_chat(message, history)
-    history.append({"role": "assistant", "content": response})
-    return history, ""
+from nivra_agent import nivra_chat, nivra_vision
 
+# ==================================================
+# Constants
+# ==================================================
 
 UPLOAD_DIR = "/tmp/uploads"
 os.makedirs(UPLOAD_DIR, exist_ok=True)
 
-def upload_image(image_base64: str):
-    img_bytes = base64.b64decode(image_base64)
-    filename = f"{uuid.uuid4()}.png"
-    path = os.path.join(UPLOAD_DIR, filename)
-
-    with open(path, "wb") as f:
-        f.write(img_bytes)
+SPACE_HOST = os.environ.get(
+    "SPACE_HOST",
+    "https://datdevsteve-nivra-ai-agent.hf.space"
+)
 
-    # Expose via HF Space static file serving
-    return f"{os.environ['SPACE_HOST']}/file={path}"
+# ==================================================
+# CHAT ENDPOINT
+# ==================================================
 
-# ===============================
-# 🆕 Vision endpoint (CRITICAL FIX)
-# ===============================
-import uuid
-import os
-
-UPLOAD_DIR = "/tmp/uploads"
-os.makedirs(UPLOAD_DIR, exist_ok=True)
+def chat_fn(message, history):
+    history = history or []
+    history.append({"role": "user", "content": message})
 
-def vision_fn(image, hint_text):
-    import uuid, os
+    response = nivra_chat(message, history)
 
-    UPLOAD_DIR = "/tmp/uploads"
-    os.makedirs(UPLOAD_DIR, exist_ok=True)
+    history.append({"role": "assistant", "content": response})
+    return history, ""
 
+# ==================================================
+# 🆕 VISION ENDPOINT (FILE-BASED, NO BASE64)
+# ==================================================
+
+def vision_fn(image: Image.Image, hint_text: str):
+    """
+    image: PIL Image (uploaded by Gradio)
+    hint_text: optional user context
+    """
+
+    if image is None:
+        return (
+            "[PRIMARY DIAGNOSIS] Inconclusive [/PRIMARY DIAGNOSIS]\n"
+            "[DIAGNOSIS DESCRIPTION] No image provided. [/DIAGNOSIS DESCRIPTION]\n"
+            "[FIRST AID] Please upload a valid image. [/FIRST AID]\n"
+            "[EMERGENCY CONSULTATION REQUIRED] No [/EMERGENCY CONSULTATION REQUIRED]"
+        )
+
+    # Save image to HF-served temp directory
     filename = f"{uuid.uuid4()}.png"
     path = os.path.join(UPLOAD_DIR, filename)
-
     image.save(path)
 
-    image_url = f"https://datdevsteve-nivra-ai-agent.hf.space/file={path}"
-    return nivra_vision(image_url, hint_text)
+    # HF Spaces automatically serve /file= paths
+    image_url = f"{SPACE_HOST}/file={path}"
+
+    print(f"📥 Vision image saved: {image_url}")
 
+    # Pass URL to agent (CRITICAL FIX)
+    return nivra_vision(image_url, hint_text)
 
-# ===============================
+# ==================================================
 # UI + API
-# ===============================
+# ==================================================
+
 with gr.Blocks(title="🩺 Nivra AI Agent") as demo:
-    gr.Markdown("# 🩺 Nivra AI Agent")
+    gr.Markdown(
+        "# 🩺 Nivra AI Agent\n"
+        "_AI-powered healthcare assistant for preliminary guidance._"
+    )
 
     # -------------------------------
     # Chat UI
     # -------------------------------
     chatbot = gr.Chatbot(show_label=False)
-    txt = gr.Textbox(placeholder="Describe your symptoms...")
+    txt = gr.Textbox(
+        placeholder="Describe your symptoms (e.g. fever, headache, rash)..."
+    )
     send = gr.Button("Send")
 
     send.click(chat_fn, inputs=[txt, chatbot], outputs=[chatbot, txt])
     txt.submit(chat_fn, inputs=[txt, chatbot], outputs=[chatbot, txt])
-    gr.Button(visible=False).click(
-        upload_image,
-        inputs=[gr.Textbox()],
-        outputs=[gr.Textbox()],
-        api_name="upload_image",
-    )
+
     # -------------------------------
-    # Hidden API endpoints (ISOLATED)
+    # 🔒 Hidden Vision API (Flutter)
     # -------------------------------
-
-    with gr.Row(visible=False):
-        vision_image = gr.Textbox()
-        vision_hint = gr.Textbox()
-        vision_out = gr.Textbox()
-
-        gr.Button(visible=False).click(
-            vision_fn,
-            inputs=[
-                gr.Image(type="pil", label="Medical Image"),
-                gr.Textbox(label="Optional description"),
+    gr.Button(visible=False).click(
+        vision_fn,
+        inputs=[
+            gr.Image(type="pil"),
+            gr.Textbox()
         ],
         outputs=[gr.Textbox()],
         api_name="vision_fn",
-)
+    )
+
+# ==================================================
+# Launch (HF-safe)
+# ==================================================
 
-# 🔴 DO NOT enable queue for Flutter SSE
 demo.launch(
     server_name="0.0.0.0",
     server_port=7860,