Update app.py

app.py

@@ -1,11 +1,11 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel, load_tool, tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, load_tool, tool
 import datetime
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# ==================== ✨ LOCATION 1: YOUR TOOLS ✨ ====================
+# ==================== YOUR TOOLS ====================
 # Existing tool from Unit 1
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -14,15 +14,13 @@ def get_current_time_in_timezone(timezone: str) -> str:
         timezone: A string representing a valid timezone (e.g., 'America/New_York').
     """
     try:
-        # Create timezone object
         tz = pytz.timezone(timezone)
-        # Get current time in that timezone
         local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
         return f"The current local time in {timezone} is: {local_time}"
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-# ✨ NEW TOOL 1: Tip Calculator
+# NEW TOOL 1: Tip Calculator
 @tool
 def calculate_tip(bill_amount: float, tip_percent: float) -> str:
     """Calculate tip amount for a restaurant bill.
@@ -33,11 +31,11 @@ def calculate_tip(bill_amount: float, tip_percent: float) -> str:
     try:
         tip = bill_amount * (tip_percent / 100)
         total = bill_amount + tip
-        return f"💰 Tip Amount: ${tip:.2f}\n💵 Total Bill: ${total:.2f}\n📊 Tip Percentage: {tip_percent}%"
+        return f"💰 Tip: ${tip:.2f}, Total: ${total:.2f} ({tip_percent}%)"
     except Exception as e:
-        return f"Error calculating tip: {str(e)}"
+        return f"Error: {str(e)}"
 
-# ✨ NEW TOOL 2: Word Counter
+# NEW TOOL 2: Word Counter
 @tool
 def word_counter(text: str) -> str:
     """Count words and characters in text.
@@ -46,13 +44,12 @@ def word_counter(text: str) -> str:
     """
     try:
         words = len(text.split())
-        characters = len(text)
-        sentences = text.count('.') + text.count('!') + text.count('?')
-        return f"📝 Text Analysis:\n• Words: {words}\n• Characters: {characters}\n• Sentences: {sentences}"
+        chars = len(text)
+        return f"📝 Words: {words}, Characters: {chars}"
     except Exception as e:
-        return f"Error analyzing text: {str(e)}"
+        return f"Error: {str(e)}"
 
-# ✨ NEW TOOL 3: BMI Calculator
+# NEW TOOL 3: BMI Calculator
 @tool
 def calculate_bmi(weight_kg: float, height_m: float) -> str:
     """Calculate Body Mass Index (BMI).
@@ -62,91 +59,44 @@ def calculate_bmi(weight_kg: float, height_m: float) -> str:
     """
     try:
         bmi = weight_kg / (height_m ** 2)
-        
         if bmi < 18.5:
-            category = "Underweight"
+            cat = "Underweight"
         elif 18.5 <= bmi < 25:
-            category = "Normal weight"
+            cat = "Normal"
         elif 25 <= bmi < 30:
-            category = "Overweight"
+            cat = "Overweight"
         else:
-            category = "Obese"
-            
-        return f"⚖️ BMI Result: {bmi:.1f}\n📋 Category: {category}"
+            cat = "Obese"
+        return f"⚖️ BMI: {bmi:.1f} ({cat})"
     except Exception as e:
-        return f"Error calculating BMI: {str(e)}"
+        return f"Error: {str(e)}"
 
-# ✨ NEW TOOL 4: Currency Converter (Simulated)
-@tool
-def convert_currency(amount: float, from_currency: str, to_currency: str) -> str:
-    """Convert currency between different types (simulated rates).
-    Args:
-        amount: Amount to convert
-        from_currency: Source currency (USD, EUR, JPY, etc.)
-        to_currency: Target currency (USD, EUR, JPY, etc.)
-    """
-    # Simulated exchange rates (for demo purposes)
-    rates = {
-        'USD': 1.0,
-        'EUR': 0.85,
-        'JPY': 110.0,
-        'GBP': 0.75,
-        'CAD': 1.25,
-        'AUD': 1.35,
-        'INR': 75.0
-    }
-    
-    try:
-        if from_currency.upper() not in rates or to_currency.upper() not in rates:
-            return f"❌ Currency not supported. Available: {', '.join(rates.keys())}"
-        
-        rate = rates[to_currency.upper()] / rates[from_currency.upper()]
-        converted = amount * rate
-        
-        return f"💱 Currency Conversion:\n{amount} {from_currency.upper()} = {converted:.2f} {to_currency.upper()}\n📈 Exchange Rate: 1 {from_currency.upper()} = {rate:.4f} {to_currency.upper()}"
-    except Exception as e:
-        return f"Error converting currency: {str(e)}"
-
-# ==================== ✨ AGENT SETUP ✨ ====================
+# ==================== AGENT SETUP ====================
 final_answer = FinalAnswerTool()
 
 # Load image generation tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-# Create model
-model = InferenceClientModel(
-    max_tokens=2096,
-    temperature=0.5,
-    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-    custom_role_conversions=None,
-)
-
-# Load system prompt from prompts.yaml file
+# Load system prompt
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
-# ==================== ✨ LOCATION 2: AGENT WITH ALL TOOLS ✨ ====================
-# ✨ UPGRADED: Added 5 new tools and increased max_steps
+# ==================== CREATE AGENT ====================
+# SIMPLIFIED: Use default model instead of InferenceClientModel
 agent = CodeAgent(
-    model=model,
     tools=[
-        final_answer,                    # Required - DON'T REMOVE
-        DuckDuckGoSearchTool(),          # Web search capability
-        image_generation_tool,           # Image generation
-        get_current_time_in_timezone,    # Existing time tool
-        calculate_tip,                   # ✨ NEW: Tip calculator
-        word_counter,                    # ✨ NEW: Word counter
-        calculate_bmi,                   # ✨ NEW: BMI calculator
-        convert_currency                 # ✨ NEW: Currency converter
+        final_answer,                    # Required
+        DuckDuckGoSearchTool(),          # Search
+        image_generation_tool,           # Images
+        get_current_time_in_timezone,    # Time
+        calculate_tip,                   # NEW
+        word_counter,                    # NEW
+        calculate_bmi                    # NEW
     ],
-    max_steps=10,                       # ✨ UPGRADED: Was 6, now 10
-    verbosity_level=2,                  # ✨ NEW: Shows detailed thinking
-    grammar=None,
-    planning_interval=None,
-    name="SuperAssistant Pro",
-    description="An enhanced AI agent with multiple useful tools",
+    max_steps=8,                        # Increased
+    verbosity_level=1,                  # Show thinking
     prompt_templates=prompt_templates
 )
 
-# ==================== ✨ LAUNCH ✨ ====================
+# ==================== LAUNCH ====================
 GradioUI(agent).launch()