Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 import joblib
-import numpy as np
 import json
 import math
 import re
@@ -12,14 +11,12 @@ import xgboost as xgb
 
 warnings.filterwarnings("ignore")
 
-# Load XGBoost model (if used in future extensions)
+# Load models
 xgb_model = xgb.Booster()
 xgb_model.load_model("xgb_model.json")
-
-# Load Random Forest model
 rf = joblib.load("rf_model.pkl")
 
-# Load tile catalog and sizes
+# Load product catalog and tile sizes
 with open("tile_catalog.json", "r", encoding="utf-8") as f:
     tile_catalog = json.load(f)
 
@@ -52,23 +49,24 @@ def create_pdf(text):
 
 def extract_tile_area(msg, unit):
     msg = msg.lower().replace("feet", "").replace("ft", "").replace("mm", "").strip()
-    msg = re.sub(r"\s*(x|×|\*|into)\s*", "x", msg)  # normalize all separators to 'x'
+    msg = re.sub(r"\s*(x|×|\*|into)\s*", "x", msg)  # Normalize separators to 'x'
+    parts = msg.split("x")
 
-    if "x" in msg:
-        try:
-            val1, val2 = map(lambda v: float(re.sub(r"[^\d.]", "", v)), msg.split("x"))
+    try:
+        if len(parts) == 2:
+            val1 = float(re.sub(r"[^\d.]", "", parts[0]))
+            val2 = float(re.sub(r"[^\d.]", "", parts[1]))
             if unit == "mm":
                 sqft = (val1 * val2) / 92903.04
             else:
                 sqft = val1 * val2
             return round(sqft, 2)
-        except:
-            return None
-    else:
-        try:
-            return float(re.sub(r"[^\d.]", "", msg))  # single number case
-        except:
+        elif len(parts) == 1 and parts[0]:
+            return float(re.sub(r"[^\d.]", "", parts[0]))
+        else:
             return None
+    except:
+        return None
 
 def chat_fn(message, history, user_state={}):
     if message.strip().lower() in ["floor", "wall"]:
@@ -114,15 +112,15 @@ def chat_fn(message, history, user_state={}):
 
     if user_state["step"] == "get_tile_size":
         area = extract_tile_area(message, user_state["unit"])
-        if area is None:
-            return reply("I couldn’t understand that size. Try something like 600 x 600, 2 x 2, or just 4."), None, user_state
+        if not area or area <= 0:
+            return reply("❌ That tile size didn’t work. Try something like:\n- `600 x 600`\n- `2 x 2`\n- `4` (if in ft)"), None, user_state
 
         user_state["tile_area"] = area
         user_state["step"] = "done"
 
         area_needed = user_state["area"]
         tile_type = user_state["tile_type"]
-        tile_needed = math.ceil((area_needed / area) * 1.1)  # 10% buffer
+        tile_needed = math.ceil((area_needed / area) * 1.1)
 
         best = []
         for tile in tile_catalog:
@@ -171,7 +169,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", radius_size="lg")) as de
 
         👉 Start by typing **"Floor"** or **"Wall"**  
         📏 Then give me the **area** and **tile size** in `mm` or `ft`  
-        ✏️ Example: `600 x 600`, `2 x 2`, or just `4` sq.ft  
+        ✏️ Example: `600 x 600`, `2 x 2`, `4`  
         📋 I’ll recommend tiles and generate a **PDF Report**!
         """,
         elem_id="header"