Update app.py

app.py

@@ -1,8 +1,6 @@
 import os
-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
@@ -10,6 +8,7 @@ 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",
@@ -33,95 +32,6 @@ MOOD_IMAGES = {
     "Any": "images/default.png"
 }
 
-def create_leaflet_map(locations):
-    markers_js = ""
-    for loc in locations:
-        markers_js += f"""
-        L.marker([{loc['lat']}, {loc['lng']}]).addTo(map)
-          .bindPopup("<b>{loc['name']}</b>");
-        """
-
-    center_lat = locations[0]['lat'] if locations else 20
-    center_lng = locations[0]['lng'] if locations else 0
-
-    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"
-      crossorigin=""
-    />
-    <script
-      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>
-    """
-    return html
-
-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)"}
-    try:
-        response = requests.get(url, params=params, headers=headers, timeout=5)
-        if response.status_code == 200:
-            data = response.json()
-            if data:
-                lat = float(data[0]["lat"])
-                lon = float(data[0]["lon"])
-                return lat, lon
-    except Exception:
-        pass
-    return None, None
-
-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)
-    blacklist = {"The", "And", "For", "With", "From", "Your", "Day", "Enjoy", "Start", "Begin"}
-    places = []
-    for match in matches:
-        if match not in blacklist and len(match) > 2:
-            places.append(match)
-    # Remove duplicates preserving order
-    seen = set()
-    unique_places = []
-    for p in places:
-        if p not in seen:
-            seen.add(p)
-            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
@@ -132,28 +42,22 @@ 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."
     )
-
-    itinerary_text = call_groq_model(prompt)
+    
+    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
     image_path = MOOD_IMAGES.get(mood, MOOD_IMAGES["Any"])
-
-    # 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)
-
-    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)
+    return itinerary_text, image_path, gr.update(visible=True)
 
 def generate_pdf(itinerary_text):
     if not itinerary_text.strip():
@@ -167,7 +71,9 @@ 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
 
@@ -188,6 +94,7 @@ def generate_pdf(itinerary_text):
     return temp_pdf.name
 
 with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")) as demo:
+    # Header with gr.HTML for styles and animation
     header_html = """
     <style>
     @keyframes colorCycle {
@@ -205,7 +112,6 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")
         color: #007777;
         font-size: 18px;
         margin-top: 4px;
-        font-weight:bold;
     }
     </style>
     <div style='text-align: center;'>
@@ -216,6 +122,7 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")
     with gr.Row():
         gr.HTML(header_html)
 
+    # Inputs
     with gr.Row():
         mood = gr.Dropdown(MOODS, label="Mood", value="Any", info="Select the mood of your trip")
         location = gr.Textbox(label="Destination", placeholder="e.g., India, Japan, France (leave blank for Any)", info="Where do you want to go?")
@@ -229,7 +136,6 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")
     with gr.Column():
         itinerary_output = gr.Markdown(label="Generated Itinerary")
         image_output = gr.Image(label="Travel Mood Image", type="filepath", visible=False)
-        map_output = gr.HTML(visible=False)
 
     with gr.Column():
         download_btn = gr.Button("📥 Download Itinerary as PDF", variant="secondary")
@@ -238,7 +144,7 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")
     generate_btn.click(
         generate_itinerary,
         inputs=[mood, location, budget, days],
-        outputs=[itinerary_output, image_output, image_output, map_output, map_output]
+        outputs=[itinerary_output, image_output, image_output]
     )
 
     download_btn.click(
@@ -249,3 +155,4 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")
 
 demo.launch()
 
+