Update app.py

app.py

@@ -110,6 +110,42 @@ def generate_itinerary(mood, location, budget, days):
 
     return itinerary_text, image_path, map_html
 
+def get_best_season(location):
+    """AI-powered function to recommend the best season to visit any city, state, or country."""
+    if not location.strip() or location.lower() == "any destination":
+        return "🌤️ Best season info is available for specific locations only."
+
+    prompt = (
+        f"Based on weather, tourism, and festivals, suggest the best season or months to visit {location}. "
+        f"Explain briefly in 25 words or less."
+    )
+
+    try:
+        response = client.chat.completions.create(
+            model="openai/gpt-oss-20b",  # You can replace this with any working model, e.g. "llama-3.1-8b"
+            messages=[
+                {"role": "system", "content": "You are an expert travel advisor."},
+                {"role": "user", "content": prompt}
+            ],
+            temperature=0.7,
+            max_tokens=120,
+            top_p=1,
+            stream=False
+        )
+
+        # Safely extract the content depending on SDK format
+        if hasattr(response.choices[0].message, "content"):
+            season_text = response.choices[0].message.content
+        else:
+            season_text = response.choices[0].message["content"]
+
+        return f"🌤️ {season_text.strip()}"
+
+    except Exception as e:
+        print("⚠️ Error getting season:", e)
+        return "🌤️ Unable to determine the best season right now."
+
+
 def generate_pdf(itinerary_text):
     if not itinerary_text.strip():
         return None, gr.update(visible=False)