Update app.py

app.py

@@ -94,12 +94,13 @@ def generate_itinerary(mood, location, budget, days):
     )
 
     # 🧩 Generate itinerary
-    completion = client.chat.completions.create(
+    completion = client.response.create(
         model="openai/gpt-oss-20B",  # Use gemma or gpt-oss if this doesn't work
         messages=[
             {"role": "system", "content": "You are an expert travel planner."},
             {"role": "user", "content": prompt}
         ],
+         input=prompt,
         temperature=0.8,
         max_tokens=2800,  # enough for multi-day itineraries
         top_p=1,
@@ -143,48 +144,24 @@ def get_best_season(location):
     try:
         print(f"🔍 Calling Groq API for location: {location}")
 
-        # Try chat model first
-        try:
-            response = client.chat.completions.create(
-                model="gemma2-9b-it",  # ✅ works with Groq
-                messages=[
-                    {"role": "system", "content": "You are a travel expert."},
-                    {"role": "user", "content": prompt}
-                ],
-                temperature=0.7,
-                max_completion_tokens=120,
-                top_p=1,
-                stream=False
-            )
-
-            if hasattr(response.choices[0].message, "content"):
-                season_text = response.choices[0].message.content
-            else:
-                season_text = response.choices[0].message["content"]
-
-        except Exception as e:
-            # ✅ Fallback for models that don’t support chat.completions
-            print("⚠️ Chat completion failed, trying text generation fallback...")
-            completion = client.completions.create(
-                model="gemma2-9b-it",
-                prompt=prompt,
-                max_tokens=120,
-                temperature=0.7,
-                top_p=1
-            )
-            season_text = completion.choices[0].text
-
-        print("✅ Received response from API.")
-        print("🧠 AI raw response:", season_text)
-
-        if not season_text.strip():
+        # Use the responses.create() method with gpt-oss-20b
+        response = client.responses.create(
+            model="openai/gpt-oss-20b",
+            input=prompt,
+            temperature=0.7,
+            max_output_tokens=120
+        )
+
+        # Access the generated text
+        season_text = response.output_text.strip() if hasattr(response, "output_text") else str(response)
+
+        if not season_text:
             return "🌤️ Season info not available right now."
 
-        return f"🌤️ {season_text.strip()}"
+        return f"🌤️ {season_text}"
 
     except Exception as e:
-        print("⚠️ Full error while getting season:\n")
-        traceback.print_exc()
+        print("⚠️ Error while getting season:\n", e)
         return "🌤️ Unable to determine the best season right now."
 
 
@@ -192,6 +169,7 @@ def get_best_season(location):
 
 
 
+
 def generate_pdf(itinerary_text):
     if not itinerary_text.strip():
         return None, gr.update(visible=False)