Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import requests
 import os
 from geopy.geocoders import Nominatim
-from smolagents import tool, CodeAgent, HfApiModel
+from smolagents import tool, CodeAgent, HfApiModel, GoogleSearchTool, VisitWebpageTool
 
 # Lấy API key từ biến môi trường
 WEATHER_API_KEY = os.environ.get("WEATHER_API_KEY")
@@ -15,6 +15,8 @@ def get_weather(lat: float, lon: float) -> dict | None:
     Args:
       lat: latitude of the location
       lon: longitude of the location
+
+    Return: dictionary about weather information or None.
     """
     url = f"https://api.openweathermap.org/data/2.5/weather?lat={lat}&lon={lon}&appid={WEATHER_API_KEY}&units=metric"
     try:
@@ -35,48 +37,16 @@ def get_weather(lat: float, lon: float) -> dict | None:
 # Khởi tạo agent
 weather_agent = CodeAgent(
     model=HfApiModel("deepseek-ai/DeepSeek-R1", max_tokens=8096),
-    tools=[get_weather],
+    tools=[get_weather, GoogleSearchTool(provider="serper"), VisitWebpageTool()],
+    name="weather_agent",
     description="Weather information provider"
 )
 
-# Hàm xử lý địa lý
-def get_coordinates(location: str) -> tuple:
-    geolocator = Nominatim(user_agent="weather_chatbot")
-    location = geolocator.geocode(location)
-    if location:
-        return (location.latitude, location.longitude)
-    return (None, None)
+
 
 # Hàm xử lý chat
 def respond(message, history):
-    # Chuyển đổi địa điểm thành tọa độ
-    lat, lon = get_coordinates(message)
-    
-    if not lat or not lon:
-        return "Không tìm thấy địa điểm. Vui lòng thử lại với tên chính xác hơn."
-    
-    # Gọi agent
-    result = weather_agent.run(f"""
-    Hãy phân tích thông tin thời tiết cho tọa độ {lat}, {lon} và trình bày kết quả theo định dạng:
-    - Điều kiện thời tiết
-    - Nhiệt độ
-    - Độ ẩm
-    - Tốc độ gió
-    """)
-    
-    # Xử lý kết quả
-    weather_data = get_weather(lat, lon)
-    if not weather_data:
-        return "Không thể lấy dữ liệu thời tiết. Vui lòng thử lại sau."
-    
-    response = f"""
-    🌤️ **Thông tin thời tiết cho {message}**:
-    - **Điều kiện**: {weather_data['condition'].capitalize()}
-    - **Nhiệt độ**: {weather_data['temperature']}°C
-    - **Độ ẩm**: {weather_data['humidity']}%
-    - **Gió**: {weather_data['wind_speed']} m/s
-    """
-    
+    response = weather_agent.run(message)
     return response
 
 # Tạo giao diện Gradio