Update app.py

app.py

@@ -28,6 +28,15 @@ MOODS = [
     "Beach & Island",
     "Any"
 ]
+TRAVELLER_TYPES = [
+    "Solo",
+    "Couple",
+    "Family",
+    "Friends",
+    "Business",
+    "Any"
+]
+
 
 MOOD_IMAGES = {
     "Adventure": "images/adventure.png",
@@ -75,10 +84,21 @@ def extract_places_and_map(itinerary_text, city_context=None):
 
     return m._repr_html_()
 
-def generate_itinerary(mood, location, budget, days):
+def generate_itinerary(mood, location, budget, days, traveller_type, current_location):
     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 = (
+    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"and the budget is {budget_text}. "
+    f"Include daily activities, key landmarks, and meal or cultural suggestions. "
+    f"Use Markdown formatting with **bold headings** for days and time slots."
+)
+
     
     prompt = (
         f"Create a {days}-day travel itinerary to {location_text}. "
@@ -227,6 +247,10 @@ 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")
@@ -253,11 +277,12 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")
 
     generate_btn.click(
         generate_itinerary,
-        inputs=[mood, location, budget, days],
+        inputs=[mood, location, budget, days, traveller_type, current_location],
         outputs=[itinerary_output, image_output, map_output],
         show_progress=True
     )
 
+
     download_btn.click(
         generate_pdf,
         inputs=[itinerary_output],