Update app.py

app.py

@@ -37,7 +37,6 @@ TRAVELLER_TYPES = [
     "Any"
 ]
 
-
 MOOD_IMAGES = {
     "Adventure": "images/adventure.png",
     "Relaxing": "images/relaxing.png",
@@ -91,21 +90,14 @@ def generate_itinerary(mood, location, budget, days, traveller_type, current_loc
     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 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}. "
-    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."
-)
-    
     completion = client.chat.completions.create(
         model="gemma2-9b-it",
         messages=[
@@ -117,10 +109,9 @@ prompt = (
         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
@@ -157,6 +148,7 @@ def generate_pdf(itinerary_text):
     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:
     
@@ -251,7 +243,6 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")
             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")
             days = gr.Slider(1, 10, value=5, step=1, label="Days")
@@ -282,7 +273,6 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")
         show_progress=True
     )
 
-
     download_btn.click(
         generate_pdf,
         inputs=[itinerary_output],
@@ -290,3 +280,4 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")
     )
 
 demo.launch()
+