Update app.py

app.py

@@ -1,81 +1,63 @@
 import gradio as gr
-from nivra_agent import nivra_chat
+from nivra_agent import nivra_chat, nivra_vision
 from indicTrans3Infer import translate_text
 
+# ===============================
+# Chat endpoint
+# ===============================
 
 def chat_fn(message, history):
     history = history or []
-
-    # Add user message
-    history.append({
-        "role": "user",
-        "content": message
-    })
-
-    # Get assistant response
+    history.append({"role": "user", "content": message})
     response = nivra_chat(message, history)
+    history.append({"role": "assistant", "content": response})
+    return history, ""
 
-    # Add assistant message
-    history.append({
-        "role": "assistant",
-        "content": response
-    })
+# ===============================
+# Translation endpoint
+# ===============================
 
-    # Return updated history + clear textbox
-    return history, ""
-    
 def indictrans_translate(text: str, target_language: str):
-    """
-    Gradio-exposed translation endpoint
-    """
-    translated = translate_text(text, target_language)
-    return translated
+    return translate_text(text, target_language)
 
+# ===============================
+# 🆕 Vision endpoint
+# ===============================
 
-with gr.Blocks(title="🩺 Nivra AI Agent") as demo:
-    gr.Markdown(
-        """
-        # 🩺 Nivra AI Agent  
-        _AI-powered healthcare assistant for preliminary medical guidance._
-        """
-    )
-
-    chatbot = gr.Chatbot(
-        elem_id="chatbot",
-        show_label=False,
-    )
+def vision_fn(image_base64: str, hint_text: str):
+    return nivra_vision(image_base64, hint_text)
 
-    # --- NEW: IndicTrans endpoint ---
-    translate_input = gr.Textbox(visible=False)
-    translate_lang = gr.Textbox(visible=False)
-    translate_output = gr.Textbox(visible=False)
+# ===============================
+# UI
+# ===============================
 
-    translate_btn = gr.Button(visible=False)
-
-    translate_btn.click(
-        indictrans_translate,
-        inputs=[translate_input, translate_lang],
-        outputs=[translate_output],
-        api_name="indictrans_translate"
-    )
+with gr.Blocks(title="🩺 Nivra AI Agent") as demo:
+    gr.Markdown("# 🩺 Nivra AI Agent")
 
-    with gr.Row():
-        txt = gr.Textbox(
-            placeholder="Describe your symptoms (e.g. fever, headache, chills)...",
-            show_label=False,
-            scale=8,
-        )
-        send = gr.Button("Send", scale=1)
+    chatbot = gr.Chatbot(show_label=False)
+    txt = gr.Textbox(placeholder="Describe your symptoms...")
+    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.Markdown(
-        "⚠️ *Not a doctor. For emergencies, consult a medical professional immediately.*"
+    # Hidden API endpoints
+    gr.Button(visible=False).click(
+        indictrans_translate,
+        inputs=[gr.Textbox(), gr.Textbox()],
+        outputs=[gr.Textbox()],
+        api_name="indictrans_translate",
+    )
+
+    gr.Button(visible=False).click(
+        vision_fn,
+        inputs=[gr.Textbox(), gr.Textbox()],
+        outputs=[gr.Textbox()],
+        api_name="vision_fn",
     )
 
 demo.launch(
     server_name="0.0.0.0",
     server_port=7860,
-    ssr_mode=False,   # ✅ fixes HF layout + heartbeat issues
+    ssr_mode=False,
 )