Update app.py

app.py

@@ -40,16 +40,20 @@ UPLOAD_DIR = "/tmp/uploads"
 os.makedirs(UPLOAD_DIR, exist_ok=True)
 
 def vision_fn(image, hint_text):
+    import uuid, os
+
+    UPLOAD_DIR = "/tmp/uploads"
+    os.makedirs(UPLOAD_DIR, exist_ok=True)
+
     filename = f"{uuid.uuid4()}.png"
     path = os.path.join(UPLOAD_DIR, filename)
 
     image.save(path)
 
-    # HF exposes files automatically via /file=
     image_url = f"https://datdevsteve-nivra-ai-agent.hf.space/file={path}"
-
     return nivra_vision(image_url, hint_text)
 
+
 # ===============================
 # UI + API
 # ===============================
@@ -82,11 +86,13 @@ with gr.Blocks(title="🩺 Nivra AI Agent") as demo:
 
         gr.Button(visible=False).click(
             vision_fn,
-            inputs=[gr.Image(type="pil"), gr.Textbox()],
-            outputs=[gr.Textbox()],
-            api_name="vision_fn",
-        )
-
+            inputs=[
+                gr.Image(type="pil", label="Medical Image"),
+                gr.Textbox(label="Optional description"),
+        ],
+        outputs=[gr.Textbox()],
+        api_name="vision_fn",
+)
 
 # 🔴 DO NOT enable queue for Flutter SSE
 demo.launch(