Update app.py

app.py

@@ -105,12 +105,12 @@ def init_agent():
     agent.init_model()
     return agent
 
-def process_final_report(agent, file, chatbot_state: List[Dict[str, str]]) -> Tuple[List[Dict[str, str]], Union[str, None]]:
+def process_final_report(agent, file, chatbot_state: List[Tuple[str, str]]) -> Tuple[List[Tuple[str, str]], Union[str, None]]:
     messages = chatbot_state if chatbot_state else []
     if file is None or not hasattr(file, "name"):
-        return messages + [{"role": "assistant", "content": "❌ Please upload a valid Excel file."}], None
+        return messages + [("assistant", "❌ Please upload a valid Excel file.")], None
 
-    messages.append({"role": "user", "content": f"Processing Excel file: {os.path.basename(file.name)}"})
+    messages.append(("user", f"Processing Excel file: {os.path.basename(file.name)}"))
     text = extract_text_from_excel(file.name)
     chunks = split_text_into_chunks(text)
     chunk_responses = [None] * len(chunks)
@@ -141,10 +141,10 @@ def process_final_report(agent, file, chatbot_state: List[Dict[str, str]]) -> Tu
 
     valid = [r for r in chunk_responses if r and not r.startswith("❌")]
     if not valid:
-        return messages + [{"role": "assistant", "content": "❌ No valid chunk results."}], None
+        return messages + [("assistant", "❌ No valid chunk results.")], None
 
     summary_prompt = f"Summarize this analysis in a final structured report:\n\n" + "\n\n".join(valid)
-    messages.append({"role": "assistant", "content": "📊 Generating final report..."})
+    messages.append(("assistant", "📊 Generating final report..."))
 
     final_report = ""
     for res in agent.run_gradio_chat(
@@ -162,8 +162,8 @@ def process_final_report(agent, file, chatbot_state: List[Dict[str, str]]) -> Tu
     with open(report_path, 'w') as f:
         f.write(f"# 🧠 Final Patient Report\n\n{cleaned}")
 
-    messages.append({"role": "assistant", "content": f"📊 Final Report:\n\n{cleaned}"})
-    messages.append({"role": "assistant", "content": f"✅ Report generated and saved: {os.path.basename(report_path)}"})
+    messages.append(("assistant", f"📊 Final Report:\n\n{cleaned}"))
+    messages.append(("assistant", f"✅ Report generated and saved: {os.path.basename(report_path)}"))
     return messages, report_path
 
 def create_ui(agent):
@@ -206,7 +206,7 @@ def create_ui(agent):
         gr.Markdown("""# 🧠 Clinical Reasoning Assistant
 Upload clinical Excel records below and click **Analyze** to generate a medical summary.
 """)
-        chatbot = gr.Chatbot(elem_classes="chatbot")
+        chatbot = gr.Chatbot(label="Chatbot", elem_classes="chatbot", type="bot")
         file_upload = gr.File(label="Upload Excel File", file_types=[".xlsx"])
         analyze_btn = gr.Button("Analyze")
         report_output = gr.File(label="Download Report", visible=False)
@@ -214,8 +214,7 @@ Upload clinical Excel records below and click **Analyze** to generate a medical
 
         def update_ui(file, current_state):
             messages, report_path = process_final_report(agent, file, current_state)
-            chat_format = [(msg["role"], msg["content"]) for msg in messages if isinstance(msg, dict)]
-            return chat_format, gr.update(visible=report_path is not None, value=report_path), messages
+            return messages, gr.update(visible=report_path is not None, value=report_path), messages
 
         analyze_btn.click(
             fn=update_ui,
@@ -232,4 +231,4 @@ if __name__ == "__main__":
         demo.launch(server_name="0.0.0.0", server_port=7860, allowed_paths=["/data/hf_cache/reports"], share=False)
     except Exception as e:
         print(f"Error: {str(e)}")
-        sys.exit(1)
+        sys.exit(1)