examples bug fix

app.py

@@ -31,7 +31,6 @@ def create_ui():
                      output,
                      generate_text_fn=generate_text,
                      clear_fn=clear)
-        # Load the examples for the interface
         load_examples(country, starting_point, query, sustainable, model, output, generate_text)
 
     return app

src/ui/components/inputs.py

@@ -12,7 +12,10 @@ def get_places():
     df = df.sort_values(by=['country', 'city'])
     return df
 
+
 def update_cities(selected_country, df):
+    if not selected_country:
+        return gr.update(choices=[], value=None, interactive=False)
     # Filter cities based on the selected country
     filtered_cities = df[df['country'] == selected_country]['city'].tolist()
     return gr.Dropdown(choices=filtered_cities, interactive=True)
@@ -32,16 +35,17 @@ def main_component() -> Tuple[gr.Dropdown, gr.Dropdown, gr.Textbox, gr.Checkbox,
     """
     df = get_places()
     country_names = list(df.country.unique())
+    cities = list(df.city.unique())
     with gr.Group():
         # Country selection dropdown
         country = gr.Dropdown(choices=country_names, multiselect=False, label="Country")
 
         # Starting point selection dropdown
-        starting_point = gr.Dropdown(choices=[], multiselect=False,
+        starting_point = gr.Dropdown(choices=cities, multiselect=False,
                                      label="City",
                                      info="Select a city as your starting point.")
 
-        # When a country is selected, update the starting point options
+        # # When a country is selected, update the starting point options
         country.select(
             fn=lambda selected_country: update_cities(selected_country, df),
             inputs=country,

src/ui/components/static.py

@@ -5,6 +5,10 @@ import gradio as gr
 from src.ui.components.inputs import get_places, update_cities, main_component
 
 
+def show_examples(examples):
+    return gr.Dataset(samples=examples)
+
+
 def load_examples(
         country: gr.components.Component,
         starting_point: gr.components.Component,
@@ -20,17 +24,20 @@ def load_examples(
     df = get_places()
     # Provide examples
     example_data = [
-        ["Austria", "Vienna", "What are some top tourist attractions?", True, "Gemini-1.0-pro"],
-        ["Germany", "Berlin", "Suggest a one week long itinerary for a trip in France.", False, "Gemini-1.5-Pro"],
+        ["Austria", "Vienna",
+         "I'm planning a trip in July and enjoy beaches, nightlife, and vibrant cities. Recommend some cities.",
+         True, "Gemini-1.0-pro"],
+        ["Germany", "Berlin",
+         "For winter travel, can you recommend destinations for Christmas markets? I love festive atmospheres.", False,
+         "Gemini-1.5-Pro"],
     ]
-    eg = gr.Examples(
+    gr.Examples(
         examples=example_data,
         inputs=[country, starting_point, query, sustainable, model],
         fn=generate_text_fn,
         outputs=[output],
         cache_examples=True,
     )
-    print(f"Examples loaded {eg.dataset.samples}")
 
     starting_point.change(
         fn=lambda selected_country: update_cities(selected_country, df),
@@ -38,8 +45,6 @@ def load_examples(
         outputs=starting_point
     )
 
-    print("Examples loaded")
-
 
 def load_buttons(
         country: gr.components.Component,