Update app.py

app.py

@@ -1,61 +1,60 @@
 import os
 import requests
 from smolagents import LiteLLMModel, ToolCallingAgent, Tool
-from smolagents.webui import GradioUI
 import wikipedia
+import gradio as gr
 
 # Optional: Hugging Face token (for private models)
 HF_TOKEN = os.getenv("HF_TOKEN")
 
 # --- Tools ---
-
 class WebSearchTool(Tool):
     """Search the web using DuckDuckGo"""
     name = "web_search"
     description = "Search the web and return concise results. Input: search query string."
-
-    def run(self, query: str):
+    
+    def forward(self, query: str):
         from smolagents import DuckDuckGoSearchTool
         tool = DuckDuckGoSearchTool()
-        return tool.run(query)
-
+        return tool.forward(query)
 
 class WikipediaTool(Tool):
     """Fetch summaries from Wikipedia"""
     name = "wikipedia_search"
     description = "Fetch Wikipedia summary for a topic. Input: topic string."
-
-    def run(self, topic: str):
+    
+    def forward(self, topic: str):
         try:
             summary = wikipedia.summary(topic, sentences=3)
             return summary
         except Exception as e:
             return f"Wikipedia lookup failed: {e}"
 
-
 class WeatherTool(Tool):
     """Get weather using Open-Meteo API"""
     name = "weather"
     description = "Get current weather. Input: city name."
-
-    def run(self, city: str):
-        # Geocoding to get coordinates
-        geocode_url = f"https://geocoding-api.open-meteo.com/v1/search?name={city}"
-        geo_resp = requests.get(geocode_url, timeout=10).json()
-        results = geo_resp.get("results")
-        if not results:
-            return f"Could not find coordinates for {city}."
-        lat = results[0]["latitude"]
-        lon = results[0]["longitude"]
-
-        # Get current weather
-        weather_url = f"https://api.open-meteo.com/v1/forecast?latitude={lat}&longitude={lon}&current_weather=true"
-        weather_resp = requests.get(weather_url, timeout=10).json()
-        weather = weather_resp.get("current_weather")
-        if not weather:
-            return f"Could not retrieve weather for {city}."
-        return f"Weather in {city}: {weather['temperature']}°C, wind {weather['windspeed']} km/h, weather code {weather['weathercode']}."
-
+    
+    def forward(self, city: str):
+        try:
+            # Geocoding to get coordinates
+            geocode_url = f"https://geocoding-api.open-meteo.com/v1/search?name={city}"
+            geo_resp = requests.get(geocode_url, timeout=10).json()
+            results = geo_resp.get("results")
+            if not results:
+                return f"Could not find coordinates for {city}."
+            lat = results[0]["latitude"]
+            lon = results[0]["longitude"]
+            
+            # Get current weather
+            weather_url = f"https://api.open-meteo.com/v1/forecast?latitude={lat}&longitude={lon}&current_weather=true"
+            weather_resp = requests.get(weather_url, timeout=10).json()
+            weather = weather_resp.get("current_weather")
+            if not weather:
+                return f"Could not retrieve weather for {city}."
+            return f"Weather in {city}: {weather['temperature']}°C, wind {weather['windspeed']} km/h, weather code {weather['weathercode']}."
+        except Exception as e:
+            return f"Weather lookup failed: {e}"
 
 # --- Initialize LLM Model ---
 model = LiteLLMModel(
@@ -68,8 +67,34 @@ agent = ToolCallingAgent(
     tools=[WebSearchTool(), WikipediaTool(), WeatherTool()],
     model=model,
     max_steps=10,
-    name="Internet Agent"
+)
+
+# --- Custom Gradio Interface ---
+def chat_with_agent(message, history):
+    """Process user message and return agent response"""
+    try:
+        result = agent.run(message)
+        return str(result)
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# Create Gradio ChatInterface
+demo = gr.ChatInterface(
+    fn=chat_with_agent,
+    title="🤖 Internet Agent",
+    description="An AI agent with web search, Wikipedia, and weather tools powered by Gemma-2-2b",
+    examples=[
+        "What's the weather in Paris?",
+        "Search for recent news about AI",
+        "Tell me about Albert Einstein from Wikipedia",
+        "What's the current temperature in Tokyo?"
+    ],
+    theme=gr.themes.Soft(),
+    retry_btn=None,
+    undo_btn=None,
+    clear_btn="Clear Chat",
 )
 
 # --- Launch Gradio Web UI ---
-GradioUI(agent).launch()
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=7860)