Update

app.py

@@ -34,16 +34,43 @@ def handle_user_turn(user_input, chat_history, user_role, sid):
     if not user_input.strip():
         return chat_history, ""
 
-    chat_history.append((user_input, None))
+    # Gradio's 'messages' type for chatbot expects dictionaries
+    # Ensure chat_history is initialized with dictionaries or converted
+    if not chat_history or not isinstance(chat_history[0], dict):
+        # Convert existing tuples to dictionaries if needed, or start fresh
+        new_chat_history = []
+        for user_msg, bot_msg in chat_history:
+            if user_msg is not None:
+                new_chat_history.append({"role": "user", "content": user_msg})
+            if bot_msg is not None:
+                new_chat_history.append({"role": "assistant", "content": bot_msg})
+        chat_history = new_chat_history
+    
+    chat_history.append({"role": "user", "content": user_input}) # Changed to dictionary format
 
     if not MODAL_BACKEND_URL:
-        chat_history[-1] = (user_input, "❗ Configuration error: `MODAL_BACKEND_URL` not set.")
+        # For an error, append as assistant message
+        chat_history.append({"role": "assistant", "content": "❗ Configuration error: `MODAL_BACKEND_URL` not set."})
         return chat_history, ""
 
     thinking_steps = ["Analyzing...", "Searching sources...", "Synthesizing..."]
+    
+    # Store the last user message to update it later
+    # This assumes the last message is always the user's latest input
+    last_user_message_index = len(chat_history) - 1
+
     for step in thinking_steps:
-        chat_history[-1] = (user_input, f"<div class='chatbot-message bot'><em>{step}</em></div>")
-        yield chat_history, ""
+        # Create a temporary 'thinking' message from the assistant
+        temp_message = {"role": "assistant", "content": f"<div class='chatbot-message bot'><em>{step}</em></div>"}
+        
+        # If there's an existing assistant thinking message, update it in place
+        # Otherwise, append a new one
+        if len(chat_history) > last_user_message_index + 1 and chat_history[last_user_message_index + 1]["role"] == "assistant":
+            chat_history[last_user_message_index + 1] = temp_message
+        else:
+            chat_history.append(temp_message)
+
+        yield chat_history, "" # Yield with the temporary thinking message
 
         time.sleep(0.4)
 
@@ -57,8 +84,14 @@ def handle_user_turn(user_input, chat_history, user_role, sid):
         reply = response.json().get("final_markdown", "No response received.")
     except Exception as e:
         reply = f"❗ Error: {str(e)}"
+    
+    # Update the last assistant message with the final reply
+    # This should replace the thinking message or be appended if no thinking message was added
+    if len(chat_history) > last_user_message_index + 1 and chat_history[last_user_message_index + 1]["role"] == "assistant":
+        chat_history[last_user_message_index + 1] = {"role": "assistant", "content": reply}
+    else:
+        chat_history.append({"role": "assistant", "content": reply}) # Should not typically happen if thinking steps run
 
-    chat_history[-1] = (user_input, reply)
     yield chat_history, ""
 
 with gr.Blocks(css=CUSTOM_CSS, title="MediAgent AI") as demo:
@@ -74,10 +107,12 @@ with gr.Blocks(css=CUSTOM_CSS, title="MediAgent AI") as demo:
       <li>🗣 Ask in English, Spanish, Hindi, or French</li>
       <li>🎯 <b>Patient</b> = simplified advice | <b>Doctor</b> = in-depth, technical insights</li>
       <li>🔗 Answers are backed by PubMed, FDA, etc.</li>
+      <li>💊 Can provide information on **drug side effects, interactions, and official label details.**</li>
+      <li>🔬 Can search for **clinical trials and medical research.**</li>
       <li>🛡️ Detects emergencies, adds safety notices</li>
     </ul>
     </div>
-    """)
+    """) # Changed Markdown content
 
     with gr.Row():
         with gr.Column(scale=1):
@@ -88,7 +123,7 @@ with gr.Blocks(css=CUSTOM_CSS, title="MediAgent AI") as demo:
                 info="Choose your user type for tailored responses"
             )
         with gr.Column(scale=3):
-            chatbot = gr.Chatbot(label="Conversation", height=480)
+            chatbot = gr.Chatbot(label="Conversation", height=480, type='messages') # Added type='messages'
 
     with gr.Row():
         user_textbox = gr.Textbox(
@@ -102,10 +137,13 @@ with gr.Blocks(css=CUSTOM_CSS, title="MediAgent AI") as demo:
         examples=[
             "What are the side effects of metformin?",
             "Explain Crohn’s disease vs Ulcerative Colitis for a doctor",
-            "¿Cuáles son los síntomas de la hipertensión?"
+            "¿Cuáles son los síntomas de la hipertensión?",
+            "Can I take ibuprofen with lisinopril? (for interactions)", # New example
+            "What is the FDA label information for acetaminophen?", # New example
+            "Are there any ongoing clinical trials for breast cancer?", # New example
         ],
         inputs=[user_textbox]
-    )
+    ) # Changed Examples content
 
     gr.ClearButton([chatbot, user_textbox], value="🔄 Clear")
 
@@ -123,4 +161,4 @@ with gr.Blocks(css=CUSTOM_CSS, title="MediAgent AI") as demo:
 
 if __name__ == "__main__":
     demo.queue()
-    demo.launch()
+    demo.launch()

requirements.txt

@@ -1,4 +1,4 @@
 # File: requirements.txt
 
-gradio==4.44.1
-requests
+gradio==5.33.0
+requests