Update app.py

app.py

@@ -5,22 +5,19 @@ import math
 import re
 from deep_translator import GoogleTranslator
 import warnings
-
 warnings.filterwarnings("ignore")
 
 # Load models
 xgb = joblib.load("xgb_model.pkl")
 rf = joblib.load("rf_model.pkl")
 
-# Load tile catalog and size mappings
+# Load tile catalog and sizes
 with open("tile_catalog.json", "r", encoding="utf-8") as f:
     tile_catalog = json.load(f)
 
 with open("tile_sizes.json", "r", encoding="utf-8") as f:
     tile_sizes = json.load(f)
 
-tile_size_map = {s["label"].lower().replace(" ", ""): s["area_sqft"] for s in tile_sizes}
-
 def translate(text, lang="en"):
     try:
         return GoogleTranslator(source="auto", target=lang).translate(text)
@@ -37,10 +34,10 @@ def extract_tile_area(msg):
             try:
                 val1 = float(re.sub(r"[^\d.]", "", parts[0]))
                 val2 = float(re.sub(r"[^\d.]", "", parts[1]))
-                if val1 > 20:  # mm
-                    sqft = (val1 * val2) / 92903.04
+                if val1 > 20:
+                    sqft = (val1 * val2) / 92903.04  # mm
                 else:
-                    sqft = val1 * val2
+                    sqft = val1 * val2  # ft
                 return round(sqft, 2)
             except:
                 return None
@@ -56,12 +53,11 @@ def extract_tile_area(msg):
     return None
 
 def chat_fn(message, history, user_state={}):
-    message_clean = message.strip().lower()
+    msg = message.strip().lower()
 
-    # Reset flow
-    if message_clean in ["restart", "reset", "start"]:
+    # Reset flow on keywords
+    if msg in ["restart", "reset", "start", "floor", "wall"]:
         user_state.clear()
-        return "🔄 Restarted. Are you looking for Floor or Wall tiles?", None, user_state
 
     if "lang" not in user_state:
         try:
@@ -71,28 +67,29 @@ def chat_fn(message, history, user_state={}):
     lang = user_state["lang"]
     def reply(text): return translate(text, lang)
 
-    # Step 1: Start
+    # Step 1: Detect tile type
     if "step" not in user_state:
-        if message_clean in ["floor", "wall"]:
+        if msg in ["floor", "wall"]:
             user_state["tile_type"] = message.capitalize()
             user_state["step"] = "get_area"
-            return reply(f"✅ You chose {user_state['tile_type']} tiles. What's the total area in sq.ft?"), None, user_state
-        return reply("👋 Hi! Are you looking for Floor or Wall tiles?"), None, user_state
+            return reply(f"✅ You selected {user_state['tile_type']} tiles. What's the total area in sq.ft?"), None, user_state
+        else:
+            return reply("👋 Welcome! Are you planning for Floor or Wall tiles? Type 'Floor' or 'Wall' to begin."), None, user_state
 
-    # Step 2: Get Area
+    # Step 2: Get area
     if user_state["step"] == "get_area":
         try:
             user_state["area"] = float(message)
             user_state["step"] = "get_tile_size"
-            return reply("📏 Enter the tile size (e.g. 2 x 2 ft, 600x600 mm, or 200 * 200):"), None, user_state
+            return reply("📏 Now enter the tile size (e.g. 2x2, 600x600 mm, 200 * 200):"), None, user_state
         except:
-            return reply("❗ Please enter a valid number for area in sq.ft (e.g. 120)."), None, user_state
+            return reply("❗ Please enter a valid number for the area (e.g. 120)."), None, user_state
 
-    # Step 3: Get Tile Size
+    # Step 3: Get tile size
     if user_state["step"] == "get_tile_size":
         area = extract_tile_area(message)
         if area is None:
-            return reply("❗ Invalid format. Try: 2 x 2, 600x600, or 1.5 * 1.5"), None, user_state
+            return reply("❗ I couldn't understand that tile size. Try: 2 x 2, 600x600 mm, 200*200."), None, user_state
 
         user_state["tile_area"] = area
         user_state["step"] = "done"
@@ -102,32 +99,32 @@ def chat_fn(message, history, user_state={}):
         tile_needed = math.ceil((total_area / area) * 1.1)
 
         # Recommendations
-        best = []
+        recommendations = []
         for tile in tile_catalog:
             if tile["type"].lower() == tile_type.lower():
                 per_box = tile["coverage"] / area
                 if per_box > 0:
                     boxes = math.ceil(tile_needed / per_box)
-                    total = boxes * tile["price"]
-                    best.append({
+                    total_price = boxes * tile["price"]
+                    recommendations.append({
                         "name": tile["name"],
                         "size": tile["size"],
                         "price": tile["price"],
                         "boxes": boxes,
-                        "total": total,
+                        "total": total_price,
                         "url": tile["url"]
                     })
 
-        best.sort(key=lambda x: x["total"])
-        top3 = best[:3]
+        recommendations.sort(key=lambda x: x["total"])
+        top3 = recommendations[:3]
 
         result = f"""
 🧱 Tile Type: {tile_type}
-📐 Area: {total_area} sq.ft
-📦 Tile Size: {round(area, 2)} sq.ft
-🔢 Tiles Needed (with 10% buffer): {tile_needed}
+📐 Area to Cover: {total_area} sq.ft
+🧮 Tile Size Area: {round(area, 2)} sq.ft
+🔢 Tiles Needed (with buffer): {tile_needed}
 
-🎯 Recommendations:
+🎯 Top Product Suggestions:
 """
         for i, t in enumerate(top3, 1):
             result += f"{i}. {t['name']} ({t['size']}) - ₹{t['price']}/box\n   ~{t['boxes']} boxes → ₹{int(t['total'])}\n   🔗 {t['url']}\n\n"
@@ -135,14 +132,14 @@ def chat_fn(message, history, user_state={}):
         result += "You can type 'Floor' or 'Wall' to estimate again."
         return reply(result), None, {}
 
-    return reply("Say 'Floor' or 'Wall' to begin a new estimate."), None, {}
+    return reply("Say 'Floor' or 'Wall' to begin."), None, {}
 
-# Gradio UI
+# Gradio interface
 with gr.Blocks() as demo:
     gr.ChatInterface(
         fn=chat_fn,
-        title="🧱 Tilo - Tile Estimator Bot",
-        description="Chat naturally to estimate tiles needed. Start by typing 'Floor' or 'Wall'.",
+        title="🧱 Tilo – Tile Estimator",
+        description="Type 'Floor' or 'Wall' to start. Then give your room size and tile size (e.g. 2x2 ft, 600x600 mm).",
         type="messages"
     )