el/backgroundremover
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Initial commit

Browse Source
This commit is contained in:
Leonard Excoffier
2024-05-01 12:31:54 -04:00
commit 333b2cf3a9
4 changed files with 71 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
Dockerfile Normal file
View File

@@ -0,0 +1,13 @@
FROM python:latest
WORKDIR /app
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt
COPY . .
EXPOSE 5124
CMD ["python", "app.py"]

43
app.py Normal file
View File

@@ -0,0 +1,43 @@
from flask import Flask, render_template, request, send_file, flash, redirect, url_for
from rembg import remove
from PIL import Image
from io import BytesIO
app = Flask(__name__)
# Update allowed extensions to include additional file types
ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif', 'bmp', 'tiff', 'webp'}
# Set max content length to 1GB (expressed in bytes)
app.config['MAX_CONTENT_LENGTH'] = 1 * 1024 * 1024 * 1024 # 1 GB limit
def file_allowed(filename):
"""
Check if the file has an allowed extension.
"""
return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
@app.route('/', methods=['GET', 'POST'])
def upload_file():
"""
Handle file upload via POST, process image to remove background, and return the processed image.
"""
if request.method == 'POST':
file = request.files.get('file', None)
if file and file_allowed(file.filename):
try:
input_image = Image.open(file.stream)
output_image = remove(input_image, post_process_mask=True)
img_io = BytesIO()
output_image.save(img_io, 'PNG')
img_io.seek(0)
return send_file(img_io, mimetype='image/png', as_attachment=True, download_name='processed_image.png')
except Exception as e:
flash('Failed to process the image. Error: {}'.format(e), 'error')
return redirect(url_for('upload_file'))
else:
flash('Please upload a valid image file (PNG, JPG, JPEG, GIF, BMP, TIFF, WEBP).', 'error')
return redirect(url_for('upload_file'))
return render_template('index.html')
if __name__ == '__main__':
app.run(host='0.0.0.0', debug=True, port=5124)

3
requirements.txt Normal file
View File

@@ -0,0 +1,3 @@
flask
rembg
pillow

12
templates/index.html Normal file
View File

@@ -0,0 +1,12 @@
<!DOCTYPE html>
<html>
<head>
<title>Background Remover</title>
</head>
<body>
<form action="/" method="post" enctype="multipart/form-data">
<input type="file" name="file" required>
<input type="submit" value="Upload and Remove Background">
</form>
</body>
</html>