43 lines
1011 B
Python
43 lines
1011 B
Python
from flask import Flask, send_file, render_template, request
|
|
import os
|
|
|
|
import numpy as np
|
|
|
|
from models.multiresolution_flow_3d import *
|
|
from models.trainer import *
|
|
|
|
tm_borg = []
|
|
|
|
app = Flask(__name__)
|
|
|
|
@app.route('/')
|
|
def home():
|
|
return render_template('index.html')
|
|
|
|
@app.route('/generate', methods=['POST'])
|
|
def generate():
|
|
# Call your Python script or function here
|
|
output_file = generate_file()
|
|
|
|
# Provide the file for download
|
|
return send_file(output_file, as_attachment=True)
|
|
|
|
def generate_file():
|
|
# This is where you generate your file
|
|
# For demonstration, we will create a simple text file
|
|
output_path = 'output.txt'
|
|
with open(output_path, 'w') as f:
|
|
f.write('Hello, this is the generated file.')
|
|
return output_path
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
|
nlevel=8
|
|
shape = (2**nlevel,2**nlevel,2**nlevel)
|
|
model = multi_scale_model(nlevel=nlevel)
|
|
tm_borg = trainer(model)
|
|
tm_borg.transfer(silent=False)
|
|
|
|
app.run(debug=True)
|