Update app.py

app.py

@@ -1,20 +1,20 @@
 import gradio as gr
 import joblib
+import numpy as np
 import json
 import math
 import re
+from fpdf import FPDF
+import tempfile
 from deep_translator import GoogleTranslator
 from xgboost import Booster
-import warnings
-
-warnings.filterwarnings("ignore")
 
 # Load models
 xgb = Booster()
 xgb.load_model("xgb_model.json")
 rf = joblib.load("rf_model.pkl")
 
-# Load tile catalog and sizes
+# Load tile catalog and size mappings
 with open("tile_catalog.json", "r", encoding="utf-8") as f:
     tile_catalog = json.load(f)
 
@@ -29,25 +29,35 @@ def translate(text, lang="en"):
     except:
         return text
 
-def extract_tile_area(msg, unit):
-    msg = msg.lower().replace("×", "x").replace("into", "x").replace("*", "x").replace("x", "x")
-    msg = msg.replace("mm", "").replace("ft", "").replace("feet", "").strip()
-    
-    # match single value (like 2 or 4)
-    if re.fullmatch(r"\d+(\.\d+)?", msg):
-        val = float(msg)
-        return round(val * val, 2) if unit == "ft" else None
+def remove_emojis(text):
+    return re.sub(r'[^\x00-\x7F]+', '', text)
+
+def create_pdf(text):
+    pdf = FPDF()
+    pdf.add_page()
+    pdf.add_font("FreeSans", "", "FreeSans.ttf", uni=True)
+    pdf.set_font("FreeSans", size=12)
+    pdf.cell(0, 10, "Tile Estimate Report", ln=True, align="C")
+    pdf.ln(5)
+    for line in text.strip().split("\n"):
+        pdf.multi_cell(0, 10, remove_emojis(line))
+    tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
+    pdf.output(tmp.name)
+    return tmp.name
+
+def extract_tile_area(msg):
+    msg = msg.lower().replace("×", "x").replace("into", "x").replace("*", "x")
+    msg = msg.replace("ft", "").replace("feet", "").replace("mm", "").strip()
     
-    # match pattern like 2x2 or 600x600
     if "x" in msg:
-        parts = re.split(r"x", 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:
+                if val1 > 20:  # treat as mm
+                    sqft = (val1 * val2) / 92903.04  # mm² to ft²
+                else:  # treat as feet
                     sqft = val1 * val2
                 return round(sqft, 2)
             except:
@@ -55,8 +65,6 @@ def extract_tile_area(msg, unit):
     return None
 
 def chat_fn(message, history, user_state={}):
-    message = message.strip().lower()
-
     if "lang" not in user_state:
         try:
             user_state["lang"] = GoogleTranslator(source="auto", target="en").detect(message)
@@ -65,58 +73,38 @@ def chat_fn(message, history, user_state={}):
     lang = user_state["lang"]
     def reply(text): return translate(text, lang)
 
-    # Reset / greet
-    if message in ["hi", "hello", "hey", "start", "help"]:
-        user_state.clear()
-        return reply("👋 Hello! Are you planning for Floor or Wall tiles?"), None, user_state
-
-    # Start
-    if message in ["floor", "wall"]:
-        user_state.clear()
-        user_state["tile_type"] = message.capitalize()
-        user_state["step"] = "get_length"
-        return reply(f"Great! You chose {user_state['tile_type']} tiles.\nPlease enter the **length** of your space in feet:"), None, user_state
+    if "step" not in user_state:
+        if message.lower() in ["floor", "wall"]:
+            user_state["tile_type"] = message.capitalize()
+            user_state["step"] = "get_area"
+            return reply(f"✅ Great! You selected {user_state['tile_type']} tiles.\nPlease enter the total area in sq.ft:"), None, user_state
+        return reply("👋 Hello! Are you looking for Floor or Wall tiles?"), None, user_state
 
-    if user_state.get("step") == "get_length":
+    if user_state["step"] == "get_area":
         try:
-            user_state["length"] = float(re.sub(r"[^\d.]", "", message))
-            user_state["step"] = "get_width"
-            return reply("Now enter the **width** of your space in feet:"), None, user_state
+            user_state["area"] = float(message)
+            user_state["step"] = "get_tile_size"
+            return reply("📏 Now enter the tile size like 2 x 2 ft or 600x600 MM or 200 * 200:"), None, user_state
         except:
-            return reply("Please enter a valid number for length (e.g. 12.5):"), None, user_state
-
-    if user_state.get("step") == "get_width":
-        try:
-            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"✅ 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
-
-    if user_state.get("step") == "get_unit":
-        if message not in ["mm", "ft"]:
-            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, 2x2, or 4):"), None, user_state
+            return reply("❗ Please enter a number for area (e.g. 120)."), None, user_state
 
-    if user_state.get("step") == "get_tile_size":
-        tile_area = extract_tile_area(message, user_state["unit"])
-        if not tile_area:
-            return reply("❌ I couldn't understand the tile size. Try `600x600`, `2x2`, or just `4`."), None, user_state
+    if user_state["step"] == "get_tile_size":
+        area = extract_tile_area(message)
+        if area is None:
+            examples = "2 x 2 ft, 600x600 MM, 200*200, 1.5x1.5"
+            return reply(f"❗ Oops, I didn't catch that size. Try formats like: {examples}"), None, user_state
 
-        user_state["tile_area"] = tile_area
+        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 / tile_area) * 1.1)
+        tile_needed = math.ceil((area_needed / area) * 1.1)
 
         best = []
         for tile in tile_catalog:
             if tile["type"].lower() == tile_type.lower():
-                per_box = tile["coverage"] / tile_area
+                per_box = tile["coverage"] / area
                 if per_box > 0:
                     boxes = math.ceil(tile_needed / per_box)
                     total = boxes * tile["price"]
@@ -132,28 +120,32 @@ def chat_fn(message, history, user_state={}):
         best.sort(key=lambda x: x["total"])
         top3 = best[:3]
 
-        summary = f"""🧱 Tile Type: {tile_type}
-📏 Area: {area_needed} sq.ft
-🧮 Tile Size Area: {tile_area} sq.ft
-🔢 Tiles Needed: {tile_needed} (10% extra included)
+        summary = f"""
+🧱 Tile Type: {tile_type}
+📐 Area: {area_needed} sq.ft
+🧮 Tile Size: {round(area, 2)} sq.ft
+🔢 Total Tiles Needed (10% buffer): {tile_needed}
 
-🎯 Top Suggestions:
+🎯 Top Recommendations:
 """
         for i, t in enumerate(top3, 1):
-            summary += f"{i}. {t['name']} ({t['size']}) – ₹{t['price']}/box (~{t['boxes']} boxes)\n🔗 {t['url']}\n"
+            summary += f"\n{i}. {t['name']} ({t['size']})\n   ₹{t['price']}/box → ~{t['boxes']} boxes\n   {t['url']}\n"
 
-        summary += "\nType 'Floor' or 'Wall' to start again."
+        summary += "\nSay 'Floor' or 'Wall' to start a new estimate."
 
-        return reply(summary), None, user_state
+        pdf_path = create_pdf(summary)
+        return reply(summary), [pdf_path], {}
 
-    return reply("Type 'hi' or 'floor' or 'wall' to begin!"), None, user_state
+    return reply("Say 'Floor' or 'Wall' to begin again."), None, {}
 
-# UI
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.ChatInterface(
+# Launch the bot
+with gr.Blocks() as demo:
+    chatbot = gr.ChatInterface(
         fn=chat_fn,
-        title="🧱 TileBot – Smart Tile Estimator",
-        description="Type `hi`, `floor`, or `wall` to begin. I'll guide you step-by-step to estimate tile needs."
+        title="🧱 TileBot – ChatGPT-Style Tile Estimator",
+        description="Chat like a human. Enter sizes naturally like 200x200, 2 x 2 ft, or 600x600 MM.\nSay 'Floor' or 'Wall' to begin.",
+        type="messages",
+        additional_outputs=[gr.File(label="📄 Download Report")]
     )
 
-demo.launch()
+demo.launch()