from flask import Flask, request, jsonify
import os

app = Flask(__name__)

@app.route('/upload', methods=['POST'])

def upload_file():
    
    base_dir = 'wizmap-catalog'
    project_name = request.form.get('project_name')
    if not project_name:
        return jsonify({'message': 'No project name provided'}), 400
    
    UPLOAD_DIR = os.path.join(base_dir, project_name)
    os.makedirs(UPLOAD_DIR, exist_ok=True)
    app.config['UPLOAD_FILE'] = UPLOAD_DIR
    if 'grid' not in request.files or 'data' not in request.files:
        return jsonify({'message': 'No file part'}), 400
    
    grid = request.files['grid']
    data = request.files['data']
    if len(grid.read()) == 0 or len(data.read()) == 0:
        return jsonify({'message': 'Data or grid is empty.'}), 404
    grid.seek(0)
    data.seek(0)
    
    grid_filename = os.path.join(app.config['UPLOAD_FILE'], grid.filename)
    data_filename = os.path.join(app.config['UPLOAD_FILE'], data.filename)
    grid.save(grid_filename)
    data.save(data_filename)

    return jsonify({'message': 'Files uploaded successfully'}), 200

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