Update app.py

app.py

@@ -4,11 +4,10 @@ from txagent import TxAgent
 from tooluniverse import ToolUniverse
 from importlib.resources import files
 
-# Configure logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-tx_app = None  # Global to hold the model
+tx_app = None  # Global TxAgent instance
 
 def init_txagent():
     logger.info("🔥 Initializing TxAgent...")
@@ -41,11 +40,14 @@ def init_txagent():
     )
 
     agent.init_model()
-    logger.info("✅ Model loading complete")
+    logger.info("✅ TxAgent fully initialized")
     return agent
 
 def respond(message, chat_history, temperature, max_new_tokens, max_tokens, multi_agent, conversation_state, max_round):
     global tx_app
+    if tx_app is None:
+        return chat_history + [("", "⚠️ Model not ready yet. Please wait a few seconds and try again.")]
+
     try:
         if not isinstance(message, str) or len(message.strip()) <= 10:
             return chat_history + [("", "Please provide a valid message longer than 10 characters.")]
@@ -78,17 +80,12 @@ def respond(message, chat_history, temperature, max_new_tokens, max_tokens, mult
         logger.error(f"Error in respond function: {str(e)}")
         yield chat_history + [("", f"⚠️ Error: {str(e)}")]
 
-# Gradio UI
+# ✅ Top-level app object that HF Spaces can detect
 with gr.Blocks(title="TxAgent Biomedical Assistant") as app:
     gr.Markdown("# 🧠 TxAgent Biomedical Assistant")
 
     chatbot = gr.Chatbot(label="Conversation", height=600, type="messages")
-
-    msg = gr.Textbox(
-        label="Your medical query",
-        placeholder="Enter your biomedical question...",
-        lines=3
-    )
+    msg = gr.Textbox(label="Your medical query", placeholder="Enter your biomedical question...", lines=3)
 
     with gr.Row():
         temp = gr.Slider(0, 1, value=0.3, label="Temperature")
@@ -99,7 +96,6 @@ with gr.Blocks(title="TxAgent Biomedical Assistant") as app:
 
     submit = gr.Button("Submit")
     clear = gr.Button("Clear")
-
     conversation_state = gr.State([])
 
     submit.click(
@@ -116,13 +112,13 @@ with gr.Blocks(title="TxAgent Biomedical Assistant") as app:
         chatbot
     )
 
-    # Hidden button + load trigger to init model safely
-    init_button = gr.Button(visible=False)
+    # ✅ hidden init trigger on page load
+    hidden_button = gr.Button(visible=False)
 
-    def load_model():
+    def initialize_agent():
         global tx_app
         tx_app = init_txagent()
         return gr.update(visible=False)
 
-    app.load(init_button.click(fn=load_model))
+    app.load(hidden_button.click(fn=initialize_agent))