Update app.py

app.py

@@ -3,8 +3,6 @@ import joblib
 import json
 import math
 import re
-from fpdf import FPDF
-import tempfile
 from deep_translator import GoogleTranslator
 import warnings
 import xgboost as xgb
@@ -23,7 +21,7 @@ with open("tile_catalog.json", "r", encoding="utf-8") as 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]
+tile_size_map = {s["label"].lower().replace(" ", ""): s["area_sqft"] for s in tile_sizes}
 
 def translate(text, lang="en"):
     try:
@@ -31,22 +29,6 @@ def translate(text, lang="en"):
     except:
         return text
 
-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, unit):
     msg = msg.lower().replace("feet", "").replace("ft", "").replace("mm", "").strip()
     msg = re.sub(r"\s*(x|×|\*|into)\s*", "x", msg)
@@ -66,11 +48,10 @@ def extract_tile_area(msg, unit):
         return None
 
 def chat_fn(message, history, user_state={}):
-    # Only reset if not mid-conversation
     if "step" not in user_state and message.strip().lower() in ["floor", "wall"]:
         user_state.clear()
 
-    # Detect language
+    # Language detection
     if "lang" not in user_state:
         try:
             user_state["lang"] = GoogleTranslator(source="auto", target="en").detect(message)
@@ -79,32 +60,31 @@ def chat_fn(message, history, user_state={}):
     lang = user_state["lang"]
     def reply(text): return translate(text, lang)
 
-    # Handle PDF request
-    if message.strip().lower() in ["pdf", "report", "download"]:
-        if "summary" in user_state:
-            pdf_path = create_pdf(user_state["summary"])
-            return reply("Here’s your PDF report 📄"), [pdf_path], user_state
-        else:
-            return reply("No estimate yet. Please start by typing 'Floor' or 'Wall'."), None, user_state
-
-    # Begin conversation
+    # Start conversation
     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 chose {user_state['tile_type']} tiles.\nWhat’s the total area to cover (in sq.ft)?"), None, user_state
-        return reply("Hi there! Are you planning for Floor or Wall tiles?"), None, user_state
+            user_state["step"] = "get_length"
+            return reply(f"You selected {user_state['tile_type']} tiles.\nEnter the length of the space in feet:"), None, user_state
+        return reply("Hi! Are you planning for Floor or Wall tiles?"), None, user_state
+
+    if user_state["step"] == "get_length":
+        try:
+            user_state["length"] = float(message)
+            user_state["step"] = "get_width"
+            return reply("Now enter the width of the space in feet:"), None, user_state
+        except:
+            return reply("That doesn't seem like a number. Please enter the length in feet (e.g. 12)."), None, user_state
 
-    # Area input
-    if user_state["step"] == "get_area":
+    if user_state["step"] == "get_width":
         try:
-            user_state["area"] = float(message)
+            user_state["width"] = float(message)
+            user_state["area"] = round(user_state["length"] * user_state["width"], 2)
             user_state["step"] = "get_unit"
-            return reply("Would you like to enter the tile size in mm or ft?"), None, user_state
+            return reply("Got it! Would you like to enter tile size in mm or ft?"), None, user_state
         except:
-            return reply("That doesn't look like a number. Please enter the area in sq.ft (e.g. 120)."), None, user_state
+            return reply("Please enter a valid width in feet (e.g. 10)."), None, user_state
 
-    # Unit input
     if user_state["step"] == "get_unit":
         if message.lower() not in ["mm", "ft"]:
             return reply("Please type either mm or ft to choose your preferred unit."), None, user_state
@@ -113,7 +93,6 @@ def chat_fn(message, history, user_state={}):
         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, 2 x 2, or 4):"), None, user_state
 
-    # Tile size input
     if user_state["step"] == "get_tile_size":
         area = extract_tile_area(message, user_state["unit"])
         if not area or area <= 0:
@@ -147,34 +126,33 @@ def chat_fn(message, history, user_state={}):
 
         summary = f"""
 🧱 Tile Type: {tile_type}
-📐 Area to Cover: {area_needed} sq.ft
+📐 Space: {user_state['length']} ft x {user_state['width']} ft
+📏 Area to Cover: {area_needed} sq.ft
 🧮 Tile Size Area: {round(area, 2)} sq.ft
 🔢 Estimated Tiles Needed: {tile_needed} (with 10% buffer)
 
-🎯 Suggested Products:
+🎯 Top Suggestions:
 """
         for i, t in enumerate(top3, 1):
             summary += f"\n{i}. {t['name']} ({t['size']})\n   ₹{t['price']} per box → ~{t['boxes']} boxes\n   {t['url']}\n"
 
-        summary += "\nYou can type 'pdf' to download the report, or start fresh with 'Floor' or 'Wall'."
+        summary += "\nType 'Floor' or 'Wall' to start another estimate."
 
-        user_state["summary"] = summary
-        pdf_path = create_pdf(summary)
-        return reply(summary), [pdf_path], user_state
+        return reply(summary), None, user_state
 
     return reply("I didn't understand that. Please continue or type 'Floor' or 'Wall' to start over."), None, user_state
 
-# Gradio Interface
+# Interface
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", radius_size="lg")) 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.
+        Welcome! I'm here to help you estimate tiles for your floor or wall.
 
-        👉 Start by typing **"Floor"** or **"Wall"**  
-        📏 Then give me the **area** and **tile size** in `mm` or `ft`  
-        ✏️ Examples: `600 x 600`, `2 x 2`, `4`, `200 * 200`  
-        📋 I’ll recommend tiles and generate a **PDF Report**
+        👉 Start with **"Floor"** or **"Wall"**  
+        📐 Enter **length and width** (in feet)  
+        📏 Then your tile size (in `mm` or `ft`) – like `600 x 600`, `2x2`, or `4`  
+        📋 I’ll recommend tiles and calculate how many you need.
         """,
         elem_id="header"
     )
@@ -183,16 +161,8 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", radius_size="lg")) as de
         fn=chat_fn,
         type="messages",
         chatbot=gr.Chatbot(height=500, bubble_full_width=False),
-        additional_outputs=[gr.File(label="📄 Download Report")]
     )
 
-    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"
-    )
+    gr.Markdown("---\nNeed to start over? Just type **Floor** or **Wall** again.")
 
 demo.launch()