Update app.py

app.py

@@ -8,10 +8,9 @@ import hashlib
 import shutil
 import re
 from datetime import datetime
-import time
 from concurrent.futures import ThreadPoolExecutor, as_completed
 
-# Setup paths
+# Setup directories
 persistent_dir = "/data/hf_cache"
 os.makedirs(persistent_dir, exist_ok=True)
 
@@ -20,16 +19,13 @@ tool_cache_dir = os.path.join(persistent_dir, "tool_cache")
 file_cache_dir = os.path.join(persistent_dir, "cache")
 report_dir = os.path.join(persistent_dir, "reports")
 
-for directory in [model_cache_dir, tool_cache_dir, file_cache_dir, report_dir]:
-    os.makedirs(directory, exist_ok=True)
+for d in [model_cache_dir, tool_cache_dir, file_cache_dir, report_dir]:
+    os.makedirs(d, exist_ok=True)
 
 os.environ["HF_HOME"] = model_cache_dir
 os.environ["TRANSFORMERS_CACHE"] = model_cache_dir
 
-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)
-
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "src")))
 from txagent.txagent import TxAgent
 
 MAX_MODEL_TOKENS = 32768
@@ -162,7 +158,7 @@ def process_final_report(agent, file, chatbot_state: List[Dict[str, str]]) -> Tu
     return messages, report_path
 
 def create_ui(agent):
-    with gr.Blocks(css=\"\"\"
+    with gr.Blocks(css="""
         html, body, .gradio-container {
             height: 100vh;
             background-color: #111827;
@@ -190,24 +186,24 @@ def create_ui(agent):
             border: 1px solid #374151;
             border-radius: 8px;
         }
-    \"\"\") as demo:
-        gr.Markdown(\"\"\"<h2 style='color:#60a5fa'>🩺 Patient History AI Assistant</h2><p>Upload a clinical Excel file and receive a structured diagnostic summary.</p>\"\"\")
+    """) as demo:
+        gr.Markdown("""<h2 style='color:#60a5fa'>🩺 Patient History AI Assistant</h2><p>Upload a clinical Excel file and receive a structured diagnostic summary.</p>""")
         with gr.Row():
             with gr.Column(scale=3):
                 chatbot = gr.Chatbot(
-                    label=\"Clinical Assistant\",
+                    label="Clinical Assistant",
                     height=700,
-                    type=\"messages\",
+                    type="messages",
                     avatar_images=[
-                        \"https://ui-avatars.com/api/?name=AI&background=2563eb&color=fff&size=128\",
-                        \"https://ui-avatars.com/api/?name=You&background=374151&color=fff&size=128\"
+                        "https://ui-avatars.com/api/?name=AI&background=2563eb&color=fff&size=128",
+                        "https://ui-avatars.com/api/?name=You&background=374151&color=fff&size=128"
                     ]
                 )
             with gr.Column(scale=1):
                 with gr.Row():
-                    file_upload = gr.File(label=\"\", file_types=[\".xlsx\"], elem_id=\"upload-btn\")
-                    analyze_btn = gr.Button(\"🧠 Analyze\", variant=\"primary\")
-                report_output = gr.File(label=\"Download Report\", visible=False, interactive=False)
+                    file_upload = gr.File(label="", file_types=[".xlsx"], elem_id="upload-btn")
+                    analyze_btn = gr.Button("🧠 Analyze", variant="primary")
+                report_output = gr.File(label="Download Report", visible=False, interactive=False)
 
         chatbot_state = gr.State(value=[])
 
@@ -219,11 +215,11 @@ def create_ui(agent):
 
     return demo
 
-if __name__ == \"__main__\":
+if __name__ == "__main__":
     try:
         agent = init_agent()
         demo = create_ui(agent)
-        demo.launch(server_name=\"0.0.0.0\", server_port=7860, allowed_paths=[\"/data/hf_cache/reports\"], share=False)
+        demo.launch(server_name="0.0.0.0", server_port=7860, allowed_paths=["/data/hf_cache/reports"], share=False)
     except Exception as e:
-        print(f\"Error: {str(e)}\")
+        print(f"Error: {str(e)}")
         sys.exit(1)