Update app.py

app.py

@@ -80,13 +80,13 @@ def suggest_items(query):
         )
     return "🤷 Sorry, no smart suggestions found. Try asking: 'Gift items under 500' or 'Shampoo for dry hair'"
 
-
+# Corrected reset_all function
 def reset_all():
     return (
         None, None, None, None, None, None, None,
-        gr.update(choices=[], visible=False),  # quantity
-        "",  # result textbox
-        {}  # data_state
+        gr.update(choices=[], visible=False),  # quantity dropdown reset
+        "",  # result textbox reset
+        {}  # data_state (gr.State) reset with raw dict, NOT gr.update()
     )
 
 # ----- UI -----
@@ -110,28 +110,58 @@ with gr.Blocks(title="RetailGenie") as demo:
             store_map = gr.Image(value=Image.open(DEFAULT_MAP), type="pil", label="🗺️ Store Map")
             reset_btn = gr.Button("🔄 Reset All")
 
-            # Dropdown chaining
-            country.change(lambda c: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c)) if c else [], value=None), inputs=country, outputs=state)
-            state.change(lambda c, s: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c, s)) if c and s else [], value=None), inputs=[country, state], outputs=city)
-            city.change(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), 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)) if all([c, s, ci, 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)) if all([c, s, ci, st, cat]) else [], value=None), 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)) if all([c, s, ci, st, cat, prod]) else [], value=None), inputs=[country, state, city, store, category, product], outputs=brand)
+            # Dropdown chaining with consistent inputs and outputs
+            country.change(
+                lambda c: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c)) if c else [], value=None),
+                inputs=country, outputs=state
+            )
+            state.change(
+                lambda c, s: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c, s)) if c and s else [], value=None),
+                inputs=[country, state], outputs=city
+            )
+            city.change(
+                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),
+                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)) if all([c, s, ci, 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)) if all([c, s, ci, st, cat]) else [], value=None),
+                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)) if all([c, s, ci, st, cat, prod]) else [], value=None),
+                inputs=[country, state, city, store, category, product], outputs=brand
+            )
 
-            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])
+            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]
+            )
 
-            quantity.change(display_quantity_info, inputs=[quantity, data_state], outputs=result)
+            quantity.change(
+                display_quantity_info,
+                inputs=[quantity, data_state],
+                outputs=result
+            )
 
-            product.change(fn=get_product_image, inputs=product, outputs=store_map)
+            product.change(
+                fn=get_product_image,
+                inputs=product,
+                outputs=store_map
+            )
 
             reset_btn.click(
-              reset_all,
-              inputs=[],
-              outputs=[
-                   country, state, city, store,
-                   category, product, brand, quantity,
-                   result, data_state
-              ]
+                reset_all,
+                inputs=[],
+                outputs=[
+                    country, state, city, store,
+                    category, product, brand, quantity,
+                    result, data_state
+                ]
             )
 
         with gr.TabItem("🎁 Smart Suggestions"):
@@ -140,6 +170,10 @@ with gr.Blocks(title="RetailGenie") as demo:
             suggest_btn = gr.Button("💡 Get Suggestions")
             suggestions_output = gr.Textbox(label="📝 Suggestions", lines=10)
 
-            suggest_btn.click(suggest_items, inputs=suggestion_input, outputs=suggestions_output)
+            suggest_btn.click(
+                suggest_items,
+                inputs=suggestion_input,
+                outputs=suggestions_output
+            )
 
 demo.launch()