Update app.py

app.py

@@ -1,3 +1,5 @@
+import spacy.cli
+spacy.cli.download("en_core_web_sm")
 import os
 import gradio as gr
 from groq import Groq
@@ -5,14 +7,16 @@ from reportlab.pdfgen import canvas
 from reportlab.lib.pagesizes import letter
 import tempfile
 import re
-from textwrap import wrap
 from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
-from reportlab.lib.pagesizes import letter
 from reportlab.lib.styles import getSampleStyleSheet
+from reportlab.lib.pagesizes import letter
+import spacy
+from geopy.geocoders import Nominatim
+import folium
 
-# Load API key securely from environment
-GROQ_API_KEY = os.getenv("GROQ_API_KEY")
-client = Groq(api_key=GROQ_API_KEY)
+# Load API key 
+GR_API_KEY = os.getenv("GR_API_KEY")
+client = Groq(api_key=GR_API_KEY)
 
 MOODS = [
     "Adventure",
@@ -36,6 +40,41 @@ MOOD_IMAGES = {
     "Any": "images/default.png"
 }
 
+nlp = spacy.load("en_core_web_sm")
+
+def clean_place_name(name):
+    return re.sub(r'^(the|a|an)\s+', '', name, flags=re.IGNORECASE).strip()
+
+def extract_places_and_map(itinerary_text, city_context=None):
+    pattern = r"(visit|explore|trip to|at|go to|see)\s+([A-Z][A-Za-zÀ-ÖØ-öø-ÿ'’\- ]+)"
+    matches = re.findall(pattern, itinerary_text, re.IGNORECASE)
+    rule_places = [m[1].strip() for m in matches]
+
+    doc = nlp(itinerary_text)
+    ner_places = [ent.text for ent in doc.ents if ent.label_ in ["GPE", "LOC", "FAC"]]
+
+    places = list(set(rule_places + ner_places))
+
+    geolocator = Nominatim(user_agent="feelaway_app")
+    locations = []
+    for place in places:
+        try:
+            query = f"{place}, {city_context}" if city_context else place
+            loc = geolocator.geocode(query, language='en', timeout=10)
+            if loc:
+                locations.append((place, loc.latitude, loc.longitude))
+        except Exception:
+            continue
+
+    if not locations:
+        return "<p>No locations found for mapping.</p>"
+
+    m = folium.Map(location=[locations[0][1], locations[0][2]], zoom_start=10)
+    for name, lat, lon in locations:
+        folium.Marker([lat, lon], popup=name).add_to(m)
+
+    return m._repr_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
@@ -45,7 +84,6 @@ def generate_itinerary(mood, location, budget, days):
         f"Create a {days}-day travel itinerary to {location_text}. "
         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."
-        f"In the last line show the Approximate budget for one person in Dollars."
     )
     
     completion = client.chat.completions.create(
@@ -61,12 +99,15 @@ def generate_itinerary(mood, location, budget, days):
     )
     
     itinerary_text = completion.choices[0].message.content
-    image_path = MOOD_IMAGES.get(mood, MOOD_IMAGES["Any"])
-    return itinerary_text, image_path, gr.update(visible=True)
+    image_path = os.path.abspath(MOOD_IMAGES.get(mood, MOOD_IMAGES["Any"]))
+
+    map_html = extract_places_and_map(itinerary_text, city_context=location_text if location_text.lower() != "any destination" else None)
+
+    return itinerary_text, image_path, map_html
 
 def generate_pdf(itinerary_text):
     if not itinerary_text.strip():
-        return None
+        return None, gr.update(visible=False)
 
     temp_dir = tempfile.gettempdir()
     temp_path = os.path.join(temp_dir, "feelaway_itinerary.pdf")
@@ -80,9 +121,9 @@ def generate_pdf(itinerary_text):
 
     styles = getSampleStyleSheet()
     style = styles["Normal"]
-    style.fontName = "Courier"
-    style.fontSize = 10
-    style.leading = 14
+    style.fontName = "Times-Roman"
+    style.fontSize = 14
+    style.leading = 18
 
     html_text = re.sub(r'\*\*(.+?)\*\*', r'<b>\1</b>', itinerary_text)
     paragraphs = html_text.split('\n')
@@ -95,12 +136,10 @@ def generate_pdf(itinerary_text):
 
     doc.build(story)
 
-    return temp_path
-
-
-
+    return temp_path, gr.update(visible=True)
+# ---------------------- UI ----------------------
 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 {
@@ -111,52 +150,118 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")
         100% { color: #20b2aa; }
     }
     .color-cycle {
-        animation: colorCycle 4s infinite;
+        animation: colorCycle 3s infinite;
+        font-weight: 800;
+        letter-spacing: 1.5px;
+        text-transform: uppercase;
+        text-shadow: 1px 1px 3px #a0d1d1;
     }
     .tagline {
         font-style: italic;
-        color: #007777;
-        font-size: 18px;
+        color: #007777 !important;
+        font-size: 20px;
         margin-top: 4px;
+        height: 30px;
+        white-space: nowrap;
+        overflow: hidden;
+        font-weight: 600;
+        animation: pulseOpacity 2s infinite;
+    }
+    .equal-row {
+        display: flex;
+        gap: 16px;
+    }
+    .equal-row > div {
+        flex: 1;
+    }
+    #map-container iframe,
+    #image-container img {
+        width: 100%;
+        height: 500px;
+        border-radius: 8px;
+        object-fit: cover;
+        display: block;
     }
     </style>
-    <div style='text-align: center;'>
-        <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 style='text-align: center; margin-top: 30px; margin-bottom: 30px;'>
+        <h1 class="color-cycle" style="font-size: 42px; margin: 0;">FEEL AWAY APP</h1>
+        <p class="tagline" id="typing-text-main"></p>
     </div>
+    <script>
+    const tagline = "Your Mood, Your Journey – Personalized Itineraries, Instantly!";
+    function typeWriter(elId) {
+        let i = 0;
+        let forward = true;
+        const el = document.getElementById(elId);
+        function typing() {
+            if (forward) {
+                if (i < tagline.length) {
+                    el.innerHTML += tagline.charAt(i);
+                    i++;
+                } else {
+                    forward = false;
+                    setTimeout(typing, 2000);
+                    return;
+                }
+            } else {
+                if (i > 0) {
+                    el.innerHTML = tagline.substring(0, i-1);
+                    i--;
+                } else {
+                    forward = true;
+                }
+            }
+            setTimeout(typing, 60);
+        }
+        typing();
+    }
+    window.onload = () => {
+        typeWriter("typing-text-main");
+    }
+    </script>
     """
-    with gr.Row():
+
+    with gr.Column(elem_id="main-content"):
         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?")
-    
-    with gr.Row():
-        budget = gr.Dropdown(["Low", "Medium", "High", "Any"], label="Budget", value="Any", info="Choose your budget level")
-        days = gr.Slider(1, 10, value=5, step=1, label="Days", info="Number of days for the itinerary")
-    
-    generate_btn = gr.Button("✨ Generate Itinerary", variant="primary")
+        with gr.Row():
+            mood = gr.Dropdown(MOODS, label="Mood", value="Any")
+            location = gr.Textbox(label="Destination", placeholder="e.g., India, Japan, France (leave blank for Any)")
+        
+        with gr.Row():
+            budget = gr.Dropdown(["Low", "Medium", "High", "Any"], label="Budget", value="Any")
+            days = gr.Slider(1, 10, value=5, step=1, label="Days")
+        
+        generate_btn = gr.Button("Generate Itinerary", variant="primary")
+
+        with gr.Column():
+            itinerary_output = gr.Markdown(label="Generated Itinerary")
 
-    with gr.Column():
-        itinerary_output = gr.Markdown(label="Generated Itinerary")
-        image_output = gr.Image(label="Travel Mood Image", type="filepath", visible=False)
+            with gr.Row(elem_classes="equal-row"):
+                with gr.Column(scale=1):
+                    gr.HTML("<div id='image-container'>")
+                    image_output = gr.Image(type="filepath", show_label=False)
+                    gr.HTML("</div>")
+                with gr.Column(scale=1):
+                    gr.HTML("<div id='map-container'>")
+                    map_output = gr.HTML(show_label=False)
+                    gr.HTML("</div>")
 
-    with gr.Column():
-        download_btn = gr.Button("📥 Download Itinerary as PDF", variant="secondary")
-        pdf_file = gr.File(label="Download PDF", file_types=[".pdf"])
+        with gr.Column():
+            download_btn = gr.Button("Download Itinerary as PDF", variant="secondary")
+            pdf_file = gr.File(file_types=[".pdf"], visible=False)
 
     generate_btn.click(
         generate_itinerary,
         inputs=[mood, location, budget, days],
-        outputs=[itinerary_output, image_output, image_output]
+        outputs=[itinerary_output, image_output, map_output],
+        show_progress=True
     )
 
     download_btn.click(
         generate_pdf,
         inputs=[itinerary_output],
-        outputs=pdf_file
+        outputs=[pdf_file, pdf_file]
     )
 
 demo.launch()