Update app.py

app.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import os
 import traceback
 import regex as re2
-from typing import List, Tuple, Dict, Any
+from typing import List, Dict, Any
 
 import gradio as gr
 import pandas as pd
@@ -16,7 +16,7 @@ from langchain_experimental.agents.agent_toolkits import create_pandas_dataframe
 
 # --- LOCAL MODULE IMPORTS ---
 from settings import (
-    HEALTHCARE_SETTINGS, GENERAL_CONVERSATION_PROMPT, USE_SCENARIO_ENGINE, DEBUG_PLAN,
+    HEALTHCARE_SETTINGS, GENERAL_CONVERSATION_PROMPT,
     COHERE_MODEL_PRIMARY, COHERE_TIMEOUT_S, USE_OPEN_FALLBACKS
 )
 from audit_log import log_event
@@ -31,7 +31,7 @@ def load_markdown_text(filepath: str) -> str:
         with open(filepath, 'r', encoding='utf-8') as f:
             return f.read()
     except FileNotFoundError:
-        return f"**Error:** The document `{os.path.basename(filepath)}` was not found. Please ensure it is in the same directory as the application."
+        return f"**Error:** The document `{os.path.basename(filepath)}` was not found."
 
 def _sanitize_text(s: str) -> str:
     if not isinstance(s, str): return s
@@ -83,7 +83,21 @@ def handle(user_msg: str, files: list) -> str:
             llm = ChatCohere(model=COHERE_MODEL_PRIMARY, temperature=0)
             enhanced_prompt = _create_enhanced_prompt(safe_in)
             
-            AGENT_PREFIX = """...""" # Your perfected agent prefix remains here
+            AGENT_PREFIX = """
+You are a data analysis agent. You have access to one or more pandas dataframes.
+You MUST respond in one of two formats.
+
+FORMAT 1: To perform a task. Your response must be a single block of text with ONLY these three sections:
+Thought: Your step-by-step reasoning.
+Action: python_repl_ast
+Action Input: The Python code to run.
+
+FORMAT 2: To give the final answer. Your response must be a single block of text with ONLY these two sections:
+Thought: I have now answered all the user's questions and can provide the final report.
+Final Answer: The complete answer, structured as the user requested.
+
+CRITICAL RULE: NEVER combine `Action` and `Final Answer` in the same response. Choose one format.
+"""
 
             agent = create_pandas_dataframe_agent(
                 llm, dataframes, agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
@@ -138,15 +152,18 @@ with gr.Blocks(theme="soft", css="style.css") as demo:
         with gr.Column(scale=2):
             with gr.Tabs():
                 with gr.TabItem("Current Assessment", id=0):
-                    chat_history_output = gr.Chatbot(label="Analysis Output", bubble_full_width=True, height=600)
+                    chat_history_output = gr.Chatbot(
+                        label="Analysis Output",
+                        type="messages",  # <-- THE FINAL FIX IS HERE
+                        height=600
+                    )
                 with gr.TabItem("Assessment History", id=1):
                     gr.Markdown("## Review Past Assessments")
                     history_dropdown = gr.Dropdown(label="Select an assessment to review", choices=[])
                     history_display = gr.Markdown(label="Selected Assessment Details")
 
     # --- FOOTER FOR LEGAL LINKS ---
-    with gr.Row():
-        gr.Markdown("---")
+    with gr.Row(): gr.Markdown("---")
     with gr.Row():
         privacy_link = gr.Button("Privacy Policy", variant="link")
         terms_link = gr.Button("Terms of Service", variant="link")