Update app.py

app.py

@@ -3,7 +3,6 @@ import time
 import requests
 import re
 import gradio as gr
-from groq import Groq
 from reportlab.pdfgen import canvas
 from reportlab.lib.pagesizes import letter
 import tempfile
@@ -11,7 +10,6 @@ from textwrap import wrap
 
 # Load API key securely from environment
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
-client = Groq(api_key=GROQ_API_KEY)
 
 MOODS = [
     "Adventure",
@@ -48,7 +46,6 @@ def create_leaflet_map(locations):
 
     html = f"""
     <div id="map" style="width: 100%; height: 400px; margin-top: 20px;"></div>
-
     <link
       rel="stylesheet"
       href="https://unpkg.com/leaflet@1.9.3/dist/leaflet.css"
@@ -58,14 +55,11 @@ def create_leaflet_map(locations):
       src="https://unpkg.com/leaflet@1.9.3/dist/leaflet.js"
       crossorigin=""
     ></script>
-
     <script>
       var map = L.map('map').setView([{center_lat}, {center_lng}], 6);
-
       L.tileLayer('https://{{s}}.tile.openstreetmap.org/{{z}}/{{x}}/{{y}}.png', {{
         attribution: '© OpenStreetMap contributors',
       }}).addTo(map);
-
       {markers_js}
     </script>
     """
@@ -73,14 +67,8 @@ def create_leaflet_map(locations):
 
 def geocode_location(place_name):
     url = "https://nominatim.openstreetmap.org/search"
-    params = {
-        "q": place_name,
-        "format": "json",
-        "limit": 1
-    }
-    headers = {
-        "User-Agent": "FeelAwayApp/1.0 (mathushawlika1221@gmail.com)"
-    }
+    params = {"q": place_name, "format": "json", "limit": 1}
+    headers = {"User-Agent": "FeelAwayApp/1.0 (mathushawlika1221@gmail.com)"}
     try:
         response = requests.get(url, params=params, headers=headers, timeout=5)
         if response.status_code == 200:
@@ -97,7 +85,6 @@ def extract_places(itinerary_text):
     # Simple heuristic: extract capitalized phrases (names)
     pattern = r"\b([A-Z][a-z]+(?:\s[A-Z][a-z]+)*)\b"
     matches = re.findall(pattern, itinerary_text)
-    # Remove common English words that aren't places (can be improved)
     blacklist = {"The", "And", "For", "With", "From", "Your", "Day", "Enjoy", "Start", "Begin"}
     places = []
     for match in matches:
@@ -112,6 +99,29 @@ def extract_places(itinerary_text):
             unique_places.append(p)
     return unique_places
 
+def call_groq_model(prompt):
+    if not GROQ_API_KEY:
+        return "**Error:** Missing GROQ_API_KEY in environment settings."
+
+    url = "https://api.groq.com/openai/v1/chat/completions"
+    headers = {"Authorization": f"Bearer {GROQ_API_KEY}", "Content-Type": "application/json"}
+    payload = {
+        "model": "gemma2-9b-it",
+        "messages": [
+            {"role": "system", "content": "You are an expert travel planner."},
+            {"role": "user", "content": prompt}
+        ],
+        "temperature": 1,
+        "max_tokens": 1024
+    }
+    try:
+        response = requests.post(url, headers=headers, json=payload, timeout=30)
+        response.raise_for_status()
+        data = response.json()
+        return data["choices"][0]["message"]["content"]
+    except Exception as e:
+        return f"**Error calling model:** {e}"
+
 def generate_itinerary(mood, location, budget, days):
     mood_text = "any mood" if mood == "Any" else mood
     budget_text = "any budget" if budget == "Any" else budget
@@ -122,37 +132,27 @@ def generate_itinerary(mood, location, budget, days):
         f"The mood of the traveler is {mood_text} and the budget is {budget_text}. "
         f"Use Markdown formatting with **bold headings** for days and time slots."
     )
-    
-    completion = client.chat.completions.create(
-        model="gemma2-9b-it",
-        messages=[
-            {"role": "system", "content": "You are an expert travel planner."},
-            {"role": "user", "content": prompt}
-        ],
-        temperature=1,
-        max_completion_tokens=1024,
-        top_p=1,
-        stream=False
-    )
-    
-    itinerary_text = completion.choices[0].message.content
+
+    itinerary_text = call_groq_model(prompt)
     image_path = MOOD_IMAGES.get(mood, MOOD_IMAGES["Any"])
 
-    # Extract place names from itinerary text
-    places = extract_places(itinerary_text)
+    # If API failed, don't proceed to map
+    if itinerary_text.startswith("**Error"):
+        return itinerary_text, None, gr.update(visible=False), "", gr.update(visible=False)
 
+    # Extract place names
+    places = extract_places(itinerary_text)
     locations = []
     for place in places:
         lat, lng = geocode_location(place)
         if lat and lng:
             locations.append({"name": place, "lat": lat, "lng": lng})
-        time.sleep(1)  # Respect Nominatim usage policy
+        time.sleep(1)
 
     if not locations:
         locations = [{"name": "World", "lat": 20, "lng": 0}]
 
     map_html = create_leaflet_map(locations)
-
     return itinerary_text, image_path, gr.update(visible=True), map_html, gr.update(visible=True)
 
 def generate_pdf(itinerary_text):
@@ -167,9 +167,7 @@ def generate_pdf(itinerary_text):
     line_height = 12
     font_size = 10
     c.setFont("Courier", font_size)
-
     max_chars_per_line = int((width - 2 * margin) / (font_size * 0.6))
-
     x = margin
     y = height - margin