from flask import Flask, request, redirect, url_for
import os

UPLOAD_FOLDER = '/var/www/downloads'  # same folder as before
ALLOWED_EXTENSIONS = set(['exe', 'txt', 'png', 'jpg', 'zip'])  # optional filter

app = Flask(__name__)
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER

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

@app.route('/upload', methods=['POST'])
def upload_file():
    if 'file' not in request.files:
        return 'No file part', 400
    file = request.files['file']
    if file.filename == '':
        return 'No selected file', 400
    if file and allowed_file(file.filename):
        filepath = os.path.join(app.config['UPLOAD_FOLDER'], file.filename)
        file.save(filepath)
        return f'File uploaded successfully: {file.filename}', 200
    return 'File type not allowed', 400

if __name__ == "__main__":
    app.run(host='0.0.0.0', port=5000)