import os
from flask import Flask, send_file, request, jsonify
from utils.handle_file import file_handler
from werkzeug.utils import secure_filename
app = Flask(__name__)
app.config['MAX_CONTENT_LENGTH'] = 11 * 1024 * 1024

@app.route("/")
def index():
    return send_file('src/index.html')


@app.route('/upload', methods=['POST'])
def upload():
    try:
        allowed_chars = request.args.get('allowed_size')
        if 'file' not in request.files and 'files' not in request.files:
            return jsonify({"error": "No files uploaded"}), 400
        if 'files' in request.files:
            files = request.files.getlist('files')
        else:
            files = request.files.getlist('file')
        if not files or not files[0].filename:
            return jsonify({"error": "No files selected"}), 400
        if len(files) == 1:
            return file_handler.process_file(files[0], allowed_chars)
        else:
            return file_handler.process_files(files, allowed_chars)
            
    except Exception as e:
        return jsonify({"error": f"An error occurred: {e}"}), 500

def main():
    app.run(host='0.0.0.0', port=7860, debug=True)

if __name__ == "__main__":
    main()