Update app.py

app.py

@@ -6,7 +6,6 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 # ==================== YOUR TOOLS ====================
-# Existing tool from Unit 1
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -20,7 +19,6 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-# NEW TOOL 1: Tip Calculator
 @tool
 def calculate_tip(bill_amount: float, tip_percent: float) -> str:
     """Calculate tip amount for a restaurant bill.
@@ -35,7 +33,6 @@ def calculate_tip(bill_amount: float, tip_percent: float) -> str:
     except Exception as e:
         return f"Error: {str(e)}"
 
-# NEW TOOL 2: Word Counter
 @tool
 def word_counter(text: str) -> str:
     """Count words and characters in text.
@@ -49,52 +46,54 @@ def word_counter(text: str) -> str:
     except Exception as e:
         return f"Error: {str(e)}"
 
-# NEW TOOL 3: BMI Calculator
-@tool
-def calculate_bmi(weight_kg: float, height_m: float) -> str:
-    """Calculate Body Mass Index (BMI).
-    Args:
-        weight_kg: Weight in kilograms
-        height_m: Height in meters
-    """
-    try:
-        bmi = weight_kg / (height_m ** 2)
-        if bmi < 18.5:
-            cat = "Underweight"
-        elif 18.5 <= bmi < 25:
-            cat = "Normal"
-        elif 25 <= bmi < 30:
-            cat = "Overweight"
-        else:
-            cat = "Obese"
-        return f"⚖️ BMI: {bmi:.1f} ({cat})"
-    except Exception as e:
-        return f"Error: {str(e)}"
-
 # ==================== AGENT SETUP ====================
 final_answer = FinalAnswerTool()
 
-# Load image generation tool from Hub
+# Load image generation tool
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
+# ⚠️ FIX: ADD THE MISSING MODEL SETUP
+# Try different model import approaches:
+
+# OPTION 1: Use HfApiModel (common in spaces)
+try:
+    from smolagents import HfApiModel
+    model = HfApiModel(
+        model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+        max_tokens=512,
+        temperature=0.5
+    )
+except ImportError:
+    # OPTION 2: Use InferenceClientModel with correct import path
+    try:
+        from smolagents.models import InferenceClientModel
+        model = InferenceClientModel(
+            model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+            max_tokens=512,
+            temperature=0.5
+        )
+    except ImportError:
+        # OPTION 3: Use default model
+        from smolagents.models import DefaultModel
+        model = DefaultModel()
+
 # Load system prompt
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
 # ==================== CREATE AGENT ====================
-# SIMPLIFIED: Use default model instead of InferenceClientModel
 agent = CodeAgent(
+    model=model,  # ⚠️ THIS WAS MISSING!
     tools=[
-        final_answer,                    # Required
-        DuckDuckGoSearchTool(),          # Search
-        image_generation_tool,           # Images
-        get_current_time_in_timezone,    # Time
-        calculate_tip,                   # NEW
-        word_counter,                    # NEW
-        calculate_bmi                    # NEW
+        final_answer,
+        DuckDuckGoSearchTool(),
+        image_generation_tool,
+        get_current_time_in_timezone,
+        calculate_tip,
+        word_counter
     ],
-    max_steps=8,                        # Increased
-    verbosity_level=1,                  # Show thinking
+    max_steps=8,
+    verbosity_level=1,
     prompt_templates=prompt_templates
 )