import os
from flask import Flask, flash, request, redirect, url_for, session
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'}

app = Flask(__name__)
app.config["SESSION_PERMANENT"] = False
app.config["SESSION_TYPE"] = "null"
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER

Session(app)
 
def allowed_file(filename):
    return '.' in filename and \
           filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS

@app.route('/rm-bg/<transparent>', methods=['GET', 'POST'])
def upload_file(transparent):
    transparent = True if transparent == "true" else False
    if request.method == 'POST':
        if 'file' not in request.files:
            flash('No file part')
            return {"status": "Failed", "message": "Please Provide file name(file)."}
        file = request.files['file']
        if file.filename == '':
            flash('No selected file')
            return  {"status": "Failed", "message": "Filename Not Found."}
        if file and allowed_file(file.filename):
            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"
    }

if __name__ == '__main__':
    # http_server = WSGIServer(('', 8000), app)
    # http_server.serve_forever()
    app.debug = True
    app.run(port=8000)