Update app.py

app.py

@@ -9,7 +9,6 @@ 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
@@ -83,18 +82,23 @@ def extract_places_and_map(itinerary_text, city_context=None):
 
     return m._repr_html_()
 
-def generate_itinerary(mood, location, budget, days, traveller_type, current_location):
+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"
-    from_text = current_location.strip() if current_location.strip() != "" else "your city"
 
+    # Prompt tailored to traveller type
     prompt = (
-        f"Create a {days}-day travel itinerary from {from_text} to {location_text}. "
-        f"The traveler’s mood is {mood_text}, traveller type is {traveller_text}, "
+        f"Create a {days}-day travel itinerary for {location_text}. "
+        f"The traveler's mood is {mood_text}, traveller type is {traveller_text}, "
         f"and the budget is {budget_text}. "
-        f"Include daily activities, key landmarks, and meal or cultural suggestions. "
+        f"Analyze the traveller type and tailor the itinerary accordingly:\n"
+        f"- Family: include kid-friendly attractions and activities.\n"
+        f"- Friends: include adventurous, social, and fun group activities.\n"
+        f"- Couple: include romantic spots, dining, and relaxation.\n"
+        f"- Solo: include safe, self-exploration options and easy transport.\n"
+        f"- Business: include short stays, meetings, and convenient travel options.\n"
         f"Use Markdown formatting with **bold headings** for days and time slots."
     )
 
@@ -239,9 +243,9 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")
         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")
-            current_location = gr.Textbox(label="Current Location (From)", placeholder="e.g., Dubai, London (optional)")
 
         with gr.Row():
             budget = gr.Dropdown(["Low", "Medium", "High", "Any"], label="Budget", value="Any")
@@ -268,7 +272,7 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")
 
     generate_btn.click(
         generate_itinerary,
-        inputs=[mood, location, budget, days, traveller_type, current_location],
+        inputs=[mood, location, budget, days, traveller_type],
         outputs=[itinerary_output, image_output, map_output],
         show_progress=True
     )
@@ -280,4 +284,3 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")
     )
 
 demo.launch()
-