Spaces:
Sleeping
Sleeping
| import spacy.cli | |
| spacy.cli.download("en_core_web_sm") | |
| import os | |
| import gradio as gr | |
| from groq import Groq | |
| from reportlab.pdfgen import canvas | |
| from reportlab.lib.pagesizes import letter | |
| import tempfile | |
| import re | |
| from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer | |
| from reportlab.lib.styles import getSampleStyleSheet | |
| from reportlab.lib.pagesizes import letter | |
| import spacy | |
| from geopy.geocoders import Nominatim | |
| import folium | |
| # Load API key | |
| GR_API_KEY = os.getenv("GR_API_KEY") | |
| client = Groq(api_key=GR_API_KEY) | |
| MOODS = [ | |
| "Adventure", | |
| "Relaxing", | |
| "Romantic", | |
| "Cultural & Heritage", | |
| "Foodie & Culinary", | |
| "Shopping & Urban", | |
| "Beach & Island", | |
| "Any" | |
| ] | |
| TRAVELLER_TYPES = [ | |
| "Solo", | |
| "Couple", | |
| "Family", | |
| "Friends", | |
| "Business", | |
| "Any" | |
| ] | |
| MOOD_IMAGES = { | |
| "Adventure": "images/adventure.png", | |
| "Relaxing": "images/relaxing.png", | |
| "Romantic": "images/romantic.png", | |
| "Cultural & Heritage": "images/cultural.png", | |
| "Foodie & Culinary": "images/foodie.png", | |
| "Shopping & Urban": "images/shopping.png", | |
| "Beach & Island": "images/beach.png", | |
| "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, traveller_type): | |
| mood_text = "any mood" if mood == "Any" else mood | |
| budget_text = "any budget" if budget == "Any" else budget | |
| traveller_text = "any traveller" if traveller_type == "Any" else traveller_type | |
| location_text = location if location.strip() != "" else "any destination" | |
| prompt = ( | |
| f"Create a {days}-day travel itinerary for a {traveller_text} visiting {location_text}. " | |
| f"Include the best season to visit the location. " | |
| f"The traveler's mood is {mood_text} and the budget is {budget_text}. " | |
| f"Include daily activities, key landmarks, meals, and cultural experiences. " | |
| f"Use Markdown formatting with **bold headings** for days and time slots." | |
| ) | |
| completion = client.chat.completions.create( | |
| model="llama-3.1-8b-instant", | |
| 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 = 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, gr.update(visible=False) | |
| temp_dir = tempfile.gettempdir() | |
| temp_path = os.path.join(temp_dir, "feelaway_itinerary.pdf") | |
| if os.path.exists(temp_path): | |
| os.remove(temp_path) | |
| doc = SimpleDocTemplate(temp_path, pagesize=letter, | |
| rightMargin=40, leftMargin=40, | |
| topMargin=40, bottomMargin=40) | |
| styles = getSampleStyleSheet() | |
| style = styles["Normal"] | |
| 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') | |
| story = [] | |
| for para in paragraphs: | |
| p = Paragraph(para if para.strip() else ' ', style) | |
| story.append(p) | |
| story.append(Spacer(1, 6)) | |
| doc.build(story) | |
| return temp_path, gr.update(visible=True) | |
| # ---------------------- UI ---------------------- | |
| with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")) as demo: | |
| header_html = """ | |
| <style> | |
| @keyframes colorCycle { | |
| 0% { color: #008080; } | |
| 25% { color: #20b2aa; } | |
| 50% { color: #40e0d0; } | |
| 75% { color: #008080; } | |
| 100% { color: #20b2aa; } | |
| } | |
| .color-cycle { | |
| 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 !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; 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.Column(elem_id="main-content"): | |
| gr.HTML(header_html) | |
| 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(): | |
| traveller_type = gr.Dropdown(TRAVELLER_TYPES, label="Traveller Type", value="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.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(file_types=[".pdf"], visible=False) | |
| generate_btn.click( | |
| generate_itinerary, | |
| inputs=[mood, location, budget, days, traveller_type], | |
| outputs=[itinerary_output, image_output, map_output], | |
| show_progress=True | |
| ) | |
| download_btn.click( | |
| generate_pdf, | |
| inputs=[itinerary_output], | |
| outputs=[pdf_file, pdf_file] | |
| ) | |
| demo.launch() | |