Update app.py

app.py

@@ -126,19 +126,16 @@ def generate_itinerary(mood, location, budget, days):
         return "🌤️ Unable to generate itinerary at this time.", None, None
 
 def get_best_season(location):
-    """AI-powered function to recommend the best season or general weather for any location."""
-    if not location.strip() or location.lower() == "any destination":
+    location_clean = location.strip()
+    if not location_clean or location_clean.lower() == "any destination":
         return "🌤️ Best season info is available for specific locations only."
-
+    
     prompt = (
-        f"Suggest the best season or time of year to visit {location} based on weather and tourism. "
-        f"Keep it short and under 25 words."
+        f"You are a travel expert. Suggest the best season or months to visit {location_clean}, "
+        f"considering weather, tourist crowds, and local events. Keep it 1-2 sentences."
     )
 
     try:
-        print(f"🔍 Calling Groq API for location: {location}")
-
-        # Use chat.completions.create() method with gpt-oss-20b
         response = client.chat.completions.create(
             messages=[
                 {"role": "system", "content": "You are a travel expert."},
@@ -150,17 +147,19 @@ def get_best_season(location):
             top_p=1
         )
 
-        season_text = response.choices[0].message.content.strip()
+        season_text = (
+            getattr(response.choices[0].message, "content", "")
+            or getattr(response.choices[0], "text", "")
+        ).strip()
 
         if not season_text:
-            return "🌤️ Season info not available right now."
-
+            # fallback if model returns nothing
+            return SEASON_FALLBACK.get(location_clean, "🌤️ Season info not available right now.")
         return f"🌤️ {season_text}"
 
     except Exception as e:
         print("⚠️ Error while getting season:\n", e)
-        return "🌤️ Unable to determine the best season right now."
-
+        return SEASON_FALLBACK.get(location_clean, "🌤️ Unable to determine the best season right now.")
 
 def generate_pdf(itinerary_text):
     if not itinerary_text.strip():