add weather forecast

app.py

@@ -5,6 +5,8 @@ import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
 
+import xml.etree.ElementTree as ET
+
 from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity!
@@ -34,6 +36,76 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
+@tool
+def get_city_id(url, city_name):
+    """A tool that get city ID from city name.
+
+    Args:
+        city_name:  The city name that want to get city id written in Japanese. (e.g., '東京', '横浜')
+
+    Returns:
+        city id. If not get, return None.
+    """
+
+    # XMLのURL
+    xml_url = "https://weather.tsukumijima.net/primary_area.xml"
+
+    try:
+        response = requests.get(url)
+        response.raise_for_status()
+        response.encoding = response.apparent_encoding
+        xml_string = response.text
+
+        root = ET.fromstring(xml_string)
+        for city in root.findall('.//city'):
+            if city.get('title') == city_name:
+                return city.get('id')
+        return None
+    except requests.exceptions.RequestException as e:
+        print(f"Error during request: {e}")
+        return None
+    except ET.ParseError as e:
+        print(f"Error parsing XML: {e}")
+        return None
+
+@tool
+def get_weather(city_id):
+    """Get city weather forecast from city id.
+
+    Args:
+        city_id: city id. (e.g., 140010)
+
+    Returns:
+        Weather forecast
+    """
+    try:
+        url = f"https://weather.tsukumijima.net/api/forecast?city={city_id}"
+        response = requests.get(url)
+        response.raise_for_status()
+
+        data_json = response.json()
+
+        # 今日の予報を取得 (forecasts[0])
+        date_str = data_json["forecasts"][0]["date"]
+        date = datetime.strptime(date_str, "%Y-%m-%d").strftime("%Y年%m月%d日")
+        title = data_json["title"]
+        weather = data_json["forecasts"][0]["telop"]
+
+        # 明日の予報を取得 (forecasts[1])
+        weather2 = data_json["forecasts"][1]["telop"]
+        
+        results = f"{date}の{title}は{weather}です。明日は{weather2}でしょう。"
+        return results
+    
+    except requests.exceptions.RequestException as e:
+        return f"天気情報の取得に失敗しました: {e}"
+        
+    except KeyError as e:
+        return f"予期しないデータ形式です: {e}"
+    except TypeError as e:
+        return f"気温データが取得できませんでした: {e}"
+
+
 final_answer = FinalAnswerTool()
 model = HfApiModel(
 max_tokens=2096,
@@ -51,7 +123,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, image_generation_tool, get_current_time_in_timezone], # add your tools here (don't remove final_answer)
+    tools=[final_answer, image_generation_tool, get_current_time_in_timezone, get_weather, get_city_id], # add your tools here (don't remove final_answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,