Update app.py

app.py

@@ -1,10 +1,19 @@
 import gradio as gr
 import os
 import pandas as pd
+from PIL import Image
 
+# ----- Config -----
 BASE_DIR = "data"
-
-# Get subfolders
+ASSETS_DIR = "assets"
+DEFAULT_MAP = os.path.join(ASSETS_DIR, "default_map.png")
+PRODUCT_IMAGE_MAP = {
+    "shampoo": os.path.join(ASSETS_DIR, "shampoo_map.png"),
+    "curd": os.path.join(ASSETS_DIR, "curd_map.png"),
+    "ghee": os.path.join(ASSETS_DIR, "ghee_map.png")
+}
+
+# ----- Utilities -----
 def get_subfolders(path):
     try:
         return sorted([f for f in os.listdir(path) if os.path.isdir(os.path.join(path, f))])
@@ -12,7 +21,6 @@ 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")])
@@ -20,7 +28,6 @@ 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)
@@ -31,7 +38,6 @@ 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)
@@ -43,7 +49,6 @@ 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
@@ -52,102 +57,53 @@ def display_quantity_info(quantity, data_dict):
     except Exception as e:
         return f"⚠️ Error: {e}"
 
-# Smart Suggestions (hardcoded logic)
-def suggest_items(query):
-    query = query.lower()
-    if "gift" in query and "500" in query:
-        return (
-            "🎁 Gift Suggestions under ₹500:\n"
-            "1. Bath & Body Gift Set - ₹499\n"
-            "2. Mini Perfume Pack - ₹349\n"
-            "3. Skin Care Hamper - ₹399\n"
-            "4. Chocolates Gift Box - ₹299"
-        )
-    if "shampoo" in query and "dry" in query:
-        return (
-            "🧴 Shampoos for Dry Hair:\n"
-            "1. Dove 500 ml - ₹325\n"
-            "2. Clinic Plus 500 ml - ₹680"
-        )
-    return "🤷 Sorry, no smart suggestions found. Try asking: 'Gift items under 500' or 'Shampoo for dry hair'"
-
-# Reset logic
+def get_product_image(product_name):
+    lower_name = product_name.strip().lower()
+    return Image.open(PRODUCT_IMAGE_MAP.get(lower_name, DEFAULT_MAP))
+
 def reset_all():
     return (
         None, None, None, None, None, None, None,
-        gr.update(choices=[], visible=False), "", {}
+        gr.update(choices=[], visible=False), "", {}, Image.open(DEFAULT_MAP)
     )
 
-# Interface
+# ----- UI -----
 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)
-                state = gr.Dropdown(label="🏙️ State", choices=[], interactive=False)
-                city = gr.Dropdown(label="🏘️ City", choices=[], interactive=False)
-                store = gr.Dropdown(label="🏪 Store", choices=[], interactive=False)
-                category = gr.Dropdown(label="🛍️ Category", choices=[], interactive=False)
-                product = gr.Dropdown(label="📦 Product", choices=[], interactive=False)
-                brand = gr.Dropdown(label="🏷️ Brand", choices=[], interactive=False)
-                quantity = gr.Dropdown(label="🔢 Quantity", visible=False)
+                country = gr.Dropdown(label="🌍 Country", choices=get_subfolders(BASE_DIR), interactive=True)
+                state = gr.Dropdown(label="🏙️ State", choices=[], interactive=True)
+                city = gr.Dropdown(label="🏘️ City", choices=[], interactive=True)
+                store = gr.Dropdown(label="🏪 Store", choices=[], interactive=True)
+                category = gr.Dropdown(label="🛍️ Category", choices=[], interactive=True)
+                product = gr.Dropdown(label="📦 Product", choices=[], interactive=True)
+                brand = gr.Dropdown(label="🏷️ Brand", choices=[], interactive=True)
+                quantity = gr.Dropdown(label="🔢 Quantity", visible=False, interactive=True)
 
             result = gr.Textbox(label="🔍 Product Info", lines=5)
             data_state = gr.State()
+            store_map = gr.Image(value=Image.open(DEFAULT_MAP), type="pil", label="🗺️ Store Map")
             reset_btn = gr.Button("🔄 Reset All")
 
-            # 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
-            )
+            # Dropdown chaining
+            country.change(lambda c: gr.update(choices=get_subfolders(os.path.join(BASE_DIR, c)) if c else [], value=None), 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), 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), 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), 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), 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), inputs=[country, state, city, store, category, product], outputs=brand)
 
-            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
-            )
+            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])
 
-            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
-            )
+            quantity.change(display_quantity_info, inputs=[quantity, data_state], outputs=result)
 
-            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]
-            )
+            product.change(fn=get_product_image, inputs=product, outputs=store_map)
 
-            quantity.change(display_quantity_info, inputs=[quantity, data_state], outputs=result)
-            reset_btn.click(reset_all, inputs=[], outputs=[country, state, city, store, category, product, brand, quantity, result, data_state])
+            reset_btn.click(reset_all, inputs=[], outputs=[country, state, city, store, category, product, brand, quantity, result, data_state, store_map])
 
-        # 🎁 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)