Update app.py

app.py

@@ -1,11 +1,33 @@
-# app.py
+from flask import Flask, jsonify, request, send_file
+from flask_cors import CORS
+from rembg import remove
+from PIL import Image
+import io
 
-import os
-import subprocess
+# Initialize the Flask app
+myapp = Flask(__name__)
+CORS(myapp)  # Enable CORS if needed
 
-if __name__ == "__main__":
-    # Run awake.py in the background
-    subprocess.Popen(["python", "wk.py"])  # Start awake.py
+@myapp.route('/')
+def home():
+    return "Welcome to the Image Background Remover!"  # Basic home response
+
+@myapp.route('/remove_background', methods=['POST'])
+def remove_background():
+    if 'image' not in request.files:
+        return jsonify({"error": "No image provided"}), 400
+
+    input_image = request.files['image'].read()  # Read the uploaded image
+    output_bytes = remove(input_image)  # Process the image with rembg
 
-    # Run the Flask app using Gunicorn
-    os.system("gunicorn -w 4 -b 0.0.0.0:5000 myapp:myapp")  # 4 worker processes
+    # Convert the output bytes back into a PIL image
+    output_image = Image.open(io.BytesIO(output_bytes))
+    img_byte_arr = io.BytesIO()
+    output_image.save(img_byte_arr, format='PNG')
+    img_byte_arr.seek(0)
+
+    return send_file(img_byte_arr, mimetype='image/png')
+
+# Add this block to make sure your app runs when called
+if __name__ == "__main__":
+    myapp.run(host='0.0.0.0', port=7860)  # Run directly if needed for testing