Update app.py

app.py

@@ -1,52 +1,79 @@
-import pandas as pd
 import gradio as gr
+import os
+import pandas as pd
 
-# Load data
-df = pd.read_csv("inventory.csv")
+BASE_DIR = "data"
 
-# Get category list
-categories = sorted(df['Category'].dropna().unique())
+def get_subfolders(path):
+    try:
+        return sorted([f for f in os.listdir(path) if os.path.isdir(os.path.join(path, f))])
+    except FileNotFoundError:
+        return []
 
-# Product selector from category
-def get_products(category):
-    products = df[df['Category'] == category]['Product Name'].unique().tolist()
-    return gr.update(choices=products, visible=True), category
+def get_csv_files(path):
+    try:
+        return sorted([f[:-4] for f in os.listdir(path) if f.endswith(".csv")])
+    except FileNotFoundError:
+        return []
 
-# Final result message
-def get_product_info(product, category):
-    match = df[(df['Category'] == category) & (df['Product Name'] == product)]
-    if not match.empty:
-        row = match.iloc[0]
+def load_csv_info(path):
+    try:
+        df = pd.read_csv(path)
+        if df.empty:
+            return "⚠️ No data found in the selected brand file."
+        row = df.iloc[0]
         if str(row["In Stock"]).strip().lower() == "yes":
-            return f"✅ {product} is available in Aisle {row['Aisle']}. We have {row['Quantity']} in stock, and it's priced at ₹{row['Price (rupees)']}."
+            return (
+                f"✅ {os.path.basename(path).replace('.csv', '')} is in stock.\n"
+                f"- Quantity: {row['Quantity']}\n"
+                f"- Floor: {row['Floor']}\n"
+                f"- Aisle: {row['Aisle']}\n"
+                f"- Price: ₹{row['Price']}"
+            )
         else:
-            return f"❌ Sorry, {product} is currently not available in stock."
-    else:
-        return "⚠️ Product not found."
+            return f"❌ {os.path.basename(path).replace('.csv', '')} is currently out of stock."
+    except Exception as e:
+        return f"⚠️ Error reading CSV: {e}"
 
-# Clear function
-def reset():
-    return gr.update(value=None), gr.update(choices=[], visible=False), "", ""
-
-# Gradio App
-with gr.Blocks(title="RetailGenie") as demo:
-    gr.Markdown("## 🧞 RetailGenie – Your Smart In-Store Assistant")
-    gr.Markdown("### 👋 Welcome! Let me help you quickly find products in your store.")
+with gr.Blocks(title="RetailGenie Navigator") as demo:
+    gr.Markdown("## 🧞 RetailGenie – Multi-Store Smart Inventory Assistant")
 
     with gr.Row():
-        with gr.Column():
-            category_input = gr.Dropdown(label="🗂️ Choose Product Category", choices=categories)
-            product_input = gr.Dropdown(label="📦 Choose a Product", visible=False)
-            clear_btn = gr.Button("🔁 Start Over")
+        country = gr.Dropdown(label="🌍 Country", choices=get_subfolders(BASE_DIR))
+        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")
 
-        with gr.Column():
-            message = gr.Textbox(label="💬 Assistant Response", interactive=False)
+    result = gr.Textbox(label="🔍 Product Info", lines=5)
 
     # Logic chaining
-    state_category = gr.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),
+                 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),
+                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)
+
+    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)
 
-    category_input.change(get_products, inputs=category_input, outputs=[product_input, state_category])
-    product_input.change(get_product_info, inputs=[product_input, state_category], outputs=message)
-    clear_btn.click(reset, outputs=[category_input, product_input, message, state_category])
+    brand.change(
+        lambda c, s, ci, st, cat, prod, b: load_csv_info(
+            os.path.join(BASE_DIR, c, s, ci, st, cat, prod, f"{b}.csv")
+        ),
+        inputs=[country, state, city, store, category, product, brand],
+        outputs=result
+    )
 
 demo.launch()