Update app.py

app.py

@@ -201,13 +201,15 @@ class HeightComparisonAgent(CodeAgent):
 # --- Instantiate the Subclassed Agent ---
 # IMPORTANT: Use the HeightComparisonAgent class, not CodeAgent directly.
 # Set verbosity_level=3 so the parent's run method (super().run) generates the verbose output.
-height_agent = HeightComparisonAgent(
-    tools=[DuckDuckGoSearchTool(), VisitWebpageTool(), parse_height_from_text, create_comparison_statement, FinalAnswerTool()],
-    model=llm_model,
-    verbosity_level=3, # <<< Crucial for GradioUI to see the steps from the parent run
-    max_steps=20, # Increased slightly just in case
-    # planning_interval=3, # Optional
-)
+if llm_model is not None:
+    height_agent = HeightComparisonAgent(
+        tools=[DuckDuckGoSearchTool(), VisitWebpageTool(), parse_height_from_text, create_comparison_statement, FinalAnswerTool()],
+        model=llm_model,
+        verbosity_level=3, # <<< Crucial for GradioUI to see the steps from the parent run
+        max_steps=20, # Increased slightly just in case
+        # planning_interval=3, # Optional
+    )
+else print('Failed to Load LiteLLM Model')
 
 # --- Launch Gradio using GradioUI and the custom agent ---
 print("--- Starting Gradio Interface with GradioUI and HeightComparisonAgent ---")