103 lines
3.3 KiB
Python
103 lines
3.3 KiB
Python
import os
|
||
import json
|
||
from flask import Flask, render_template, request, jsonify, send_from_directory
|
||
from node_engine import NodeEngine
|
||
from database import init_db
|
||
from config import Config
|
||
|
||
app = Flask(__name__)
|
||
app.config.from_object(Config)
|
||
|
||
# Ensure directories exist
|
||
os.makedirs('configs/nodes', exist_ok=True)
|
||
os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
|
||
os.makedirs('static/css', exist_ok=True)
|
||
os.makedirs('static/js', exist_ok=True)
|
||
|
||
# Initialize engine and DB
|
||
engine = NodeEngine()
|
||
init_db()
|
||
|
||
@app.route('/')
|
||
def index():
|
||
# Reload engine to pick up any manual JSON changes
|
||
nodes_config = engine.get_all_node_configs()
|
||
return render_template('board.html', node_configs=nodes_config)
|
||
|
||
@app.route('/api/nodes/configs')
|
||
def get_node_configs():
|
||
return jsonify(engine.get_all_node_configs())
|
||
|
||
@app.route('/api/create_node', methods=['POST'])
|
||
def create_node():
|
||
node_type_id = request.args.get('type')
|
||
x = int(request.args.get('x', 100))
|
||
y = int(request.args.get('y', 100))
|
||
|
||
# Get config
|
||
configs = engine.get_all_node_configs()
|
||
node_data = next((c for c in configs if c['id'] == node_type_id), None)
|
||
|
||
if not node_data:
|
||
return "Node type not found", 404
|
||
|
||
instance_id = f"inst_{node_type_id}_{os.urandom(2).hex()}"
|
||
return render_template('partials/node.html',
|
||
node_data=node_data,
|
||
node_instance_id=instance_id,
|
||
x=x, y=y)
|
||
|
||
@app.route('/api/upload_asset', methods=['POST'])
|
||
def upload_asset():
|
||
if 'file' not in request.files:
|
||
return "No file", 400
|
||
file = request.files['file']
|
||
if file.filename == '':
|
||
return "No selected file", 400
|
||
|
||
filename = f"upload_{os.urandom(4).hex()}_{file.filename}"
|
||
upload_dir = os.path.join(app.root_path, 'static', 'uploads')
|
||
os.makedirs(upload_dir, exist_ok=True)
|
||
filepath = os.path.join(upload_dir, filename)
|
||
file.save(filepath)
|
||
|
||
print(f"File saved to: {filepath}")
|
||
|
||
# URL for access via browser
|
||
image_url = f"/static/uploads/{filename}"
|
||
|
||
return f'''
|
||
<div class="result-container">
|
||
<img src="{image_url}" class="generated-art animated fadeIn">
|
||
<div class="meta-info">✅ 上传成功: {file.filename}</div>
|
||
<input type="hidden" name="uploaded_url" value="{image_url}">
|
||
</div>
|
||
'''
|
||
|
||
@app.route('/api/run_node/<node_id>', methods=['POST'])
|
||
def run_node(node_id):
|
||
data = request.form.to_dict()
|
||
result = engine.execute_node(node_id, data)
|
||
|
||
if result.get('type') == 'image':
|
||
return f'''
|
||
<div class="result-container">
|
||
<img src="{result["url"]}" class="generated-art animated fadeIn">
|
||
<div class="meta-info">处理完成 | 耗时: {result["time"]}s</div>
|
||
</div>
|
||
'''
|
||
else:
|
||
error_msg = result.get('error', '未知错误')
|
||
return f'<div class="error-badge">❌ 错误: {error_msg}</div>'
|
||
|
||
@app.route('/api/chat', methods=['POST'])
|
||
def chat():
|
||
message = request.form.get('message', '')
|
||
return f'''
|
||
<div class="msg user">{message}</div>
|
||
<div class="msg ai">已接收指令。如果您想生成图片，请点击右侧“Nano-banana”节点的“启动生成”按钮。</div>
|
||
'''
|
||
|
||
if __name__ == '__main__':
|
||
app.run(debug=True, port=5000)
|