Create app.py

app.py

@@ -0,0 +1,149 @@
+import gradio as gr
+from smolagents import CodeAgent, DuckDuckGoSearchTool, tool, InferenceClientModel
+import datetime
+import pytz
+import math
+import os
+
+# ============ TOOLS ============
+
+@tool
+def web_search_tool(query: str) -> str:
+    """Search the web for current information using DuckDuckGo.
+    
+    Args:
+        query: The search query string to look up.
+    """
+    try:
+        search = DuckDuckGoSearchTool()
+        results = search(query)
+        if not results:
+            return "No results found for your query."
+        return results
+    except Exception as e:
+        return f"Search error: {str(e)}"
+
+@tool
+def time_tool(timezone: str = "UTC") -> str:
+    """Get current time in a specific timezone.
+    
+    Args:
+        timezone: The timezone to check (e.g., 'Asia/Tokyo') or 'list' to see options.
+    """
+    try:
+        if timezone.lower() == "list":
+            timezones = ["UTC", "America/New_York", "Europe/London", 
+                        "Asia/Tokyo", "Australia/Sydney", "Europe/Paris"]
+            return "Available timezones:\n" + "\n".join(f"• {tz}" for tz in timezones)
+        
+        tz = pytz.timezone(timezone)
+        time = datetime.datetime.now(tz)
+        return (f"⏰ Time in {timezone}:\n"
+                f"• Date: {time.strftime('%Y-%m-%d')}\n"
+                f"• Time: {time.strftime('%H:%M:%S')}\n"
+                f"• Day: {time.strftime('%A')}")
+    except Exception as e:
+        return f"Error: {str(e)}. Try 'list' to see available timezones."
+
+@tool
+def calculator_tool(expression: str) -> str:
+    """Perform mathematical calculations.
+    
+    Args:
+        expression: The math expression to evaluate (e.g., '2 + 2' or 'sqrt(16)').
+    """
+    try:
+        # Simple cleanup for user-friendly operators
+        expr = expression.replace('^', '**').replace('×', '*').replace('÷', '/')
+        safe_dict = {
+            'abs': abs, 'round': round, 'min': min, 'max': max,
+            'sqrt': math.sqrt, 'pow': pow, 'sin': math.sin, 
+            'cos': math.cos, 'tan': math.tan, 'pi': math.pi, 'e': math.e
+        }
+        result = eval(expr, {"__builtins__": {}}, safe_dict)
+        return f"🧮 Calculation Result: {result}"
+    except Exception as e:
+        return f"Calculation error: {str(e)}"
+
+@tool
+def unit_converter_tool(query: str) -> str:
+    """Convert between different units of measurement.
+    
+    Args:
+        query: The conversion string (e.g., '10 km to miles').
+    """
+    try:
+        parts = query.lower().split()
+        if len(parts) < 4:
+            return "Format: '10 km to miles' or '100 celsius to fahrenheit'"
+        
+        value = float(parts[0])
+        from_unit = parts[1]
+        to_unit = parts[3]
+        
+        conversions = {
+            'km': {'miles': 0.621371, 'meters': 1000},
+            'miles': {'km': 1.60934, 'meters': 1609.34},
+            'meters': {'km': 0.001, 'miles': 0.000621371},
+            'kg': {'pounds': 2.20462, 'grams': 1000},
+            'pounds': {'kg': 0.453592},
+            'liters': {'gallons': 0.264172}
+        }
+        
+        if from_unit == 'celsius' and to_unit == 'fahrenheit':
+            result = (value * 9/5) + 32
+        elif from_unit == 'fahrenheit' and to_unit == 'celsius':
+            result = (value - 32) * 5/9
+        elif from_unit in conversions and to_unit in conversions[from_unit]:
+            result = value * conversions[from_unit][to_unit]
+        else:
+            return f"Can't convert {from_unit} to {to_unit}."
+            
+        return f"📏 {value} {from_unit} = {result:.4f} {to_unit}"
+    except Exception as e:
+        return f"Conversion error: {str(e)}"
+
+# ============ AGENT SETUP ============
+
+HF_TOKEN = os.getenv("HUGGINGFACE_TOKEN", "")
+
+# Using HfApiModel ensures the correct message format (list of dicts)
+# Qwen-2.5-Coder is highly recommended for CodeAgent tool use
+model = InferenceClientModel(
+    model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+    token=HF_TOKEN # Make sure your token has "Inference" permissions
+)
+
+agent = CodeAgent(
+    model=model,
+    tools=[web_search_tool, time_tool, calculator_tool],
+    max_steps=5,
+    additional_authorized_imports=['math', 'datetime', 'pytz']
+)
+
+# ============ GRADIO UI ============
+
+def chat_with_agent(message, history):
+    if not HF_TOKEN:
+        return "Please set your HUGGINGFACE_TOKEN as an environment variable to use the AI."
+    try:
+        # agent.run returns the final answer string
+        response = agent.run(message)
+        return str(response)
+    except Exception as e:
+        return f"Agent Error: {str(e)}"
+
+demo = gr.ChatInterface(
+    fn=chat_with_agent,
+    title="🤖 Alfred AI Assistant",
+    description="I am Alfred, your tool-equipped assistant. I can search the web, calculate math, convert units, and check the time!",
+    examples=[
+        "What time is it in Tokyo?",
+        "Convert 150 pounds to kg",
+        "Calculate the square root of 144 plus 50",
+        "Search for the latest news on AI agents"
+    ],
+)
+
+if __name__ == "__main__":
+    demo.launch(theme=gr.themes.Soft())