Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import os
 import pandas as pd
-from transformers import pipeline
+from PIL import Image
 
 # ----- Config -----
 BASE_DIR = "data"
@@ -55,7 +55,26 @@ def display_quantity_info(quantity, data_dict):
         else:
             return f"❌ Sorry, {quantity} is currently not in stock."
     except Exception as e:
-        return f"⚠️ Error: {e}"
+        return f"⚠ Error: {e}"
+
+# Smart Suggestions Function
+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'"
 
 def reset_all():
     return (
@@ -65,61 +84,20 @@ def reset_all():
         {}  # data_state (gr.State) reset with raw dict
     )
 
-# ----- Chatbot setup -----
-# Use a text2text-generation pipeline with a chat-capable model
-chat_model_name = "facebook/blenderbot-400M-distill"  # You can replace this with any chat-capable model
-chat_generator = pipeline("text2text-generation", model=chat_model_name)
-
-# We'll maintain chat history as a list of strings (user and bot messages)
-chat_history = []
-
-def chat_with_bot(user_message, history):
-    global chat_history
-    if history is None:
-        history = []
-        chat_history = []
-
-    # Append user message to history
-    chat_history.append(f"User: {user_message}")
-
-    # Create prompt by joining conversation history with a Bot prompt
-    prompt = "\n".join(chat_history) + "\nBot:"
-
-    # Generate response (adjust max_length and other params as needed)
-    response = chat_generator(prompt, max_length=100, do_sample=True)[0]['generated_text']
-
-    # Extract bot reply by removing prompt prefix if present
-    bot_reply = response[len(prompt):].strip()
-    if not bot_reply:
-        bot_reply = response.strip()  # fallback if slicing fails
-
-    # Append bot reply to history
-    chat_history.append(f"Bot: {bot_reply}")
-
-    # Update Gradio chat history (list of (user, bot) tuples)
-    history.append((user_message, bot_reply))
-
-    return history, history
-
-def clear_chat():
-    global chat_history
-    chat_history = []
-    return [], []
-
 # ----- UI -----
 with gr.Blocks(title="RetailGenie") as demo:
-    gr.Markdown("# 🧞‍♂️ RetailGenie – In-Store Smart Assistant")
+    gr.Markdown("# 🧞‍♂ RetailGenie – In-Store Smart Assistant")
 
     with gr.Tabs():
         with gr.TabItem("🧭 Navigator"):
             with gr.Row():
                 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)
+                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)
+                category = gr.Dropdown(label="🛍 Category", choices=[], interactive=True)
                 product = gr.Dropdown(label="📦 Product", choices=[], interactive=True)
-                brand = gr.Dropdown(label="🏷️ Brand", 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)
@@ -173,11 +151,14 @@ with gr.Blocks(title="RetailGenie") as demo:
 
         with gr.TabItem("🎁 Smart Suggestions"):
             gr.Markdown("### 🤖 Ask RetailGenie for Recommendations")
-            chatbot_ui = gr.Chatbot(label="RetailGenie Chatbot")
-            msg = gr.Textbox(show_label=False, placeholder="Type your message here and press Enter")
-            clear = gr.Button("Clear Chat")
-
-            msg.submit(chat_with_bot, inputs=[msg, chatbot_ui], outputs=[chatbot_ui, chatbot_ui])
-            clear.click(clear_chat, inputs=[], outputs=[chatbot_ui, chatbot_ui])
+            suggestion_input = gr.Textbox(label="Ask something like:", placeholder="Gift items under 500", lines=1)
+            suggest_btn = gr.Button("💡 Get Suggestions")
+            suggestions_output = gr.Textbox(label="📝 Suggestions", lines=10)
+
+            suggest_btn.click(
+                suggest_items,
+                inputs=suggestion_input,
+                outputs=suggestions_output
+            )
 
-demo.launch()
+demo.launch()