Update app.py

app.py

@@ -8,17 +8,18 @@ from fpdf import FPDF
 import tempfile
 from deep_translator import GoogleTranslator
 import warnings
-import xgboost as xgb  # ✅ Correct import for native XGBoost model
+import xgboost as xgb
 
 warnings.filterwarnings("ignore")
 
-# ✅ Load models
+# Load XGBoost model (native JSON format)
 xgb_model = xgb.Booster()
-xgb_model.load_model("xgb_model.json")  # ✅ Load native .json format
+xgb_model.load_model("xgb_model.json")
 
+# Load Random Forest model
 rf = joblib.load("rf_model.pkl")
 
-# ✅ Load catalog and sizes
+# Load tile catalog and sizes
 with open("tile_catalog.json", "r", encoding="utf-8") as f:
     tile_catalog = json.load(f)
 
@@ -50,29 +51,30 @@ def create_pdf(text):
     return tmp.name
 
 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 = msg.split("x")
-        if len(parts) == 2:
-            try:
-                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
+    msg = msg.lower().replace("feet", "").replace("ft", "").replace("mm", "").strip()
+    parts = re.split(r"\s*[x×*into]+\s*", msg)
+    if len(parts) == 2:
+        try:
+            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
+    elif len(parts) == 1:
+        try:
+            return float(parts[0])  # Direct sqft entry
+        except:
+            return None
     return None
 
 def chat_fn(message, history, user_state={}):
-    # Reset state if user types Floor or Wall
     if message.strip().lower() in ["floor", "wall"]:
         user_state.clear()
 
-    # Language detection
     if "lang" not in user_state:
         try:
             user_state["lang"] = GoogleTranslator(source="auto", target="en").detect(message)
@@ -81,7 +83,6 @@ def chat_fn(message, history, user_state={}):
     lang = user_state["lang"]
     def reply(text): return translate(text, lang)
 
-    # Handle PDF command
     if message.strip().lower() in ["pdf", "report", "download"]:
         if "summary" in user_state:
             pdf_path = create_pdf(user_state["summary"])
@@ -89,7 +90,6 @@ def chat_fn(message, history, user_state={}):
         else:
             return reply("No estimate yet. Please start by typing 'Floor' or 'Wall'."), None, user_state
 
-    # Start flow
     if "step" not in user_state:
         if message.lower() in ["floor", "wall"]:
             user_state["tile_type"] = message.capitalize()
@@ -111,12 +111,12 @@ def chat_fn(message, history, user_state={}):
         user_state["unit"] = message.lower()
         user_state["step"] = "get_tile_size"
         unit_label = "mm" if user_state["unit"] == "mm" else "feet"
-        return reply(f"Please enter the tile size in {unit_label} (e.g. 600 x 600):"), None, user_state
+        return reply(f"Please enter the tile size in {unit_label} (e.g. 600 x 600 or 2 x 2 or 4):"), None, 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 or 2 x 2."), None, user_state
+            return reply("I couldn’t understand that size. Try something like 600 x 600 or 2 x 2 or 4."), None, user_state
 
         user_state["tile_area"] = area
         user_state["step"] = "done"
@@ -163,20 +163,39 @@ def chat_fn(message, history, user_state={}):
 
     return reply("Type 'Floor' or 'Wall' to begin a new estimate."), None, user_state
 
-# ✅ Launch with clean intro
-with gr.Blocks() as demo:
+
+# --- Modern Gradio UI with Theme ---
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", radius_size="full")) as demo:
+    gr.Markdown(
+        """
+        # 🧱 <span style="color:#3B82F6">TileBot</span> – Smart Tile Estimator  
+        Welcome! I'm here to help you estimate tile needs for your floor or wall.
+
+        👉 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  
+        📋 I’ll recommend tiles and generate a **PDF Report**!
+        """,
+        elem_id="header"
+    )
+
     chatbot = gr.ChatInterface(
         fn=chat_fn,
-        title="🧱 TileBot – Your Tile Estimation Assistant",
-        description=(
-            "🧱 TileBot is here to help you estimate tiles for your space.\n\n"
-            "Start by telling me if you're working on a Floor or Wall. Then share your room size and tile size in mm or feet — "
-            "I'll do the rest.\n\n"
-            "I’ll calculate how many tiles you need, recommend suitable products, and give you a PDF report.\n\n"
-            "Type 'Floor' or 'Wall' to begin!"
-        ),
+        title=None,
+        description=None,
         type="messages",
-        additional_outputs=[gr.File(label="📄 Download Report")]
+        chatbot=gr.Chatbot(height=500, bubble_full_width=False),
+        additional_outputs=[gr.File(label="📄 Download Report")],
+        clear_btn="Clear",
+    )
+
+    gr.Markdown(
+        """
+        ---
+        💡 Tip: Type `"pdf"` anytime to get a downloadable report.  
+        🚀 Need a new estimate? Just say `"Floor"` or `"Wall"` to start over.
+        """,
+        elem_id="footer"
     )
 
 demo.launch()