Update app.py

app.py

@@ -1,6 +1,6 @@
 # ✅ Fully optimized app.py for Hugging Face Space with persistent 150GB storage
 
-#import sys, os, json, gradio as gr, pandas as pd, pdfplumber, hashlib, shutil, re, time
+import sys, os, json, gradio as gr, pandas as pd, pdfplumber, hashlib, shutil, re, time
 from concurrent.futures import ThreadPoolExecutor, as_completed
 from threading import Thread
 
@@ -28,11 +28,6 @@ os.environ.update({
 LOCAL_TXAGENT_PATH = os.path.join(model_cache_dir, "mims-harvard", "TxAgent-T1-Llama-3.1-8B")
 LOCAL_RAG_PATH = os.path.join(model_cache_dir, "mims-harvard", "ToolRAG-T1-GTE-Qwen2-1.5B")
 
-# Manual download using snapshot_download (only if needed)
-# from huggingface_hub import snapshot_download
-# snapshot_download("mims-harvard/TxAgent-T1-Llama-3.1-8B", local_dir=LOCAL_TXAGENT_PATH, local_dir_use_symlinks=False)
-# snapshot_download("mims-harvard/ToolRAG-T1-GTE-Qwen2-1.5B", local_dir=LOCAL_RAG_PATH, local_dir_use_symlinks=False)
-
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "src")))
 from txagent.txagent import TxAgent
 
@@ -104,7 +99,6 @@ def init_agent():
     agent.init_model()
     return agent
 
-# Lazy load
 agent_container = {"agent": None}
 def get_agent():
     if agent_container["agent"] is None:
@@ -142,10 +136,10 @@ Medical Records:\n{extracted[:15000]}
                     if isinstance(chunk, str): final_response += chunk
                     elif isinstance(chunk, list): final_response += "".join([c.content for c in chunk if hasattr(c, 'content')])
                 cleaned = final_response.replace("[TOOL_CALLS]", "").strip()
-                updated_history = history + [{"role": "user", "content": message}, {"role": "assistant", "content": cleaned}]
+                updated_history = history + [[message, cleaned]]
                 return updated_history, None
             except Exception as e:
-                return history + [{"role": "user", "content": message}, {"role": "assistant", "content": f"❌ Error: {str(e)}"}], None
+                return history + [[message, f"❌ Error: {str(e)}"]], None
 
         send_btn.click(analyze, inputs=[msg_input, chatbot, state, file_upload], outputs=[chatbot, gr.File()])
         msg_input.submit(analyze, inputs=[msg_input, chatbot, state, file_upload], outputs=[chatbot, gr.File()])
@@ -159,4 +153,4 @@ if __name__ == "__main__":
         show_error=True,
         allowed_paths=["/data/reports"],
         share=False
-    )
+    )