Update app.py

app.py

@@ -1,10 +1,19 @@
 import pandas as pd
 from itertools import permutations
-from flask import Flask, jsonify, request
-from flask_cors import CORS
+from fastapi import FastAPI, File, UploadFile, Form, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
+from typing import List
 
-app = Flask(__name__)
-CORS(app)
+app = FastAPI()
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Allows all origins
+    allow_credentials=True,
+    allow_methods=["*"],  # Allows all methods
+    allow_headers=["*"],  # Allows all headers
+)
 
 class Box:
     def __init__(self, length, width, height, quantity, box_type):
@@ -103,19 +112,23 @@ def pack_boxes(truck, boxes):
 
     return packed_positions
 
-@app.route('/upload', methods=['POST'])
-def upload_file():
-    file = request.files['file']
+@app.post("/upload/")
+async def upload_file(
+    file: UploadFile = File(...),
+    length: float = Form(...),
+    width: float = Form(...),
+    height: float = Form(...),
+):
     if not file:
-        return jsonify({"error": "No file uploaded"}), 400
+        raise HTTPException(status_code=400, detail="No file uploaded")
     
     ext = file.filename.split('.')[-1].lower()
     if ext == 'csv':
-        data = pd.read_csv(file)
+        data = pd.read_csv(file.file)
     elif ext in ['xls', 'xlsx']:
-        data = pd.read_excel(file)
+        data = pd.read_excel(file.file)
     else:
-        return jsonify({"error": "Unsupported file format"}), 400
+        raise HTTPException(status_code=400, detail="Unsupported file format")
     
     # Convert dimensions from CM to inches
     data['PieceLength'] = data['PieceLength'] / 2.54
@@ -129,9 +142,9 @@ def upload_file():
     ]
 
     # Convert truck dimensions from feet to inches
-    truck_length = float(request.form['length']) * 12  # Convert to inches
-    truck_width = float(request.form['width']) * 12    # Convert to inches
-    truck_height = float(request.form['height']) * 12  # Convert to inches
+    truck_length = length * 12  # Convert to inches
+    truck_width = width * 12    # Convert to inches
+    truck_height = height * 12  # Convert to inches
 
     truck = Truck(truck_length, truck_width, truck_height)
 
@@ -151,7 +164,4 @@ def upload_file():
     ]
     print(f"quantity {[box_data[i]['quantity'] for i in range(len(box_data))]}")
 
-    return jsonify(box_data)
-
-if __name__ == '__main__':
-    app.run(debug=True , host = "0.0.0.0" , port=8888)
+    return {"boxes": box_data}