Update app.py

app.py

@@ -154,8 +154,7 @@ def create_ui(agent: TxAgent):
         def analyze_potential_oversights(message: str, history: list, conversation: list, files: list):
             start_time = time.time()
             try:
-                history.append({"role": "user", "content": message})
-                history.append({"role": "assistant", "content": "⏳ Analyzing records for potential oversights..."})
+                history = history + [{"role": "user", "content": message}, {"role": "assistant", "content": "⏳ Analyzing records for potential oversights..."}]
                 yield history, None
 
                 extracted_data = ""
@@ -191,8 +190,8 @@ Medical Records:\n{extracted_data[:15000]}
                     elif isinstance(chunk, list):
                         response += "".join([c.content for c in chunk if hasattr(c, 'content')])
 
-                    history[-1]["content"] = response.replace("[TOOL_CALLS]", "").strip()
-                    yield history, None
+                    cleaned = response.replace("[TOOL_CALLS]", "").strip()
+                    yield history[:-1] + [{"role": "assistant", "content": cleaned}], None
 
                 final_output = response.replace("[TOOL_CALLS]", "").strip()
                 if not final_output:
@@ -204,7 +203,7 @@ Medical Records:\n{extracted_data[:15000]}
                     if os.path.exists(possible_report):
                         report_path = possible_report
 
-                history[-1] = {"role": "assistant", "content": final_output}
+                history = history[:-1] + [{"role": "assistant", "content": final_output}]
                 yield history, report_path
 
             except Exception as e: