Update app.py

app.py

@@ -1,23 +1,10 @@
 import gradio as gr
 import os
 import pandas as pd
-from PIL import Image, ImageDraw
 
 BASE_DIR = "data"
-ASSETS_DIR = "assets"
-STORE_MAP_FILE = os.path.join(ASSETS_DIR, "store_map_clinicplus.png")
-
-
-# Approximate mapping from Aisle number to bounding box on the store map
-AISLE_TO_BOX = {
-    # Example: '14': (x, y, width, height) - you can later load this from config
-    "14": (220, 150, 40, 120),
-    "15": (270, 150, 40, 120),
-    "16": (320, 150, 40, 120),
-    # Add more as needed...
-}
-
 
+# Get subfolders
 def get_subfolders(path):
     try:
         return sorted([f for f in os.listdir(path) if os.path.isdir(os.path.join(path, f))])
@@ -25,7 +12,7 @@ def get_subfolders(path):
         print(f"❌ Error reading subfolders from {path}: {e}")
         return []
 
-
+# Get CSV files
 def get_csv_files(path):
     try:
         return sorted([f[:-4] for f in os.listdir(path) if f.endswith(".csv")])
@@ -33,7 +20,7 @@ def get_csv_files(path):
         print(f"❌ Error reading CSVs from {path}: {e}")
         return []
 
-
+# Load quantities from brand CSV
 def get_quantities_from_csv(path):
     try:
         df = pd.read_csv(path)
@@ -44,12 +31,11 @@ def get_quantities_from_csv(path):
         print(f"❌ Error loading CSV: {e}")
         return gr.update(choices=[], visible=False), {}
 
-
+# Show result based on quantity
 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":
             msg = (
                 f"✅ {quantity} is available!\n"
@@ -57,6 +43,7 @@ def display_quantity_info(quantity, data_dict):
                 f"• Aisle: {row['Aisle']}\n"
                 f"• Price: ₹{row['Price']}"
             )
+            # Show only Offer field if it exists and is not empty
             if "Offer" in row and pd.notna(row["Offer"]) and row["Offer"].strip():
                 msg += f"\n• 🎉 Offer: {row['Offer']}"
             return msg
@@ -65,26 +52,7 @@ def display_quantity_info(quantity, data_dict):
     except Exception as e:
         return f"⚠️ Error: {e}"
 
-
-def generate_map_with_highlight(quantity, data_dict):
-    try:
-        df = pd.DataFrame(data_dict)
-        row = df[df["Quantity"] == quantity].iloc[0]
-        aisle = str(row.get("Aisle", "")).strip()
-
-        image = Image.open(STORE_MAP_FILE).convert("RGBA")
-        draw = ImageDraw.Draw(image)
-
-        if aisle in AISLE_TO_BOX:
-            x, y, w, h = AISLE_TO_BOX[aisle]
-            draw.rectangle([x, y, x + w, y + h], outline="red", width=5)
-
-        return image
-    except Exception as e:
-        print(f"⚠️ Map render error: {e}")
-        return Image.open(STORE_MAP_FILE)
-
-
+# Smart Suggestions (hardcoded logic)
 def suggest_items(query):
     query = query.lower()
     if "gift" in query and "500" in query:
@@ -103,28 +71,19 @@ def suggest_items(query):
         )
     return "🤷 Sorry, no smart suggestions found. Try asking: 'Gift items under 500' or 'Shampoo for dry hair'"
 
-
+# Reset logic
 def reset_all():
     return (
-        None,  # country - Dropdown
-        None,  # state - Dropdown
-        None,  # city - Dropdown
-        None,  # store - Dropdown
-        None,  # category - Dropdown
-        None,  # product - Dropdown
-        None,  # brand - Dropdown
-        gr.update(choices=[], visible=False),  # quantity - Dropdown
-        "",  # result - Textbox
-        {},  # data_state - State (MUST be a raw dict or string, not gr.update)
-        Image.open(STORE_MAP_FILE)  # store_map - Image (if using type="pil")
+        None, None, None, None, None, None, None,
+        gr.update(choices=[], visible=False), "", {}
     )
 
-
-
+# Interface
 with gr.Blocks(title="RetailGenie") as demo:
     gr.Markdown("# 🧞‍♂️ RetailGenie – In-Store Smart Assistant")
 
     with gr.Tabs():
+        # 🧭 Navigator Tab
         with gr.TabItem("🧭 Navigator"):
             with gr.Row():
                 country = gr.Dropdown(label="🌍 Country", choices=get_subfolders(BASE_DIR), value=None)
@@ -137,23 +96,58 @@ with gr.Blocks(title="RetailGenie") as demo:
                 quantity = gr.Dropdown(label="🔢 Quantity", visible=False)
 
             result = gr.Textbox(label="🔍 Product Info", lines=5)
-            store_map = gr.Image(label="🗺️ Store Map", value=Image.open(STORE_MAP_FILE), type="pil")
-            data_state = gr.State(value={}) 
+            data_state = gr.State()
             reset_btn = gr.Button("🔄 Reset All")
 
-            country.change(lambda c: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c)) if c else [], value=None, interactive=True), inputs=country, outputs=state)
-            state.change(lambda c, s: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c, s)) if c and s else [], value=None, interactive=True), inputs=[country, state], outputs=city)
-            city.change(lambda c, s, ci: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c, s, ci)) if c and s and ci else [], value=None, interactive=True), 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)) if all([c, s, ci, st]) else [], value=None, interactive=True), 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)) if all([c, s, ci, st, cat]) else [], value=None, interactive=True), 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)) if all([c, s, ci, st, cat, prod]) else [], value=None, interactive=True), inputs=[country, state, city, store, category, product], outputs=brand)
+            # Dropdown chain logic
+            country.change(
+                lambda c: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c)) if c else [], value=None, interactive=True),
+                inputs=country,
+                outputs=state
+            )
+
+            state.change(
+                lambda c, s: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c, s)) if c and s else [], value=None, interactive=True),
+                inputs=[country, state],
+                outputs=city
+            )
+
+            city.change(
+                lambda c, s, ci: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c, s, ci)) if c and s and ci else [], value=None, interactive=True),
+                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)) if all([c, s, ci, st]) else [], value=None, interactive=True),
+                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)) if all([c, s, ci, st, cat]) else [], value=None, interactive=True),
+                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)) if all([c, s, ci, st, cat, prod]) else [], value=None, interactive=True),
+                inputs=[country, state, city, store, category, product],
+                outputs=brand
+            )
 
-            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")) if all([c, s, ci, st, cat, prod, b]) else (gr.update(choices=[], visible=False), {}), inputs=[country, state, city, store, category, product, brand], outputs=[quantity, data_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")
+                ) if all([c, s, ci, st, cat, prod, b]) else (gr.update(choices=[], visible=False), {}),
+                inputs=[country, state, city, store, category, product, brand],
+                outputs=[quantity, data_state]
+            )
 
             quantity.change(display_quantity_info, inputs=[quantity, data_state], outputs=result)
-            quantity.change(generate_map_with_highlight, inputs=[quantity, data_state], outputs=store_map)
-            reset_btn.click(reset_all, inputs=[], outputs=[country, state, city, store, category, product, brand, quantity, result, data_state, store_map])
+            reset_btn.click(reset_all, inputs=[], outputs=[country, state, city, store, category, product, brand, quantity, result, data_state])
 
+        # 🎁 Suggestions Tab
         with gr.TabItem("🎁 Smart Suggestions"):
             gr.Markdown("### 🤖 Ask RetailGenie for Recommendations")
             suggestion_input = gr.Textbox(label="Ask something like:", placeholder="Gift items under 500", lines=1)