Update app.py

app.py

@@ -20,10 +20,10 @@ def get_quantities_from_csv(path):
     try:
         df = pd.read_csv(path)
         if df.empty or "Quantity" not in df.columns:
-            return gr.update(choices=[], visible=False), gr.State({})
+            return gr.update(choices=[], visible=False), {}
         return gr.update(choices=df["Quantity"].dropna().tolist(), visible=True), df.to_dict()
     except Exception as e:
-        return gr.update(choices=[], visible=False), gr.State({})
+        return gr.update(choices=[], visible=False), {}
 
 def display_quantity_info(quantity, data_dict):
     try:
@@ -55,10 +55,14 @@ with gr.Blocks(title="RetailGenie Navigator") as demo:
         quantity = gr.Dropdown(label="🔢 Quantity", visible=False)
 
     result = gr.Textbox(label="🔍 Product Info", lines=5)
+    data_state = gr.State()
 
     # Logic chaining
-    country.change(lambda c: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c)), value=None),
-                   inputs=country, outputs=state)
+    country.change(
+        lambda c: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c)), value=None),
+        inputs=country,
+        outputs=state
+    )
 
     state.change(
         lambda c, s: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c, s)), value=None),
@@ -66,31 +70,41 @@ with gr.Blocks(title="RetailGenie Navigator") as demo:
         outputs=city
     )
 
-    city.change(lambda c, s, ci: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c, s, ci)), value=None),
-                inputs=[country, state, city], outputs=store)
+    city.change(
+        lambda c, s, ci: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c, s, ci)), 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)), value=None),
-                 inputs=[country, state, city, store], outputs=category)
+    store.change(
+        lambda c, s, ci, st: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c, s, ci, st)), value=None),
+        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),
-                    inputs=[country, state, city, store, category], outputs=product)
+    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),
+        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),
-                   inputs=[country, state, city, store, category, product], outputs=brand)
+    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),
+        inputs=[country, state, city, store, category, product],
+        outputs=brand
+    )
 
-    # When brand is selected → get quantities + dataframe 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")
         ),
         inputs=[country, state, city, store, category, product, brand],
-        outputs=[quantity, gr.State()]
+        outputs=[quantity, data_state]
     )
 
-    # When quantity is selected → lookup and display info
     quantity.change(
         display_quantity_info,
-        inputs=[quantity, gr.State()],
+        inputs=[quantity, data_state],
         outputs=result
     )