Update app.py

app.py

@@ -1,10 +1,6 @@
-# ============================================
-# ALFRED - Your AI Butler Agent
-# HuggingFace AI Agents Course - Unit 1
-# ============================================
-
+import os
 import gradio as gr
-from smolagents import CodeAgent, DuckDuckGoSearchTool, VisitWebpageTool, HfApiModel, tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel, tool
 
 # ============================================
 # CUSTOM TOOLS FOR ALFRED
@@ -22,198 +18,142 @@ def calculator(operation: str) -> str:
         The result of the calculation as a string
     """
     try:
-        # Only allow safe mathematical operations
         allowed_chars = set('0123456789+-*/(). ')
         if all(c in allowed_chars for c in operation):
             result = eval(operation)
             return f"The result of {operation} is: {result}"
         else:
-            return "Invalid operation. Please use only numbers and basic operators (+, -, *, /)."
+            return "Invalid operation. Please use only numbers and basic operators."
     except Exception as e:
-        return f"Could not calculate that expression. Error: {str(e)}"
+        return f"Could not calculate. Error: {str(e)}"
 
 
 @tool
-def get_current_time(timezone: str = "UTC") -> str:
+def get_current_time() -> str:
     """
-    Gets the current date and time.
-    
-    Args:
-        timezone: The timezone to get time for (default is UTC)
+    Gets the current date and time in UTC.
     
     Returns:
         The current date and time as a string
     """
     from datetime import datetime
-    try:
-        now = datetime.utcnow()
-        return f"The current date and time (UTC) is: {now.strftime('%Y-%m-%d %H:%M:%S')}"
-    except Exception as e:
-        return f"Could not get the time. Error: {str(e)}"
+    now = datetime.utcnow()
+    return f"Current date and time (UTC): {now.strftime('%Y-%m-%d %H:%M:%S')}"
 
 
-@tool  
-def text_summarizer(text: str) -> str:
+@tool
+def get_weather_info(city: str) -> str:
     """
-    Summarizes a given text into key points.
+    Gets a simple weather description for a city (mock data for demo).
     
     Args:
-        text: The text to summarize
+        city: The name of the city to get weather for
     
     Returns:
-        A summarized version of the text
+        A weather description string
     """
-    if len(text) < 100:
-        return f"Text is already short: {text}"
-    
-    # Simple summarization: return first 500 characters with ellipsis
-    summary = text[:500] + "..." if len(text) > 500 else text
-    return f"Summary: {summary}"
+    # This is mock data - in production you'd use a real weather API
+    import random
+    conditions = ["sunny", "partly cloudy", "cloudy", "rainy", "windy"]
+    temp = random.randint(15, 30)
+    condition = random.choice(conditions)
+    return f"Weather in {city}: {temp}°C, {condition}"
 
 
 # ============================================
-# ALFRED AGENT SETUP
+# MODEL SETUP - Using free-tier compatible model
 # ============================================
 
-# Custom system prompt to give Alfred his personality
-ALFRED_SYSTEM_PROMPT = """You are Alfred, a polite, professional, and highly capable AI butler assistant.
-
-Your characteristics:
-- You are courteous and address users respectfully
-- You are thorough and provide complete answers
-- You use your available tools wisely to help users
-- You explain your reasoning when solving problems
-- You admit when you don't know something
-
-Available tools:
-1. DuckDuckGoSearchTool - Search the web for information
-2. VisitWebpageTool - Visit and read webpage content
-3. calculator - Perform mathematical calculations
-4. get_current_time - Get the current date and time
-5. text_summarizer - Summarize long texts
-
-Always think step by step and use the most appropriate tool for each task.
-"""
+model = InferenceClientModel(
+    model_id="Qwen/Qwen2.5-7B-Instruct",  # Smaller model that works on free tier
+    token=os.environ.get("HF_TOKEN"),
+)
 
-# Initialize the language model (using HuggingFace's free inference API)
-model = HfApiModel()
+# ============================================
+# ALFRED AGENT SETUP
+# ============================================
 
-# Create Alfred with all tools
 alfred = CodeAgent(
     tools=[
         DuckDuckGoSearchTool(),
-        VisitWebpageTool(),
         calculator,
         get_current_time,
-        text_summarizer,
+        get_weather_info,
     ],
     model=model,
-    system_prompt=ALFRED_SYSTEM_PROMPT,
-    max_steps=10,  # Maximum number of steps Alfred can take
-    verbosity_level=1  # Show some logging for debugging
+    max_steps=5,
+    verbosity_level=1,
 )
 
-
 # ============================================
-# GRADIO INTERFACE
+# CHAT FUNCTION
 # ============================================
 
 def chat_with_alfred(message, history):
     """
     Process user message and return Alfred's response.
-    
-    Args:
-        message: The user's input message
-        history: Chat history (for context)
-    
-    Returns:
-        Alfred's response as a string
     """
     if not message.strip():
-        return "I beg your pardon, but you haven't asked me anything. How may I assist you?"
+        return "Please ask me something. How may I assist you today?"
     
     try:
-        # Run Alfred on the user's query
         response = alfred.run(message)
-        
-        # Format the response nicely
-        formatted_response = f"🎩 **Alfred's Response:**\n\n{str(response)}"
-        return formatted_response
-        
+        return str(response)
     except Exception as e:
-        error_message = (
-            f"I do apologize, but I encountered an issue while processing your request.\n\n"
-            f"**Error details:** {str(e)}\n\n"
-            f"Please try rephrasing your question or ask something else."
-        )
-        return error_message
-
-
-# Create the Gradio interface
-with gr.Blocks(
-    title="Alfred - AI Butler Agent",
-    theme=gr.themes.Soft()
-) as demo:
+        error_msg = str(e)
+        # Provide helpful error messages
+        if "404" in error_msg or "not found" in error_msg.lower():
+            return "I'm having trouble connecting to my AI backend. Please check the HF_TOKEN and model availability."
+        elif "rate" in error_msg.lower() or "limit" in error_msg.lower():
+            return "I've hit a rate limit. Please wait a moment and try again."
+        else:
+            return f"I encountered an issue: {error_msg}"
+
+
+# ============================================
+# GRADIO INTERFACE
+# ============================================
+
+with gr.Blocks(title="Alfred - AI Butler Agent") as demo:
     
-    # Header
     gr.Markdown(
         """
         # 🎩 Alfred - Your AI Butler
         
-        Welcome! I am Alfred, your personal AI assistant. I can help you with:
-        - 🔍 **Web searches** - Find information on any topic
-        - 🌐 **Reading webpages** - Extract content from URLs
-        - 🔢 **Calculations** - Perform mathematical operations
-        - 🕐 **Time queries** - Get current date and time
-        - 📝 **Text summarization** - Summarize long texts
+        Good day! I am Alfred, your personal AI assistant. I am at your service.
         
-        *How may I be of service today?*
+        ### My Capabilities:
+        - 🔍 *Web Search* - I can search the internet for information
+        - 🧮 *Calculator* - I can perform mathematical calculations
+        - 🕐 *Current Time* - I can tell you the current time (UTC)
+        - 🌤️ *Weather* - I can provide weather information
         
         ---
         """
     )
     
-    # Chat interface
     chatbot = gr.ChatInterface(
         fn=chat_with_alfred,
         examples=[
-            "What is the latest news about artificial intelligence?",
-            "Calculate 1547 * 32 + 89",
+            "What is artificial intelligence?",
+            "Calculate 25 * 4 + 10",
             "What time is it now?",
-            "Search for information about climate change solutions",
-            "What are the top programming languages in 2024?",
+            "What's the weather in London?",
+            "Search for the latest news about AI",
+            "Calculate (100 + 50) * 2 / 5",
         ],
-        retry_btn="🔄 Retry",
-        undo_btn="↩️ Undo",
-        clear_btn="🗑️ Clear",
     )
     
-    # Footer
     gr.Markdown(
         """
         ---
-        
-        ### About Alfred
-        
-        Alfred is an AI agent built using the **smolagents** framework from HuggingFace.
-        This agent was created as part of the **HuggingFace AI Agents Course - Unit 1**.
-        
-        **Framework:** smolagents | **Model:** HuggingFace Inference API
-        
-        *Built with ❤️ for the AI Agents Course*
+        Built with [smolagents](https://github.com/huggingface/smolagents) and Gradio | Unit 1 Project
         """
     )
 
-
 # ============================================
-# LAUNCH THE APPLICATION
+# LAUNCH APP
 # ============================================
 
-if __name__ == "__main__":
-    # Launch with share=False for local testing
-    # Set share=True if you want a public link
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=False
-    )
+if _name_ == "_main_":
+    demo.launch()