Upload app.py

app.py

@@ -11,6 +11,7 @@ app = Flask(__name__)
 CORS(app)  # 允许跨域请求
 
 original_pipeline = copy.deepcopy(pipeline)  # 保存原始 pipeline
+original_graph = copy.deepcopy(graph)  # 保存原始 graph
 
 @app.route("/")
 def index():
@@ -20,6 +21,8 @@ def index():
 def reset_pipeline():
     global pipeline
     pipeline = original_pipeline
+    global graph
+    graph = original_graph
     return jsonify({"message": "Pipeline reset successfully"})
 
 @app.route("/run_pipeline", methods=["POST"])