Update app.py

app.py

@@ -1,22 +1,55 @@
 import gradio as gr
 from nivra_agent import nivra_chat
 
-
 def chat_fn(message, history):
-    # DO NOT touch history
-    # ChatInterface manages it internally
-    return nivra_chat(message, history)
-
-
-demo = gr.ChatInterface(
-    fn=chat_fn,
-    title="🩺 Nivra AI Agent",
-    description="Space to access Nivra's Agentic Interface",
-    examples=[
-        "I have fever and chills",
-        "Patient presents skin rash and itching",
-        "Patient presents stomach pain and vomiting",
-    ],
-)
+    history = history or []
+    response = nivra_chat(message, history)
+    history.append((message, response))
+    return history, ""
+
+CSS = """
+#chatbot {
+    height: 420px;
+    overflow-y: auto;
+    border-radius: 12px;
+}
+.gradio-container {
+    max-width: 900px !important;
+    margin: auto;
+}
+"""
+
+with gr.Blocks(css=CSS, 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",
+        bubble_full_width=False,
+        show_label=False,
+    )
 
-demo.launch()
+    with gr.Row():
+        txt = gr.Textbox(
+            placeholder="Describe your symptoms (e.g. fever, headache, chills)...",
+            scale=8,
+            show_label=False,
+        )
+        send = gr.Button("Send", scale=1)
+
+    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.*"
+    )
+
+demo.launch(
+    server_name="0.0.0.0",
+    server_port=7860,
+    ssr_mode=False,   # 🔥 THIS fixes layout weirdness on HF
+)