Update app.py

app.py

@@ -109,11 +109,11 @@ def generate_itinerary(mood, location, budget, days):
     # Generate map HTML from itinerary text and user location
     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
+    return itinerary_text, image_path, map_html,gr.update(visible=True)
 
 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")
@@ -142,7 +142,7 @@ def generate_pdf(itinerary_text):
 
     doc.build(story)
 
-    return temp_path
+    return temp_path,gr.update(visible=True)
 
 
 with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")) as demo:
@@ -182,24 +182,24 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")
         
         with gr.Row():
             with gr.Column():
-                image_output = gr.Image(label="Travel Mood Image", type="filepath", visible=True)
+                image_output = gr.Image(label="Travel Mood Image",type="filepath", visible=False)
             with gr.Column():
-                map_output = gr.HTML(label="Map of Suggested Places", elem_id="map-container")
+                map_output = gr.HTML(label="Map of Suggested Places",elem_id="map-container")
 
     with gr.Column():
-        download_btn = gr.Button("Download Itinerary as PDF", variant="secondary")
-        pdf_file = gr.File(label="Download PDF", file_types=[".pdf"])
+        download_btn = gr.Button("Download Itinerary as PDF",variant="secondary")
+        pdf_file = gr.File(label="Download PDF", file_types=[".pdf"],visible=False)
 
     generate_btn.click(
         generate_itinerary,
         inputs=[mood, location, budget, days],
-        outputs=[itinerary_output, image_output, map_output],
+        outputs=[itinerary_output, image_output, map_output, image_output],
         show_progress=True
     )
     download_btn.click(
         generate_pdf,
         inputs=[itinerary_output],
-        outputs=pdf_file
+        outputs=[pdf_file, pdf_file]
     )
 
 demo.launch()