Update app.py

app.py

@@ -5,16 +5,19 @@ from datetime import datetime, timedelta
 from functools import lru_cache
 import gradio as gr
 import openai
-OPENAI_API_KEY
-# -------------------- CONFIGURATION --------------------
-# Use the provided OpenAI API key (lecturer paid for it)
-OPENAI_API_KEY = "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"
-
-# Optional: set OpenWeatherMap API key as environment variable (lecturer will provide)
-OPENWEATHER_API_KEY = os.getenv("OPENWEATHER_API_KEY", "")  # Set this in Hugging Face Secrets
-SERP_API_KEY = os.getenv("SERPAPI_API_KEY", "")            # Optional for real‑time attractions
 
-client = openai.OpenAI(api_key=OPENAI_API_KEY)
+# -------------------- CONFIGURATION --------------------
+# Read API keys from environment variables (set in Hugging Face Secrets)
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+OPENWEATHER_API_KEY = os.getenv("OPENWEATHER_API_KEY", "")
+SERP_API_KEY = os.getenv("SERPAPI_API_KEY", "")
+
+# Warn if OpenAI key is missing (but continue with fallbacks)
+if not OPENAI_API_KEY:
+    print("⚠️ OPENAI_API_KEY not set. AI itinerary generation will fall back to manual mode.")
+    client = None
+else:
+    client = openai.OpenAI(api_key=OPENAI_API_KEY)
 
 # Local attractions database (can be expanded)
 try:
@@ -28,7 +31,7 @@ except FileNotFoundError:
 # -------------------- TOOL IMPLEMENTATIONS --------------------
 def get_weather(city: str) -> dict:
     """
-    Return current weather for any city using lecturer's OpenWeatherMap API.
+    Return current weather for any city using OpenWeatherMap API (if key set).
     Returns a dict with keys: city, temperature, condition, humidity, wind_speed, precipitation.
     On error, returns dict with 'error' key.
     """
@@ -56,7 +59,6 @@ def get_weather(city: str) -> dict:
         if response.status_code != 200:
             return {"error": f"Weather API error: {data.get('message', 'unknown')}"}
 
-        # Extract relevant fields
         return {
             "city": city,
             "temperature": data['main']['temp'],
@@ -123,6 +125,13 @@ def fetch_attractions_via_serp(city: str) -> list:
 
 def fetch_attractions_via_openai(city: str) -> list:
     """Use OpenAI to generate a list of top attractions for any city."""
+    if not client:
+        # If no OpenAI client, return generic list
+        return [
+            {"name": f"City center", "type": "landmark", "entry_fee": "free", "duration_hours": 2},
+            {"name": f"Local museum", "type": "museum", "entry_fee": "varies", "duration_hours": 2},
+            {"name": f"Popular park", "type": "nature", "entry_fee": "free", "duration_hours": 2}
+        ]
     try:
         prompt = f"""
 List the top 8 tourist attractions in {city}. For each, provide:
@@ -146,7 +155,6 @@ Do not include any other text.
         if isinstance(attractions, dict) and "attractions" in attractions:
             attractions = attractions["attractions"]
         elif not isinstance(attractions, list):
-            # Try to extract list
             attractions = [attractions]
         # Normalize to our format
         result = []
@@ -160,7 +168,6 @@ Do not include any other text.
         return result
     except Exception as e:
         print(f"OpenAI attractions error: {e}")
-        # Fallback to a simple generic list
         return [
             {"name": f"City center", "type": "landmark", "entry_fee": "free", "duration_hours": 2},
             {"name": f"Local museum", "type": "museum", "entry_fee": "varies", "duration_hours": 2},
@@ -180,9 +187,10 @@ def get_attractions(city: str) -> dict:
         if serp_attractions:
             return {"city": city, "attractions": serp_attractions, "source": "SerpAPI"}
 
-    # 3. Fallback to OpenAI (always works, unless quota exceeded)
+    # 3. Fallback to OpenAI (or generic if no key)
     openai_attractions = fetch_attractions_via_openai(city)
-    return {"city": city, "attractions": openai_attractions, "source": "OpenAI (AI‑generated)"}
+    source = "OpenAI" if client else "generic fallback"
+    return {"city": city, "attractions": openai_attractions, "source": source}
 
 def get_tripadvisor_link(query):
     """Fetch a TripAdvisor link using SerpAPI (optional)."""
@@ -240,6 +248,10 @@ def generate_itinerary(destination, start_date, num_days, budget_type, departure
 
         # 5. Attempt to call OpenAI for itinerary
         try:
+            if not client:
+                # No OpenAI key – show manual plan
+                raise Exception("OpenAI client not available")
+
             prompt = f"""
 You are an expert travel planner. Create a detailed, day-wise travel itinerary based on the verified information below.
 
@@ -297,26 +309,10 @@ Follow these rules:
                                 break
                 itinerary = "\n".join(enhanced)
 
-        except openai.RateLimitError as e:
-            # Quota exceeded – fallback to manual itinerary
-            itinerary = f"""
-# 🌍 Manual Itinerary for {destination} (OpenAI quota exceeded)
-
-**Weather:** {weather_text}
-**Attractions available:**
-{attractions_list}
-{source_note}
-**Budget:** {budget_text if budget_text else 'Not specified'}
-
-*Since the AI service is temporarily unavailable (quota exceeded), please use the above information to plan your trip manually. You can search for opening hours, entry fees, and book activities online.*
-
-🛏️ [Search Hotels on Booking.com](https://www.booking.com/searchresults.html?ss={destination.replace(' ', '+')})
-✈️ [Search Flights on Skyscanner](https://www.skyscanner.co.in/)
-"""
         except Exception as e:
-            # Other OpenAI errors
+            # Fallback to manual itinerary (no AI)
             itinerary = f"""
-# ⚠️ Itinerary generation failed (OpenAI error: {str(e)})
+# 🌍 Manual Itinerary for {destination}
 
 **Weather:** {weather_text}
 **Attractions available:**
@@ -324,7 +320,7 @@ Follow these rules:
 {source_note}
 **Budget:** {budget_text if budget_text else 'Not specified'}
 
-Please try again later or use the above information to plan manually.
+*Since the AI service is unavailable (API key missing or quota exceeded), please use the above information to plan your trip manually. You can search for opening hours, entry fees, and book activities online.*
 
 🛏️ [Search Hotels on Booking.com](https://www.booking.com/searchresults.html?ss={destination.replace(' ', '+')})
 ✈️ [Search Flights on Skyscanner](https://www.skyscanner.co.in/)
@@ -353,7 +349,7 @@ with gr.Blocks(css=css) as demo:
     gr.Markdown("""
     # 🌍 TouristGuide AI Agent
     Your personal travel planner powered by real-time weather, attractions, and OpenAI.  
-    Works for **any city** – if no local data or SerpAPI, it uses OpenAI to suggest attractions.
+    Works for **any city** – if no local data, it uses SerpAPI or OpenAI to suggest attractions.
     """)
     with gr.Row():
         destination = gr.Textbox(label="Destination (any city)", placeholder="e.g., Paris, Tokyo, New York", scale=2)