Update app.py

app.py

@@ -9,7 +9,6 @@ from concurrent.futures import ThreadPoolExecutor, as_completed
 import hashlib
 import shutil
 import time
-from functools import lru_cache
 from threading import Thread
 import re
 import tempfile
@@ -19,7 +18,6 @@ current_dir = os.path.dirname(os.path.abspath(__file__))
 src_path = os.path.abspath(os.path.join(current_dir, "src"))
 sys.path.insert(0, src_path)
 
-# Cache directories
 base_dir = "/data"
 model_cache_dir = os.path.join(base_dir, "txagent_models")
 tool_cache_dir = os.path.join(base_dir, "tool_cache")
@@ -148,10 +146,6 @@ def create_ui(agent: TxAgent):
 
         def analyze_potential_oversights(message: str, history: list, conversation: list, files: list):
             try:
-                history.append({"role": "user", "content": message})
-                history.append({"role": "assistant", "content": "⏳ Analyzing records for potential oversights..."})
-                yield history, None
-
                 extracted_data = ""
                 file_hash_value = ""
 
@@ -190,18 +184,23 @@ Medical Records:\n{extracted_data[:15000]}
                 if not cleaned:
                     cleaned = "No clear oversights identified. Recommend comprehensive review."
 
+                updated_history = history + [
+                    {"role": "user", "content": message},
+                    {"role": "assistant", "content": cleaned}
+                ]
+
                 report_path = None
                 if file_hash_value:
                     possible_report = os.path.join(report_dir, f"{file_hash_value}_report.txt")
                     if os.path.exists(possible_report):
                         report_path = possible_report
 
-                history[-1] = {"role": "assistant", "content": cleaned}
-                yield history, report_path
+                yield updated_history, report_path
 
             except Exception as e:
-                history.append({"role": "assistant", "content": f"❌ Analysis failed: {str(e)}"})
-                yield history, None
+                updated_history = history + [{"role": "user", "content": message},
+                                             {"role": "assistant", "content": f"❌ Analysis failed: {str(e)}"}]
+                yield updated_history, None
 
         inputs = [msg_input, chatbot, conversation_state, file_upload]
         outputs = [chatbot, download_output]