Added model warmup

app.py

@@ -565,6 +565,25 @@ def clear_chat():
     system_prompt_initialized = False
     return [], ""
 
+def warmup_agent():
+    """Warm up the agent with a test query to preload everything."""
+    logger.info("Warming up agent with test query...")
+    try:
+        current_agent = get_agent()
+        
+        # Initialize system prompt
+        initialize_system_prompt(current_agent)
+        
+        # Run a simple test query
+        test_response = current_agent.run(input="Hello, this is a warmup test.")
+        logger.info(f"Agent warmup completed successfully! Test response length: {len(test_response)} chars")
+        
+        # Clear the test interaction from memory
+        current_agent.memory.clear()
+        
+    except Exception as e:
+        logger.error(f"Agent warmup failed: {e}")
+    
 # --- UI: Interface Creation ---
 def create_interface():
     """Creates and configures the complete Gradio interface."""
@@ -634,9 +653,24 @@ def create_interface():
         return demo
 
 # --- Main Execution ---
+# At the end of your app.py file, replace the main execution block:
 if __name__ == "__main__":
     try:
-        logger.info("Starting Mimir...")
+        logger.info("=" * 50)
+        logger.info("Starting Mimir Application")
+        logger.info("=" * 50)
+        
+        # Step 1: Preload the model and agent
+        logger.info("Loading AI model...")
+        start_time = time.time()
+        agent = create_langchain_agent()
+        load_time = time.time() - start_time
+        logger.info(f"Model loaded successfully in {load_time:.2f} seconds")
+        
+        # Step 2: Warm up the model
+        logger.info("Warming up model...")
+        warmup_agent()
+        
         interface = create_interface()
         interface.queue()
         interface.launch(
@@ -645,5 +679,7 @@ if __name__ == "__main__":
             debug=True,
             favicon_path="assets/favicon.ico"
         )
+        
     except Exception as e:
-        logger.error(f"Failed to launch Mimir: {e}")
+        logger.error(f"❌ Failed to launch Mimir: {e}")
+        raise