Update app.py

app.py

@@ -4,8 +4,9 @@ import os
 from geopy.geocoders import Nominatim
 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")
+# Lấy API keys từ biến môi trường
+OPENWEATHER_API_KEY = os.environ.get("OPENWEATHER_API_KEY")
+SERPER_API_KEY = os.environ.get("SERPER_API_KEY")
 
 # Định nghĩa tool get_weather
 @tool
@@ -15,10 +16,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"
+    url = f"https://api.openweathermap.org/data/2.5/weather?lat={lat}&lon={lon}&appid={OPENWEATHER_API_KEY}&units=metric"
     try:
         response = requests.get(url)
         data = response.json()
@@ -34,20 +33,47 @@ def get_weather(lat: float, lon: float) -> dict | None:
         print(f"Weather API Error: {e}")
         return None
 
+# Hàm chuyển đổi địa điểm thành tọa độ
+@tool
+def get_coordinates(location: str) -> tuple[float, float] | None:
+    """Convert location name to geographic coordinates"""
+    geolocator = Nominatim(user_agent="weather_agent")
+    location = geolocator.geocode(location)
+    return (location.latitude, location.longitude) if location else None
+
 # Khởi tạo agent
 weather_agent = CodeAgent(
     model=HfApiModel("deepseek-ai/DeepSeek-R1", max_tokens=8096),
-    tools=[get_weather, GoogleSearchTool(provider="serper"), VisitWebpageTool()],
+    tools=[
+        get_weather,
+        get_coordinates,
+        GoogleSearchTool(provider="serper", api_key=SERPER_API_KEY),
+        VisitWebpageTool()
+    ],
     name="weather_agent",
-    description="Weather information provider"
+    description="Weather information provider",
+    additional_authorized_imports=["geopy"]
 )
 
-
-
 # Hàm xử lý chat
 def respond(message, history):
-    response = weather_agent.run(message)
-    return response
+    try:
+        # Thêm prompt engineering để hướng dẫn agent
+        system_prompt = """Bạn là trợ lý thời tiết chuyên nghiệp. Hãy:
+        1. Xác định tọa độ địa lý từ tên địa điểm
+        2. Lấy thông tin thời tiết từ API
+        3. Trả lời bằng tiếng Việt với định dạng markdown
+        4. Kiểm tra độ chính xác của dữ liệu"""
+        
+        response = weather_agent.run(
+            f"{system_prompt}\n\nYêu cầu người dùng: {message}"
+        )
+        
+        return response if response else "Không thể lấy thông tin thời tiết"
+    
+    except Exception as e:
+        print(f"Error: {e}")
+        return "Đã xảy ra lỗi khi xử lý yêu cầu"
 
 # Tạo giao diện Gradio
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
@@ -57,7 +83,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     chatbot = gr.ChatInterface(
         respond,
         examples=["Hà Nội", "New York", "Tokyo", "Paris"],
-        title="Weather Chatbot"
+        title="Weather Chatbot",
+        additional_inputs_accordion_name="Advanced Options"
     )
 
 if __name__ == "__main__":