application.py

import os
import urllib.request
from flask import Flask, flash, request, redirect, render_template, send_file, send_from_directory
from werkzeug.utils import secure_filename
import pandas as pd
from output_printer_final import print_result
import gurobipy as gb 




app = Flask(__name__)
UPLOAD_FOLDER = 'uploads/'
basedir = os.path.abspath(os.path.dirname(__file__))
app.secret_key = "secret key"
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
file_dir = os.path.join(basedir, app.config['UPLOAD_FOLDER'])
if not os.path.exists(file_dir):
    os.makedirs(file_dir)


ALLOWED_EXTENSIONS = set(['csv'])

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

def generateresult(filenames, file_dir):
	""" filename = filenames[0]
	filepath=os.path.join(file_dir, filename)
	a=pd.read_csv(filepath, header=None)
	b=2*a """
	# filedownloadpath=os.path.join(file_dir, 'download-'+filename.split('.')[0]+'.xlsx')
	print_result(file_dir, filenames)
	# return(b)




@app.route('/')
def upload_form():
	return render_template('upload_files.html')



@app.route('/', methods=['POST'])
def upload_file():
	if request.method == 'POST':
		filenames = []
		if 'file1' not in request.files:
			flash('No file part')
			return redirect(request.url)
		file = request.files['file1']
		if file.filename == '':
			flash('No file selected for uploading')
			return redirect(request.url)
		if file and allowed_file(file.filename):
			filename = secure_filename(file.filename)
			filenames.append(filename)
			file.save(os.path.join(file_dir, filename))
			# downloadfilename = file_dir + '\download-file.csv'
			# b.to_csv(os.path.join(file_dir, downloadfilename))
			# return render_template('simple.html',  tables=[b.to_html(classes='data')], titles=b.columns.values) #redirect('/download/<downloadfilename>')
		else:
			flash('Allowed file types csv')
			return redirect(request.url)

		file = request.files['file2']
		if file.filename == '':
			flash('No file selected for uploading')
			return redirect(request.url)
		if file and allowed_file(file.filename):
			filename = secure_filename(file.filename)
			filenames.append(filename)
			file.save(os.path.join(file_dir, filename))
		else:
			flash('Allowed file types csv')
			return redirect(request.url)

		file = request.files['file3']
		if file.filename == '':
			flash('No file selected for uploading')
			return redirect(request.url)
		if file and allowed_file(file.filename):
			filename = secure_filename(file.filename)
			filenames.append(filename)
			file.save(os.path.join(file_dir, filename))
		else:
			flash('Allowed file types csv')
			return redirect(request.url)
		
		file = request.files['file4']
		if file.filename == '':
			flash('No file selected for uploading')
			return redirect(request.url)
		if file and allowed_file(file.filename):
			filename = secure_filename(file.filename)
			filenames.append(filename)
			file.save(os.path.join(file_dir, filename))
		else:
			flash('Allowed file types csv')
			return redirect(request.url)
			
		flash('Files successfully uploaded, the page will direct to the result soon!')
		""" b = generateresult(filenames, file_dir)
		downloadfilename = file_dir + '\model-reopen-solution.xlsx'
		b.to_csv(os.path.join(file_dir, downloadfilename)) """
		generateresult(filenames, file_dir)
		redirect('/')
		return render_template('download.html')


@app.route("/download/<filename>", methods=['GET'])
def download_file(filename):
    directory='uploads'
    try:
        return send_from_directory(directory, filename=filename, as_attachment=True)
    except FileNotFoundError:
        abort(404)