Update app.py

app.py

@@ -221,13 +221,23 @@ async def upscale_image(image_bytes: bytes, filename: str) -> Optional[str]:
 
 @app.route('/upscale', methods=['POST'])
 async def upscale_endpoint():
-    if 'file' not in request.files:
+    # Check for file in multipart/form-data
+    if 'file' in request.files:
+        file = request.files['file']
+        if file.filename == '':
+            print("File filename is empty")
+            return jsonify({"error": "No file selected"}), 400
+        filename = secure_filename(file.filename)
+        image_bytes = file.read()
+    # Fallback for raw binary data
+    elif request.content_type and 'application/octet-stream' in request.content_type:
+        image_bytes = request.get_data()
+        filename = f"uploaded_{generate_uuid()}.bin"  # Default filename for raw binary
+    else:
+        print("No file part in request.files and no raw binary data")
         return jsonify({"error": "No file provided"}), 400
-    file = request.files['file']
-    if file.filename == '':
-        return jsonify({"error": "No file selected"}), 400
-    filename = secure_filename(file.filename)
-    image_bytes = file.read()
+
+    print(f"Received file: {filename}, size: {len(image_bytes)} bytes")
     upscaled_url = await upscale_image(image_bytes, filename)
     if upscaled_url:
         return jsonify({"upscaled_url": upscaled_url}), 200