Update flask_app.py

flask_app.py

@@ -5,7 +5,7 @@ from flask_session import Session
 from werkzeug.utils import secure_filename
 from example_inference import example_inference
 from flask import send_from_directory
-
+import base64
 
 UPLOAD_FOLDER = 'images'
 ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg'}
@@ -23,9 +23,7 @@ def allowed_file(filename):
 
 @app.route('/rm-bg/<transparent>', methods=['GET', 'POST'])
 def upload_file(transparent):
-    print(transparent)
     transparent = True if transparent == "true" else False
-    print(transparent)
     if request.method == 'POST':
         if 'file' not in request.files:
             flash('No file part')
@@ -35,11 +33,14 @@ def upload_file(transparent):
             flash('No selected file')
             return  {"status": "Failed", "message": "Filename Not Found."}
         if file and allowed_file(file.filename):
-            filename = secure_filename('normal_image.png')
-            file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
-            im_path = f"{os.path.dirname(os.path.abspath(__file__))}/images/{filename}"
-            rm_image_path = example_inference(im_path, transparent)
-            return send_from_directory(app.config["UPLOAD_FOLDER"], rm_image_path)
+            img_data = file.read()
+            img_base64 = base64.b64encode(img_data)
+            if isinstance(img_base64, bytes):
+                img_base64 = img_base64.decode('utf-8')
+                print('yes')
+            image_bytes = base64.b64decode(img_base64)
+            rm_image_path = example_inference(image_bytes, transparent)
+            return 'data:image/png;base64,' + rm_image_path
     return {
         "message": "Get Request not allowed"
     }