Update app.py

app.py

@@ -1,27 +1,19 @@
 import gradio as gr
 import requests
 import os
-from geopy.geocoders import Nominatim
 from smolagents import tool, CodeAgent, HfApiModel, GoogleSearchTool, VisitWebpageTool
 
 # 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
 def get_weather(lat: float, lon: float) -> dict | None:
-    """
-    Call API to retrieve weather information for the given location.
-    Args:
-      lat: latitude of the location
-      lon: longitude of the location
-    """
     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()
         if response.status_code == 200:
+            data = response.json()
             return {
                 "condition": data["weather"][0]["description"],
                 "temperature": data["main"]["temp"],
@@ -30,51 +22,50 @@ def get_weather(lat: float, lon: float) -> dict | None:
             }
         return None
     except Exception as e:
-        print(f"Weather API Error: {e}")
+        print(f"Lỗi API thời tiết: {e}")
         return None
 
-
-# Khởi tạo agent
 weather_agent = CodeAgent(
     model=HfApiModel("deepseek-ai/DeepSeek-R1", max_tokens=8096, provider="together"),
-    tools=[
-        GoogleSearchTool(provider="serper"),
-        VisitWebpageTool(),
-        get_weather
-    ],
-    name="weather_agent",
-    description="Fetch the weather condition of the given location."
+    tools=[GoogleSearchTool(provider="serper"), get_weather],
+    name="Weather Expert",
+    description="Xử lý yêu cầu thời tiết theo quy trình"
 )
 
-# Hàm xử lý chat
 def respond(message, history):
     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"""
+        system_prompt = """BẠN LÀ CHUYÊN GIA THỜI TIẾT:
+        1. Phân tích địa điểm từ input 
+        2. Google search "tọa độ [địa điểm]"
+        3. Trích xuất LAT/LON dạng số thực
+        4. Gọi get_weather(LAT, LON)
+        5. Trả lời markdown tiếng Việt
+        
+        LƯU Ý:
+        - Luôn kiểm tra số liệu
+        - Nếu không tìm được tọa độ → hỏi lại
+        - Đơn vị: °C, %, m/s"""
         
         response = weather_agent.run(
-            f"{system_prompt}\n\nYêu cầu người dùng: {message}"
+            f"[Hệ thống] {system_prompt}\n[Người dùng] {message}"
         )
         
-        return response if response else "Không thể lấy thông tin thời tiết"
+        return response if response else "Không thể xử lý yêu cầu"
     
     except Exception as e:
-        print(f"Error: {e}")
-        return "Đã xảy ra lỗi khi xử lý yêu cầu"
+        print(f"[DEBUG] {str(e)}")
+        return "Xin lỗi, tôi gặp sự cố xử lý yêu cầu này"
 
-# Tạo giao diện Gradio
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🗺️ Weather Chat Agent")
-    gr.Markdown("Nhập địa điểm để xem thông tin thời tiết")
-    
+    gr.Markdown("# 🌦️ Weather Assistant")
+    gr.Markdown("Nhập địa điểm bất kỳ để tra cứu thời tiết")
     chatbot = gr.ChatInterface(
         respond,
-        examples=["Hà Nội", "New York", "Tokyo", "Paris"],
-        title="Weather Chatbot",
+        title="Tra cứu Thời tiết",
+        submit_btn="Gửi",
+        retry_btn="Thử lại",
+        undo_btn="Hủy",
+        clear_btn="Xóa"
     )
 
 if __name__ == "__main__":