Update app.py

app.py

@@ -8,7 +8,8 @@ import os
 import xgboost as xgb
 import logging
 
-app = Flask(__name__, static_folder=".", static_url_path="")
+# === Flask Setup ===
+app = Flask(__name__, static_folder='.', static_url_path='')
 CORS(app)
 logging.basicConfig(level=logging.INFO)
 
@@ -29,61 +30,55 @@ 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)
 
-# === Serve Frontend ===
+# === Serve index.html ===
 @app.route('/')
 def serve_index():
-    return send_from_directory('.', 'index.html')
+    return app.send_static_file('index.html')
 
+# === Serve static assets (if any JS, CSS, images) ===
 @app.route('/<path:path>')
-def serve_static(path):
+def serve_static_files(path):
     return send_from_directory('.', path)
 
-# === Calculate tile requirement ===
+# === Calculate tiles endpoint ===
 @app.route('/calculate', methods=['POST'])
 def calculate():
     try:
         data = request.get_json()
-        required_fields = ['tile_type', 'length', 'width', 'tile_size']
-        for field in required_fields:
+        for field in ['tile_type', 'length', 'width', 'tile_size']:
             if field not in data:
                 return jsonify({"error": f"Missing field: {field}"}), 400
 
         tile_type = data['tile_type'].lower()
-        if tile_type not in ['floor', 'wall']:
-            return jsonify({"error": "Invalid tile type"}), 400
-
-        tile_size = data['tile_size']
-        if tile_size not in tile_sizes:
-            return jsonify({"error": "Invalid tile size"}), 400
-
-        # Validate and compute area
         length = float(data['length'])
         width = float(data['width'])
         validate_positive_number(length, "length")
         validate_positive_number(width, "width")
-        area = length * width
-        validate_positive_number(area, "area")
 
-        # Calculate tile needs
-        tile_info = tile_sizes[tile_size]
+        if tile_type not in ['floor', 'wall']:
+            return jsonify({"error": "Invalid tile type"}), 400
+        if data['tile_size'] not in tile_sizes:
+            return jsonify({"error": "Invalid tile size"}), 400
+
+        area = length * width
+        tile_info = tile_sizes[data['tile_size']]
         area_per_tile = tile_info['length'] * tile_info['width']
         tiles_needed = math.ceil((area / area_per_tile) * 1.1)
         tiles_per_box = tile_info.get('tiles_per_box', 10)
         boxes_needed = math.ceil(tiles_needed / tiles_per_box)
 
-        # Find matching products
         matching_products = [
             {
                 **p,
                 "link": p.get("url", "#")
             }
             for p in tile_catalog
-            if p['type'].lower() == tile_type and p['size'] == tile_size
+            if p['type'].lower() == tile_type and p['size'] == data['tile_size']
         ]
 
         return jsonify({
             "tile_type": tile_type,
-            "tile_size": tile_size,
+            "tile_size": data['tile_size'],
             "length": round(length, 2),
             "width": round(width, 2),
             "area": round(area, 2),
@@ -97,7 +92,7 @@ def calculate():
         app.logger.error(f"Error in /calculate: {str(e)}")
         return jsonify({"error": "Internal server error"}), 500
 
-# === Recommendation endpoint ===
+# === Recommend endpoint ===
 @app.route('/recommend', methods=['POST'])
 def recommend():
     try:
@@ -108,9 +103,6 @@ def recommend():
                 return jsonify({"error": f"Missing field: {field}"}), 400
 
         tile_type = data['tile_type'].lower()
-        if tile_type not in ['floor', 'wall']:
-            return jsonify({"error": "Invalid tile type"}), 400
-
         length = float(data['length'])
         width = float(data['width'])
         validate_positive_number(length, "length")
@@ -123,7 +115,6 @@ def recommend():
         if not isinstance(data['price_range'], list) or len(data['price_range']) != 2:
             return jsonify({"error": "Invalid price range"}), 400
 
-        # Prepare features
         features = prepare_features({
             **data,
             "area": area
@@ -151,11 +142,12 @@ def recommend():
         app.logger.error(f"Error in /recommend: {str(e)}")
         return jsonify({"error": "Internal server error"}), 500
 
-# === Utility Functions ===
+# === Helper Functions ===
 def prepare_features(data):
     tile_type_num = 0 if data['tile_type'] == 'floor' else 1
     price_per_sqft = data['price_range'][1] / data['coverage']
     budget_efficiency = data['coverage'] / data['price_range'][1]
+
     return np.array([[tile_type_num, data['area'], data['coverage'],
                       data['price_range'][0], data['price_range'][1],
                       price_per_sqft, budget_efficiency]])
@@ -196,6 +188,6 @@ def validate_positive_number(value, field):
     if not isinstance(value, (int, float)) or value <= 0:
         raise ValueError(f"{field} must be a positive number")
 
-# === Run the App ===
+# === Start the server ===
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860, debug=False)