Update app.py

app.py

@@ -4,25 +4,23 @@ import pandas as pd
 
 BASE_DIR = "data"
 
-# Helper: Get subfolders at a given path
+# Get subfolders
 def get_subfolders(path):
-    print(f"📁 Reading subfolders from: {path}")
     try:
         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: {e}")
+        print(f"❌ Error reading subfolders from {path}: {e}")
         return []
 
-# Helper: Get all .csv filenames (without .csv)
+# Get CSV files
 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 Exception as e:
-        print(f"❌ Error: {e}")
+        print(f"❌ Error reading CSVs from {path}: {e}")
         return []
 
-# Read CSV for brand and return available quantity options
+# Load quantities from brand CSV
 def get_quantities_from_csv(path):
     try:
         df = pd.read_csv(path)
@@ -30,9 +28,10 @@ def get_quantities_from_csv(path):
             return gr.update(choices=[], visible=False), {}
         return gr.update(choices=df["Quantity"].dropna().tolist(), visible=True), df.to_dict()
     except Exception as e:
+        print(f"❌ Error loading CSV: {e}")
         return gr.update(choices=[], visible=False), {}
 
-# Show info based on selected quantity
+# Show result based on quantity
 def display_quantity_info(quantity, data_dict):
     try:
         df = pd.DataFrame(data_dict)
@@ -47,45 +46,38 @@ def display_quantity_info(quantity, data_dict):
         else:
             return f"❌ Sorry, {quantity} is currently not in stock."
     except Exception as e:
-        return f"⚠️ Error processing quantity info: {e}"
+        return f"⚠️ Error: {e}"
 
-# Reset all fields
+# Reset all
 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
+        None, None, None, None, None, None, None,
+        gr.update(choices=[], visible=False),
+        "", {}
     )
 
-# Interface
+# UI
 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)
-    quantity = gr.Dropdown(label="🔢 Quantity", visible=False)
+        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)
+        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: Chain each dropdown to the next
+    # Logic chaining
     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)), value=None, interactive=True),
         inputs=country,
         outputs=state
     )