from flask import Flask, render_template, request, send_file, redirect, url_for, flash
import os
from werkzeug.utils import secure_filename
import corelate

app = Flask(__name__)
app.secret_key = os.urandom(24)

UPLOAD_FOLDER = os.path.dirname(os.path.abspath(__file__))
ALLOWED_EXTENSIONS = {'txt','csv'}

app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024

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

@app.route('/')
def index():
    report1_exists = os.path.exists(corelate.outputfile1)
    report2_exists = os.path.exists(corelate.outputfile2)
    errors_exist = os.path.exists(corelate.errorfile)
    
    report1_data = []
    report2_data = []
    error_data = []
    
    if report1_exists:
        with open(corelate.outputfile1, 'r') as f:
            for line in f:
                parts = line.strip().split('\t')
                if len(parts) >= 3:
                    report1_data.append({
                        'zipcode': parts[0],
                        'customers': parts[1],
                        'distances': parts[2]
                    })
    
    if report2_exists:
        with open(corelate.outputfile2, 'r') as f:
            for line in f:
                parts = line.strip().split('\t')
                if len(parts) >= 2:
                    report2_data.append({
                        'distance': parts[0],
                        'customers': parts[1]
                    })
    
    if errors_exist:
        with open(corelate.errorfile, 'r') as f:
            error_data = f.readlines()
    
    return render_template('index.html', 
                           report1_data=report1_data,
                           report2_data=report2_data,
                           error_data=error_data,
                           report1_exists=report1_exists,
                           report2_exists=report2_exists,
                           errors_exist=errors_exist)

@app.route('/run-script', methods=['GET'])
def run_script():
    try:
        corelate.process_data()
        flash('Script executed successfully!', 'success')
    except Exception as e:
        flash(f'Error running script: {str(e)}', 'error')
    
    return redirect(url_for('index'))

@app.route('/download/<report_type>')
def download_file(report_type):
    if report_type == 'report1':
        return send_file(corelate.outputfile1, as_attachment=True)
    elif report_type == 'report2':
        return send_file(corelate.outputfile2, as_attachment=True)
    elif report_type == 'errors':
        return send_file(corelate.errorfile, as_attachment=True)
    else:
        flash('Invalid report type requested', 'error')
        return redirect(url_for('index'))

@app.route('/upload', methods=['POST'])
def upload_file():
    if 'file' not in request.files:
        flash('No file part', 'error')
        return redirect(url_for('index'))
    
    file = request.files['file']
    file_type = request.form.get('file_type')
    
    if file.filename == '':
        flash('No file selected', 'error')
        return redirect(url_for('index'))
    
    if file and allowed_file(file.filename):
        filename = secure_filename(file.filename)
        
        if file_type == 'zipcodedata':
            target_file = corelate.statedatafile
        elif file_type == 'ziplist':
            target_file = corelate.ziplist
        elif file_type == 'servicecenters':
            target_file = corelate.servicecenters
        else:
            flash('Invalid file type', 'error')
            return redirect(url_for('index'))
        
        file.save(os.path.join(app.config['UPLOAD_FOLDER'], target_file))
        flash(f'File successfully uploaded and replaced {target_file}', 'success')
    else:
        flash('File type not allowed. Please upload .txt files only.', 'error')
    
    return redirect(url_for('index'))

if __name__ == "__main__":
    import os
    port = int(os.environ.get("PORT", "7860"))
    app.run(host="0.0.0.0", port=port, debug=False)