Update app.py

app.py

@@ -1,8 +1,10 @@
-
+import sys
+import os
+import pandas as pd
 import pdfplumber
 import json
 import gradio as gr
-from typing import List
+from typing import List, Tuple
 from concurrent.futures import ThreadPoolExecutor, as_completed
 import hashlib
 import shutil
@@ -168,14 +170,18 @@ def format_response_for_ui(response: str) -> str:
         return "\n".join(formatted)
     return cleaned
 
-def analyze(message: str, history: list, files: list):
+def analyze(message: str, history: List[Tuple[str, str]], files: list):
     start_time = datetime.now()
     logger.info(f"Starting analysis for message: {message[:100]}...")
     if files:
         logger.info(f"Processing {len(files)} uploaded files")
 
-    history = history + [{"role": "user", "content": message},
-                         {"role": "assistant", "content": "⏳ Analyzing records for potential oversights..."}]
+    # Initialize chat history in the correct format if empty
+    if history is None:
+        history = []
+    
+    # Add user message to history
+    history.append([message, None])
     yield history, None
 
     extracted = ""
@@ -190,7 +196,7 @@ def analyze(message: str, history: list, files: list):
                 logger.info(f"Processed {len(files)} files, extracted {len(extracted)} characters")
         except Exception as e:
             logger.error(f"Error processing files: {str(e)}")
-            history[-1] = {"role": "assistant", "content": f"❌ Error processing files: {str(e)}"}
+            history[-1][1] = f"❌ Error processing files: {str(e)}"
             yield history, None
             return
 
@@ -230,7 +236,7 @@ Medical Records:
             formatted_partial = format_response_for_ui(partial_response)
             
             if formatted_partial:
-                history[-1] = {"role": "assistant", "content": formatted_partial}
+                history[-1][1] = formatted_partial
                 yield history, None
 
         full_response = "".join(response_chunks)
@@ -241,7 +247,7 @@ Medical Records:
             final_output = "No clear oversights identified. Recommend comprehensive review."
             logger.info("No significant findings detected in analysis")
         
-        history[-1] = {"role": "assistant", "content": final_output}
+        history[-1][1] = final_output
 
         # Save report
         report_path = None
@@ -260,7 +266,7 @@ Medical Records:
 
     except Exception as e:
         logger.error(f"Error during analysis: {str(e)}", exc_info=True)
-        history[-1] = {"role": "assistant", "content": f"❌ Error during analysis: {str(e)}"}
+        history[-1][1] = f"❌ Error during analysis: {str(e)}"
         yield history, None
 
 def create_ui(agent):