Update app.py

app.py

@@ -4,23 +4,25 @@ import pandas as pd
 
 BASE_DIR = "data"
 
+# Helper: Get subfolders at a given path
 def get_subfolders(path):
-    print(f"📂 Reading subfolders from: {path}")
+    print(f"📁 Reading subfolders from: {path}")
     try:
-        folders = sorted([f for f in os.listdir(path) if os.path.isdir(os.path.join(path, f))])
-        print(f"✅ Found: {folders}")
-        return folders
+        return sorted([f for f in os.listdir(path) if os.path.isdir(os.path.join(path, f))])
     except Exception as e:
-        print(f"❌ Error reading {path}: {e}")
+        print(f"❌ Error: {e}")
         return []
 
-
+# Helper: Get all .csv filenames (without .csv)
 def get_csv_files(path):
+    print(f"📄 Reading CSV files from: {path}")
     try:
         return sorted([f[:-4] for f in os.listdir(path) if f.endswith(".csv")])
-    except FileNotFoundError:
+    except Exception as e:
+        print(f"❌ Error: {e}")
         return []
 
+# Read CSV for brand and return available quantity options
 def get_quantities_from_csv(path):
     try:
         df = pd.read_csv(path)
@@ -30,6 +32,7 @@ def get_quantities_from_csv(path):
     except Exception as e:
         return gr.update(choices=[], visible=False), {}
 
+# Show info based on selected quantity
 def display_quantity_info(quantity, data_dict):
     try:
         df = pd.DataFrame(data_dict)
@@ -46,40 +49,52 @@ def display_quantity_info(quantity, data_dict):
     except Exception as e:
         return f"⚠️ Error processing quantity info: {e}"
 
+# Reset all fields
+def reset_all():
+    return (
+        None,  # country
+        None,  # state
+        None,  # city
+        None,  # store
+        None,  # category
+        None,  # product
+        None,  # brand
+        gr.update(choices=[], visible=False),  # quantity
+        "",  # result
+        {}   # data state
+    )
+
+# 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")
-        city = gr.Dropdown(label="🏘️ City")
-        store = gr.Dropdown(label="🏪 Store")
-        category = gr.Dropdown(label="🛍️ Category")
-        product = gr.Dropdown(label="📦 Product")
-        brand = gr.Dropdown(label="🏷️ Brand")
+        state = gr.Dropdown(label="🏙️ State", value=None)
+        city = gr.Dropdown(label="🏘️ City", value=None)
+        store = gr.Dropdown(label="🏪 Store", value=None)
+        category = gr.Dropdown(label="🛍️ Category", value=None)
+        product = gr.Dropdown(label="📦 Product", value=None)
+        brand = gr.Dropdown(label="🏷️ Brand", value=None)
         quantity = gr.Dropdown(label="🔢 Quantity", visible=False)
 
     result = gr.Textbox(label="🔍 Product Info", lines=5)
     data_state = gr.State()
 
-    # Logic chaining
+    reset_btn = gr.Button("🔄 Reset All")
+
+    # Logic: Chain each dropdown to the next
     country.change(
-    lambda c: (
-        print(f"🔽 Selected country: {c}"),
-        gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c)), value=None)
-    )[1],
-    inputs=country,
-    outputs=state
-)
+        lambda c: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c)), value=None),
+        inputs=country,
+        outputs=state
+    )
 
     state.change(
-    lambda c, s: (
-        print(f"🔽 Selected state: {s} under country: {c}"),
-        gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c, s)), value=None)
-    )[1],
-    inputs=[country, state],
-    outputs=city
-)
+        lambda c, s: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c, s)), 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)), value=None),
@@ -119,4 +134,10 @@ with gr.Blocks(title="RetailGenie Navigator") as demo:
         outputs=result
     )
 
+    reset_btn.click(
+        reset_all,
+        inputs=[],
+        outputs=[country, state, city, store, category, product, brand, quantity, result, data_state]
+    )
+
 demo.launch()