Update app.py

app.py

@@ -52,62 +52,60 @@ def display_quantity_info(quantity, data_dict):
 def reset_all():
     return (
         None, None, None, None, None, None, None,
-        gr.update(choices=[], visible=False),
-        "", {}
+        gr.update(choices=[], visible=False), "", {}
     )
 
-# UI
+# Interface
 with gr.Blocks(title="RetailGenie Navigator") as demo:
     gr.Markdown("## 🧞 RetailGenie – Multi-Store Smart Inventory Assistant")
 
     with gr.Row():
         country = gr.Dropdown(label="🌍 Country", choices=get_subfolders(BASE_DIR), value=None)
-        state = gr.Dropdown(label="🏙️ State", choices=[], visible=True, interactive=False)
-        city = gr.Dropdown(label="🏘️ City", choices=[], visible=True, interactive=False)
-        store = gr.Dropdown(label="🏪 Store", choices=[], visible=True, interactive=False)
-        category = gr.Dropdown(label="🛍️ Category", choices=[], visible=True, interactive=False)
-        product = gr.Dropdown(label="📦 Product", choices=[], visible=True, interactive=False)
-        brand = gr.Dropdown(label="🏷️ Brand", choices=[], visible=True, interactive=False)
+        state = gr.Dropdown(label="🏙️ State", choices=[], interactive=False)
+        city = gr.Dropdown(label="🏘️ City", choices=[], interactive=False)
+        store = gr.Dropdown(label="🏪 Store", choices=[], interactive=False)
+        category = gr.Dropdown(label="🛍️ Category", choices=[], interactive=False)
+        product = gr.Dropdown(label="📦 Product", choices=[], interactive=False)
+        brand = gr.Dropdown(label="🏷️ Brand", choices=[], interactive=False)
         quantity = gr.Dropdown(label="🔢 Quantity", visible=False)
 
     result = gr.Textbox(label="🔍 Product Info", lines=5)
     data_state = gr.State()
-
     reset_btn = gr.Button("🔄 Reset All")
 
-    # Logic chaining
+    # Logic chaining with safety checks
     country.change(
-        lambda c: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c)), value=None, interactive=True),
+        lambda c: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c)) if c else [], value=None, interactive=True),
         inputs=country,
         outputs=state
     )
 
     state.change(
-        lambda c, s: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c, s)), value=None, interactive=True),
+        lambda c, s: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c, s)) if c and s else [], value=None, interactive=True),
         inputs=[country, state],
         outputs=city
     )
 
     city.change(
-        lambda c, s, ci: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c, s, ci)), value=None, interactive=True),
+        lambda c, s, ci: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c, s, ci)) if c and s and ci else [], value=None, interactive=True),
         inputs=[country, state, city],
         outputs=store
     )
 
     store.change(
-        lambda c, s, ci, st: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c, s, ci, st)), value=None, interactive=True),
+        lambda c, s, ci, st: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c, s, ci, st)) if c and s and ci and st else [], value=None, interactive=True),
         inputs=[country, state, city, store],
         outputs=category
     )
 
     category.change(
-        lambda c, s, ci, st, cat: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c, s, ci, st, cat)), value=None, interactive=True),
+        lambda c, s, ci, st, cat: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c, s, ci, st, cat)) if c and s and ci and st and cat else [], value=None, interactive=True),
         inputs=[country, state, city, store, category],
         outputs=product
     )
 
     product.change(
-        lambda c, s, ci, st, cat, prod: gr.update(choices=get_csv_files(os.path.join(BASE_DIR, c, s, ci, st, cat, prod)), value=None, interactive=True),
+        lambda c, s, ci, st, cat, prod: gr.update(choices=get_csv_files(os.path.join(BASE_DIR, c, s, ci, st, cat, prod)) if c and s and ci and st and cat and prod else [], value=None, interactive=True),
         inputs=[country, state, city, store, category, product],
         outputs=brand
     )
@@ -115,7 +113,7 @@ with gr.Blocks(title="RetailGenie Navigator") as demo:
     brand.change(
         lambda c, s, ci, st, cat, prod, b: get_quantities_from_csv(
             os.path.join(BASE_DIR, c, s, ci, st, cat, prod, f"{b}.csv")
-        ),
+        ) if all([c, s, ci, st, cat, prod, b]) else (gr.update(choices=[], visible=False), {}),
         inputs=[country, state, city, store, category, product, brand],
         outputs=[quantity, data_state]
     )