Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 import sys
 import random
 import gradio as gr
+from multiprocessing import freeze_support  # Added for multiprocessing support
 
 # Add src to path
 sys.path.append(os.path.join(os.path.dirname(__file__), "src"))
@@ -62,19 +63,6 @@ question_examples = [
     ["A 30-year-old patient is on Prozac for depression and now diagnosed with WHIM syndrome. Is Xolremdi suitable?"]
 ]
 
-# Initialize agent
-agent = TxAgent(
-    model_name,
-    rag_model_name,
-    tool_files_dict=new_tool_files,
-    force_finish=True,
-    enable_checker=True,
-    step_rag_num=10,
-    seed=100,
-    additional_default_tools=["DirectResponse", "RequireClarification"]
-)
-agent.init_model()
-
 # === Create Gradio UI ===
 def create_ui():
     with gr.Blocks(css=css) as demo:
@@ -127,9 +115,8 @@ def create_ui():
 
         chatbot.retry(
             handle_retry,
-            chatbot, chatbot,
-            temperature, max_new_tokens, max_tokens,
-            multi_agent, conversation_state, max_round
+            inputs=[chatbot, chatbot, temperature, max_new_tokens, max_tokens, multi_agent, conversation_state, max_round],
+            outputs=chatbot
         )
 
         # === Chat Interface setup
@@ -152,8 +139,7 @@ def create_ui():
     return demo
 
 if __name__ == "__main__":
-    # Initialize multiprocessing safely
-    freeze_support()
+    freeze_support()  # Initialize multiprocessing support
     
     # Initialize agent
     agent = TxAgent(