Update app.py

app.py

@@ -1,7 +1,4 @@
 import os
-import time
-import re
-import requests
 import gradio as gr
 from groq import Groq
 from reportlab.pdfgen import canvas
@@ -35,49 +32,6 @@ MOOD_IMAGES = {
     "Any": "images/default.png"
 }
 
-# --- Map related functions ---
-def extract_places(itinerary_text):
-    pattern = r"\b([A-Z][a-z]+(?:\s[A-Z][a-z]+)*)\b"
-    matches = re.findall(pattern, itinerary_text)
-    blacklist = {"Day", "Morning", "Evening", "Afternoon", "Breakfast", "Lunch", "Dinner"}
-    places = [m for m in matches if m not in blacklist and len(m) > 2]
-    return list(dict.fromkeys(places))  # unique order preserved
-
-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 and response.json():
-            data = response.json()[0]
-            return float(data["lat"]), float(data["lon"])
-    except:
-        pass
-    return None, None
-
-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>');\n"
-
-    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" />
-    <script src="https://unpkg.com/leaflet@1.9.3/dist/leaflet.js"></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 generate_itinerary(mood, location, budget, days):
     mood_text = "any mood" if mood == "Any" else mood
     budget_text = "any budget" if budget == "Any" else budget
@@ -103,76 +57,100 @@ def generate_itinerary(mood, location, budget, days):
     
     itinerary_text = completion.choices[0].message.content
     image_path = MOOD_IMAGES.get(mood, MOOD_IMAGES["Any"])
-
-    # --- Map generation ---
-    places = extract_places(itinerary_text)
-    locations = []
-    for p in places:
-        lat, lng = geocode_location(p)
-        if lat and lng:
-            locations.append({"name": p, "lat": lat, "lng": lng})
-        time.sleep(1)  # respect Nominatim usage policy
-    if not locations:
-        locations = [{"name": "World", "lat": 20, "lng": 0}]
-    map_html = create_leaflet_map(locations)
-
-    return itinerary_text, image_path, gr.update(value=map_html, visible=True)
+    return itinerary_text, image_path, gr.update(visible=True)
 
 def generate_pdf(itinerary_text):
     if not itinerary_text.strip():
         return None
+
     temp_pdf = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
     c = canvas.Canvas(temp_pdf.name, pagesize=letter)
     width, height = letter
+
     margin = 40
     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
+
+    lines = []
     for paragraph in itinerary_text.split('\n'):
-        for line in wrap(paragraph, width=max_chars_per_line):
-            if y < margin + line_height:
-                c.showPage()
-                c.setFont("Courier", font_size)
-                y = height - margin
-            c.drawString(x, y, line)
-            y -= line_height
+        wrapped = wrap(paragraph, width=max_chars_per_line)
+        lines.extend(wrapped if wrapped else [''])
+
+    for line in lines:
+        if y < margin + line_height:
+            c.showPage()
+            c.setFont("Courier", font_size)
+            y = height - margin
+        c.drawString(x, y, line)
+        y -= line_height
+
     c.save()
     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 {
+        0% { color: #008080; }
+        25% { color: #20b2aa; }
+        50% { color: #40e0d0; }
+        75% { color: #008080; }
+        100% { color: #20b2aa; }
+    }
+    .color-cycle {
+        animation: colorCycle 4s infinite;
+    }
+    .tagline {
+        font-style: italic;
+        color: #007777;
+        font-size: 18px;
+        margin-top: 4px;
+    }
+    </style>
     <div style='text-align: center;'>
-        <h1 style="color: teal; font-size: 38px; margin-bottom: 0;">FEEL AWAY APP</h1>
-        <p style="font-style: italic; color: #007777; font-size: 18px; margin-top: 4px;">
-        "Your Mood, Your Journey – Personalized Itineraries, Instantly"</p>
+        <h1 class="color-cycle" style="font-size: 38px; margin-bottom: 0;">FEEL AWAY APP</h1>
+        <p class="tagline">"Your Mood, Your Journey – Personalized Itineraries, Instantly"</p>
     </div>
     """
-    gr.HTML(header_html)
+    with gr.Row():
+        gr.HTML(header_html)
 
+    # Inputs
     with gr.Row():
-        mood = gr.Dropdown(MOODS, label="Mood", value="Any")
-        location = gr.Textbox(label="Destination", placeholder="e.g., India, Japan, France")
+        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?")
+    
     with gr.Row():
-        budget = gr.Dropdown(["Low", "Medium", "High", "Any"], label="Budget", value="Any")
-        days = gr.Slider(1, 15, value=5, step=1, label="Days")
-
+        budget = gr.Dropdown(["Low", "Medium", "High", "Any"], label="Budget", value="Any", info="Choose your budget level")
+        days = gr.Slider(1, 15, value=5, step=1, label="Days", info="Number of days for the itinerary")
+    
     generate_btn = gr.Button("✨ Generate Itinerary", variant="primary")
-    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)
 
-    download_btn = gr.Button("📥 Download Itinerary as PDF", variant="secondary")
-    pdf_file = gr.File(label="Download PDF", file_types=[".pdf"])
+    with gr.Column():
+        itinerary_output = gr.Markdown(label="Generated Itinerary")
+        image_output = gr.Image(label="Travel Mood Image", type="filepath", visible=False)
+
+    with gr.Column():
+        download_btn = gr.Button("📥 Download Itinerary as PDF", variant="secondary")
+        pdf_file = gr.File(label="Download PDF", file_types=[".pdf"])
 
     generate_btn.click(
         generate_itinerary,
         inputs=[mood, location, budget, days],
-        outputs=[itinerary_output, image_output, map_output]
+        outputs=[itinerary_output, image_output, image_output]
     )
 
-    download_btn.click(generate_pdf, inputs=[itinerary_output], outputs=pdf_file)
+    download_btn.click(
+        generate_pdf,
+        inputs=[itinerary_output],
+        outputs=pdf_file
+    )
 
-demo.launch()
+demo.launch()