Update app.py

app.py

@@ -16,7 +16,25 @@ def days_before_season(local_time:str, season:str)-> str: #it's import to specif
         local_time: local time
         season: a season of the year
     """
-    days = season_start - local_time
+    # days = season_start - local_time
+    # Current time
+    current_time = datetime.strptime(local_time, "%Y-%m-%d %H:%M:%S")
+
+    if season == "spring":
+        spring_start = datetime(2025, 03, 01)
+        # Calculate the difference
+        days = (spring_start - current_time).days
+    elif season == "summer":
+        summer_start = datetime(2025, 06, 01)
+        # Calculate the difference
+        days = (summer_start - current_time).days
+    elif season == "autumn":
+        autumn_start = datetime(2025, 09, 01)
+        # Calculate the difference
+        days = (autumn_start - current_time).days
+    elif season == "winter":
+        return "I cannot calculate days until the winter because it's currently a winter"
+
     return f"Until {season} are {days} left."
 
 @tool