Update app.py

app.py

@@ -1,6 +1,5 @@
 import os
 import json
-import time
 from typing import List, Dict, Any
 
 import gradio as gr
@@ -26,20 +25,16 @@ rag_store = RagStore(index_dir=INDEX_DIR)
 
 # ------------- Handlers -------------
 
-
 def seed_sample_data() -> str:
     from sample_data_seed import seed
-
     n = seed()
     return f"Seeded sample data: {n} files."
 
-
 def handle_build_index(progress=gr.Progress(track_tqdm=True)) -> str:
     docs = build_corpus(CORPUS_DIRS)
     rag_store.build(docs)
     return f"Built index with {len(docs)} documents."
 
-
 def handle_generate_plan(
     city: str,
     hours: float,
@@ -49,20 +44,19 @@ def handle_generate_plan(
     pace: str,
     date: str,
     force_weather: str | None,
+    mode_pref: str,
+    show_costs: bool,
 ):
     if not rag_store.available():
         return None, None, "Index not built yet. Click 'Build/Update RAG Index'."
 
-    # Geocode city -> lat/lon
     loc = geocode_city(city)
     if not loc:
         return None, None, f"Failed to geocode: {city}"
     lat, lon = loc["lat"], loc["lon"]
 
-    # Weather summary
     weather = get_weather_summary(lat, lon, date=date, override=force_weather or None)
 
-    # Generate plan
     plan = generate_plan(
         rag=rag_store,
         center=(lat, lon),
@@ -73,12 +67,12 @@ def handle_generate_plan(
         pace=pace,
         weather=weather,
         date=date,
+        mode_pref=mode_pref,
+        include_costs=show_costs,
     )
 
-    # Map HTML, itinerary table, narrative text
     return plan["map_html"], plan["table"], plan["narrative"]
 
-
 with gr.Blocks(css=".small {font-size: 12px}") as demo:
     (
         city_in,
@@ -89,6 +83,8 @@ with gr.Blocks(css=".small {font-size: 12px}") as demo:
         pace_in,
         date_in,
         weather_in,
+        mode_in,
+        show_costs_in,
         build_btn,
         seed_btn,
         gen_btn,
@@ -110,6 +106,8 @@ with gr.Blocks(css=".small {font-size: 12px}") as demo:
             pace_in,
             date_in,
             weather_in,
+            mode_in,
+            show_costs_in,
         ],
         outputs=[map_out, table_out, text_out],
     )