Update app.py

app.py

@@ -1,66 +1,52 @@
 import pandas as pd
 import gradio as gr
 
-# Load inventory
+# Load data
 df = pd.read_csv("inventory.csv")
 
-# Extract category list
+# Get category list
 categories = sorted(df['Category'].dropna().unique())
 
-# Step 1: Ask for category
-def ask_category():
-    return "Please select a product category:", gr.Dropdown(choices=categories, label="Product Category"), None
-
-# Step 2: Based on category, ask for product
-def ask_product(category):
+# Product selector from category
+def get_products(category):
     products = df[df['Category'] == category]['Product Name'].unique().tolist()
-    return (
-        f"Now select a product from {category}:",
-        gr.update(choices=products, visible=True),  # ✅ Fix applied here
-        category
-    )
+    return gr.update(choices=products, visible=True), category
 
-# Step 3: Show product info
-def show_result(product, category):
+# 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]
-        if row["In Stock"].strip().lower() == "yes":
-            return f"✅ Your **{product}** is in **{row['Aisle']}**. Quantity available: **{row['Quantity']}**, Price: ₹{row['Price (rupees)']}."
+        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)']}."
         else:
-            return f"❌ Sorry, **{product}** is currently **not available** in stock."
+            return f"❌ Sorry, {product} is currently not available in stock."
     else:
-        return "Product not found."
+        return "⚠️ Product not found."
 
-# Gradio interface with state
-with gr.Blocks(title="🧞 RetailGenie Guided Assistant") as demo:
-    gr.Markdown("## 🛍️ RetailGenie – Smart In-Store Assistant")
-    gr.Markdown("Start by selecting a product category:")
+# Clear function
+def reset():
+    return gr.update(value=None), gr.update(choices=[], visible=False), "", ""
 
-    state_category = gr.State()
-    
-    # Step 1
-    cat_btn = gr.Button("Select Category")
-    cat_out = gr.Textbox(label="Message")
-    cat_dropdown = gr.Dropdown(choices=categories, visible=False)
-    
-    # Step 2
-    prod_out = gr.Textbox(label="Message")
-    prod_dropdown = gr.Dropdown(visible=False)
-    
-    # Step 3
-    final_out = gr.Textbox(label="Result")
+# 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.")
 
-    # Step 1 callback
-    def show_cat_options():
-        return "Please choose a product category:", gr.update(visible=True), None
-    cat_btn.click(show_cat_options, outputs=[cat_out, cat_dropdown, state_category])
+    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")
 
-    # Step 2 callback
-    cat_dropdown.change(ask_product, inputs=cat_dropdown, outputs=[prod_out, prod_dropdown, state_category])
+        with gr.Column():
+            message = gr.Textbox(label="💬 Assistant Response", interactive=False)
+
+    # Logic chaining
+    state_category = gr.State()
 
-    # Step 3 callback
-    prod_dropdown.change(show_result, inputs=[prod_dropdown, state_category], outputs=final_out)
+    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])
 
 demo.launch()