Update app.py

app.py

@@ -16,24 +16,30 @@ def get_csv_files(path):
     except FileNotFoundError:
         return []
 
-def load_csv_info(path):
+def get_quantities_from_csv(path):
     try:
         df = pd.read_csv(path)
-        if df.empty:
-            return "⚠️ No data found in the selected brand file."
-        row = df.iloc[0]
+        if df.empty or "Quantity" not in df.columns:
+            return gr.update(choices=[], visible=False), gr.State({})
+        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({})
+
+def display_quantity_info(quantity, data_dict):
+    try:
+        df = pd.DataFrame(data_dict)
+        row = df[df["Quantity"] == quantity].iloc[0]
         if str(row["In Stock"]).strip().lower() == "yes":
             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']}"
+                f"✅ {quantity} is available!\n"
+                f"• Floor: {row['Floor']}\n"
+                f"• Aisle: {row['Aisle']}\n"
+                f"• Price: ₹{row['Price']}"
             )
         else:
-            return f"❌ {os.path.basename(path).replace('.csv', '')} is currently out of stock."
+            return f"❌ Sorry, {quantity} is currently not in stock."
     except Exception as e:
-        return f"⚠️ Error reading CSV: {e}"
+        return f"⚠️ Error processing quantity info: {e}"
 
 with gr.Blocks(title="RetailGenie Navigator") as demo:
     gr.Markdown("## 🧞 RetailGenie – Multi-Store Smart Inventory Assistant")
@@ -46,6 +52,7 @@ with gr.Blocks(title="RetailGenie Navigator") as demo:
         category = gr.Dropdown(label="🛍️ Category")
         product = gr.Dropdown(label="📦 Product")
         brand = gr.Dropdown(label="🏷️ Brand")
+        quantity = gr.Dropdown(label="🔢 Quantity", visible=False)
 
     result = gr.Textbox(label="🔍 Product Info", lines=5)
 
@@ -68,11 +75,19 @@ with gr.Blocks(title="RetailGenie Navigator") as demo:
     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: load_csv_info(
+        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()]
+    )
+
+    # When quantity is selected → lookup and display info
+    quantity.change(
+        display_quantity_info,
+        inputs=[quantity, gr.State()],
         outputs=result
     )