Update app.py

app.py

@@ -451,6 +451,7 @@ weather_api_key = os.environ.get("WEATHER_API_KEY")
 
 
 def get_time_tool2(query: str) -> datetime:
+
     try:
         # Step 1: 獲取地點
         location_prompt = f"""
@@ -472,7 +473,8 @@ def get_time_tool2(query: str) -> datetime:
 
         # Step 3: 用 GPT 推算具體時間
         time_query_prompt = f"""
-        Given the current time in {location}, answer the following:
+        Given that today is {now.strftime('%A')}, {now.strftime('%Y-%m-%d')}, 
+        answer the following:
 
         Query: "{query}"
 
@@ -486,19 +488,22 @@ def get_time_tool2(query: str) -> datetime:
         time_str = time_response.choices[0].text.strip()
 
         # Step 4: 解析並格式化時間
-        # 嘗試解析時間
-        if "AM" in time_str or "PM" in time_str:
-            target_time = datetime.strptime(time_str, "%I:%M %p")
-        else:
-            target_time = datetime.strptime(time_str, "%Y-%m-%d %H:%M:%S")
+        try:
+            # 嘗試解析時間
+            if "AM" in time_str or "PM" in time_str:
+                target_time = datetime.strptime(time_str, "%I:%M %p")
+            else:
+                target_time = datetime.strptime(time_str, "%Y-%m-%d %H:%M:%S")
 
-        # 添加時區
-        target_time = target_time.replace(tzinfo=ZoneInfo(tz_str))
+            # 添加時區
+            target_time = target_time.replace(tzinfo=ZoneInfo(tz_str))
 
-        return target_time.strftime("%Y-%m-%d %H:%M:%S")
+            return target_time.strftime("%Y-%m-%d %H:%M:%S")
+        except Exception as e:
+            return f"Error in processing time: {e}"
 
     except Exception as e:
-        return f"Error in processing time: {e}"
+        return f"Error in retrieving location or time information: {e}"
         
 def weather_agent_tool(query: str) -> str:
     """Weather Agent: Return current, hourly, or historical weather info using WeatherAPI."""