67 lines
2.1 KiB
Python
67 lines
2.1 KiB
Python
|
|
import os
|
|
import uuid
|
|
from flask import Flask, render_template, request, send_file
|
|
import subprocess
|
|
|
|
app = Flask(__name__)
|
|
|
|
UPLOAD_FOLDER = 'uploads'
|
|
GENERATED_IMAGES_FOLDER = 'generated_images'
|
|
ALLOWED_EXTENSIONS = {'csv'}
|
|
|
|
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
|
|
app.config['GENERATED_IMAGES_FOLDER'] = GENERATED_IMAGES_FOLDER
|
|
|
|
# 確保目錄存在
|
|
os.makedirs(UPLOAD_FOLDER, exist_ok=True)
|
|
os.makedirs(GENERATED_IMAGES_FOLDER, exist_ok=True)
|
|
|
|
def allowed_file(filename):
|
|
return '.' in filename and \
|
|
filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
|
|
|
|
|
|
@app.route("/")
|
|
def index():
|
|
return render_template("index.html")
|
|
@app.route("/hello")
|
|
def hello():
|
|
return "Hello, World!"
|
|
|
|
@app.route('/upload_csv', methods=['POST'])
|
|
def upload_csv():
|
|
print("Received request to upload CSV file")
|
|
if 'csvFile' not in request.files:
|
|
return "No file part", 400
|
|
file = request.files['csvFile']
|
|
if file.filename == '':
|
|
return "No selected file", 400
|
|
if file and allowed_file(file.filename):
|
|
|
|
unique_filename = str(uuid.uuid4()) + '.csv'
|
|
csv_path = os.path.join(app.config['UPLOAD_FOLDER'], unique_filename)
|
|
file.save(csv_path)
|
|
print(f"CSV file saved to {csv_path}")
|
|
|
|
try:
|
|
# 呼叫外部 Python 腳本來生成圖片
|
|
subprocess.run([os.path.join(os.getcwd(), '.venv', 'Scripts', 'python'), 'generate_images.py', csv_path], check=True)
|
|
except subprocess.CalledProcessError as e:
|
|
print(f"Error generating images: {e}")
|
|
return "Error generating images", 500
|
|
except FileNotFoundError as e:
|
|
print(f"File not found: {e}")
|
|
return "File not found", 404
|
|
except Exception as e:
|
|
print(f"An unexpected error occurred: {e}")
|
|
return "An unexpected error occurred", 500
|
|
|
|
|
|
return "File uploaded successfully", 200
|
|
|
|
|
|
if __name__ == '__main__':
|
|
# 當在生產環境部署時，不建議使用 debug=True
|
|
# 可以指定 host='0.0.0.0' 讓外部可訪問 (在防火牆允許的情況下)
|
|
app.run(debug=True, host='127.0.0.1', port=5000) |