Update app.py

app.py

@@ -5,15 +5,14 @@ import math
 import re
 from deep_translator import GoogleTranslator
 import warnings
+from xgboost import Booster
 
 warnings.filterwarnings("ignore")
 
 # Load models
-from xgboost import Booster
 xgb = Booster()
 xgb.load_model("xgb_model.json")
-
-rf = joblib.load("rf_model.pkl")  # Needs scikit-learn
+rf = joblib.load("rf_model.pkl")  # Requires scikit-learn
 
 # Load tile catalog and sizes
 with open("tile_catalog.json", "r", encoding="utf-8") as f:
@@ -33,7 +32,7 @@ def translate(text, lang="en"):
 def extract_tile_area(msg, unit):
     msg = msg.lower().replace("×", "x").replace("into", "x").replace("*", "x")
     msg = msg.replace("mm", "").replace("ft", "").replace("feet", "").strip()
-    
+
     if "x" in msg:
         parts = re.split(r"x", msg)
         if len(parts) == 2:
@@ -48,7 +47,6 @@ def extract_tile_area(msg, unit):
             except:
                 return None
     else:
-        # Single value fallback for ft
         try:
             val = float(re.sub(r"[^\d.]", "", msg))
             return val if unit == "ft" else None
@@ -77,7 +75,7 @@ def chat_fn(message, history, user_state={}):
         user_state["step"] = "get_length"
         return reply(f"Great! You chose {user_state['tile_type']} tiles.\nPlease enter the **length** of the space in feet:"), None, user_state
 
-    # Stepwise logic
+    # Stepwise input
     if user_state.get("step") == "get_length":
         try:
             user_state["length"] = float(re.sub(r"[^\d.]", "", message))
@@ -91,7 +89,7 @@ def chat_fn(message, history, user_state={}):
             user_state["width"] = float(re.sub(r"[^\d.]", "", message))
             user_state["area"] = round(user_state["length"] * user_state["width"], 2)
             user_state["step"] = "get_unit"
-            return reply(f"Your total area is {user_state['area']} sq.ft.\nWould you like to enter tile size in **mm** or **ft**?"), None, user_state
+            return reply(f"✅ Total area is {user_state['area']} sq.ft.\nWould you like to enter tile size in **mm** or **ft**?"), None, user_state
         except:
             return reply("Please enter a valid number for width (e.g. 10):"), None, user_state
 
@@ -100,12 +98,12 @@ def chat_fn(message, history, user_state={}):
             return reply("Please type either **mm** or **ft** to choose your preferred unit."), None, user_state
         user_state["unit"] = message
         user_state["step"] = "get_tile_size"
-        return reply(f"Enter the tile size in {message.upper()} (e.g. 600x600 or 2x2):"), None, user_state
+        return reply(f"Enter the tile size in {message.upper()} (e.g. 600x600 or 2x2 or 4):"), None, user_state
 
     if user_state.get("step") == "get_tile_size":
         area_per_tile = extract_tile_area(message, user_state["unit"])
         if not area_per_tile:
-            return reply("I couldn't understand that size. Try something like `600x600`, `2x2`, or just `4` if square in ft."), None, user_state
+            return reply("I couldn't understand that size. Try something like `600x600`, `2x2`, or just `4` (for square tile in ft)."), None, user_state
 
         user_state["tile_area"] = area_per_tile
         user_state["step"] = "done"
@@ -151,18 +149,15 @@ def chat_fn(message, history, user_state={}):
 
     return reply("👋 Please type 'Floor' or 'Wall' to begin tile estimation."), None, user_state
 
-# UI setup
-with gr.Blocks() as demo:
+# Gradio UI
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.ChatInterface(
         fn=chat_fn,
         title="🧱 TileBot – Smart Tile Estimator",
         description=(
             "Plan your tile needs easily. Just type `hi` to begin or start with `Floor` or `Wall`.\n"
-            "I’ll ask for room size, tile size, and recommend the best fit tiles."
-        ),
-        theme=gr.themes.Soft(),
-        retry_btn=None,
-        undo_btn=None
+            "I'll ask for room size, tile size, and recommend the best-fit tiles for your space."
+        )
     )
 
 demo.launch()