Update app.py

app.py

@@ -1,22 +1,75 @@
 import os
+import requests
+from smolagents import LiteLLMModel, ToolCallingAgent, Tool
+from smolagents.webui import GradioUI
+import wikipedia
 
+# Optional: Hugging Face token (for private models)
 HF_TOKEN = os.getenv("HF_TOKEN")
 
-import os
-import datetime
-from tools.final_answer import FinalAnswerTool
-from Gradio_UI import GradioUI
+# --- 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):
+        from smolagents import DuckDuckGoSearchTool
+        tool = DuckDuckGoSearchTool()
+        return tool.run(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):
+        try:
+            summary = wikipedia.summary(topic, sentences=3)
+            return summary
+        except Exception as e:
+            return f"Wikipedia lookup failed: {e}"
 
-from smolagents import tool, LiteLLMModel, ToolCallingAgent, DuckDuckGoSearchTool  # or langchain.tools import tool
 
-final_answer = FinalAnswerTool()
+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']}."
+
+
+# --- Initialize LLM Model ---
 model = LiteLLMModel(
-    model_id="huggingface/google/gemma-2-2b-it"
+    model_id="huggingface/google/gemma-2-2b-it",
+    hf_token=HF_TOKEN
 )
 
+# --- Initialize Tool-Calling Agent ---
 agent = ToolCallingAgent(
-    tools=[DuckDuckGoSearchTool()],
+    tools=[WebSearchTool(), WikipediaTool(), WeatherTool()],
     model=model,
-    max_steps=10
+    max_steps=10,
+    name="Internet Agent"
 )
-GradioUI(agent).launch()
+
+# --- Launch Gradio Web UI ---
+GradioUI(agent).launch()