Update app.py

app.py

@@ -9,6 +9,7 @@ import shutil
 import re
 from datetime import datetime
 import time
+from concurrent.futures import ThreadPoolExecutor, as_completed
 
 # Configuration and setup
 persistent_dir = "/data/hf_cache"
@@ -31,7 +32,6 @@ sys.path.insert(0, src_path)
 
 from txagent.txagent import TxAgent
 
-# Constants
 MAX_MODEL_TOKENS = 32768
 MAX_CHUNK_TOKENS = 8192
 MAX_NEW_TOKENS = 2048
@@ -131,8 +131,6 @@ def init_agent():
     return agent
 
 def process_final_report(agent, file, chatbot_state: List[Dict[str, str]]) -> Tuple[List[Dict[str, str]], Union[str, None]]:
-    from concurrent.futures import ThreadPoolExecutor, as_completed
-
     messages = chatbot_state if chatbot_state else []
     report_path = None
 
@@ -145,7 +143,7 @@ def process_final_report(agent, file, chatbot_state: List[Dict[str, str]]) -> Tu
         messages.append({"role": "assistant", "content": "⏳ Extracting and analyzing data..."})
 
         extracted_text = extract_text_from_excel(file.name)
-        chunks = split_text_into_chunks(extracted_text, max_tokens=MAX_CHUNK_TOKENS)
+        chunks = split_text_into_chunks(extracted_text)
         chunk_responses = [None] * len(chunks)
 
         def analyze_chunk(index: int, chunk: str) -> Tuple[int, str]:
@@ -153,7 +151,6 @@ def process_final_report(agent, file, chatbot_state: List[Dict[str, str]]) -> Tu
             prompt_tokens = estimate_tokens(prompt)
             if prompt_tokens > MAX_MODEL_TOKENS:
                 return index, f"❌ Chunk {index+1} prompt too long ({prompt_tokens} tokens). Skipping..."
-
             response = ""
             try:
                 for result in agent.run_gradio_chat(
@@ -175,10 +172,9 @@ def process_final_report(agent, file, chatbot_state: List[Dict[str, str]]) -> Tu
                                 response += r.content
             except Exception as e:
                 return index, f"❌ Error analyzing chunk {index+1}: {str(e)}"
-
             return index, clean_response(response)
 
-        with ThreadPoolExecutor(max_workers=4) as executor:
+        with ThreadPoolExecutor(max_workers=1) as executor:
             futures = [executor.submit(analyze_chunk, i, chunk) for i, chunk in enumerate(chunks)]
             for future in as_completed(futures):
                 i, result = future.result()
@@ -268,11 +264,11 @@ def process_final_report(agent, file, chatbot_state: List[Dict[str, str]]) -> Tu
         messages.append({"role": "assistant", "content": f"❌ Error processing file: {str(e)}"})
 
     return messages, report_path
+
 def create_ui(agent):
-    """Create the Gradio UI for the patient history analysis tool."""
     with gr.Blocks(
         title="Patient History Chat",
-        css="""
+        css=\"\"\"
         .gradio-container {
             max-width: 900px !important;
             margin: auto;
@@ -307,42 +303,24 @@ def create_ui(agent):
             padding-left: 1.2em;
             margin: 0.4em 0;
         }
-        """
+        \"\"\"
     ) as demo:
-        gr.Markdown("""
-        <h2 style='color:#182848'>🏥 Patient History Analysis Tool</h2>
-        <p style='color:#444;'>Upload an Excel file containing clinical data. The assistant will analyze it for patterns, inconsistencies, and recommendations.</p>
-        """)
+        gr.Markdown(\"\"\"\n<h2 style='color:#182848'>🏥 Patient History Analysis Tool</h2>\n<p style='color:#444;'>Upload an Excel file containing clinical data. The assistant will analyze it for patterns, inconsistencies, and recommendations.</p>\n\"\"\")
 
         with gr.Row():
             with gr.Column(scale=3):
                 chatbot = gr.Chatbot(
-                    label="Clinical Assistant",
+                    label=\"Clinical Assistant\",
                     show_copy_button=True,
                     height=600,
-                    type="messages",
-                    avatar_images=(
-                        None,
-                        "https://i.imgur.com/6wX7Zb4.png"
-                    ),
+                    type=\"messages\",
+                    avatar_images=(None, \"https://i.imgur.com/6wX7Zb4.png\"),
                     render_markdown=True
                 )
             with gr.Column(scale=1):
-                file_upload = gr.File(
-                    label="Upload Excel File",
-                    file_types=[".xlsx"],
-                    height=100
-                )
-                analyze_btn = gr.Button(
-                    "🧠 Analyze Patient History",
-                    variant="primary",
-                    elem_classes="primary"
-                )
-                report_output = gr.File(
-                    label="Download Report",
-                    visible=False,
-                    interactive=False
-                )
+                file_upload = gr.File(label=\"Upload Excel File\", file_types=[\".xlsx\"], height=100)
+                analyze_btn = gr.Button(\"🧠 Analyze Patient History\", variant=\"primary\", elem_classes=\"primary\")
+                report_output = gr.File(label=\"Download Report\", visible=False, interactive=False)
 
         chatbot_state = gr.State(value=[])
 
@@ -350,36 +328,24 @@ def create_ui(agent):
             messages, report_path = process_final_report(agent, file, current_state)
             formatted_messages = []
             for msg in messages:
-                role = msg.get("role")
-                content = msg.get("content", "")
-                if role == "assistant":
-                    content = content.replace("- ", "\n- ")  # Ensure bullet formatting
-                    content = f"<div class='chat-message-content'>{content}</div>"
-                formatted_messages.append({"role": role, "content": content})
+                role = msg.get(\"role\")
+                content = msg.get(\"content\", \"\")
+                if role == \"assistant\":
+                    content = content.replace(\"- \", \"\\n- \")
+                    content = f\"<div class='chat-message-content'>{content}</div>\"
+                formatted_messages.append({\"role\": role, \"content\": content})
             report_update = gr.update(visible=report_path is not None, value=report_path)
             return formatted_messages, report_update, formatted_messages
 
-        analyze_btn.click(
-            fn=update_ui,
-            inputs=[file_upload, chatbot_state],
-            outputs=[chatbot, report_output, chatbot_state],
-            api_name="analyze"
-        )
+        analyze_btn.click(fn=update_ui, inputs=[file_upload, chatbot_state], outputs=[chatbot, report_output, chatbot_state], api_name=\"analyze\")
 
     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,
-            show_error=True,
-            allowed_paths=["/data/hf_cache/reports"],
-            share=False
-        )
+        demo.launch(server_name=\"0.0.0.0\", server_port=7860, show_error=True, 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)