Update Background-Remover-Flask-App-using-Python-main/app.py

Background-Remover-Flask-App-using-Python-main/app.py

@@ -1,12 +1,11 @@
-import cv2
+import streamlit as st
 import os
 from rembg import remove
 from PIL import Image
 from werkzeug.utils import secure_filename
-from flask import Flask,request,render_template
 
 UPLOAD_FOLDER = 'static/uploads'
-ALLOWED_EXTENSIONS = set(['png', 'jpg', 'jpeg','webp'])
+ALLOWED_EXTENSIONS = set(['png', 'jpg', 'jpeg', 'webp'])
 
 if 'static' not in os.listdir('.'):
     os.mkdir('static')
@@ -14,34 +13,30 @@ if 'static' not in os.listdir('.'):
 if 'uploads' not in os.listdir('static/'):
     os.mkdir('static/uploads')
 
-app = Flask(__name__)
-app.config['SEND_FILE_MAX_AGE_DEFAULT'] = 0
-app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
-app.secret_key = "secret key"
-
 def allowed_file(filename):
     return '.' in filename and filename.rsplit('.', 1)[1] in ALLOWED_EXTENSIONS
 
-def remove_background(input_path,output_path):
-    input = Image.open(input_path)
-    output = remove(input)
-    output.save(output_path)
+def remove_background(input_path, output_path):
+    input_image = Image.open(input_path)
+    output_image = remove(input_image)
+    output_image.save(output_path)
+
+def main():
+    st.title("Remove Background")
 
+    uploaded_file = st.file_uploader("Choose a file", type=['png', 'jpg', 'jpeg', 'webp'])
 
-@app.route('/')
-def home():
-    return render_template('home.html')
+    if uploaded_file is not None and allowed_file(uploaded_file.name):
+        filename = secure_filename(uploaded_file.name)
+        file_path = os.path.join(UPLOAD_FOLDER, filename)
+        uploaded_file.save(file_path)
 
-@app.route('/remback',methods=['POST'])
-def remback():
-    file = request.files['file']
-    if file and allowed_file(file.filename):
-        filename = secure_filename(file.filename)
-        file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
-        rembg_img_name = filename.split('.')[0]+"_rembg.png"
-        remove_background(UPLOAD_FOLDER+'/'+filename,UPLOAD_FOLDER+'/'+rembg_img_name)
-        return render_template('home.html',org_img_name=filename,rembg_img_name=rembg_img_name)
+        rembg_img_name = filename.split('.')[0] + "_rembg.png"
+        remove_background(file_path, os.path.join(UPLOAD_FOLDER, rembg_img_name))
 
+        st.image(Image.open(file_path), caption='Original Image', use_column_width=True)
+        st.image(Image.open(os.path.join(UPLOAD_FOLDER, rembg_img_name)),
+                 caption='Image with Background Removed', use_column_width=True)
 
 if __name__ == '__main__':
-    app.run(debug=True)
+    main()