Upload 2 files

app.py

@@ -9,11 +9,17 @@ from citekit.utils.utils import parse_html_config
 app = Flask(__name__)
 CORS(app)  # 允许跨域请求
 
+original_pipeline = pipeline 
 
 @app.route("/")
 def index():
     return send_file("index.html")
 
+@app.route("/reset_pipeline", methods=["POST"])
+def reset_pipeline():
+    global pipeline
+    pipeline = original_pipeline
+    return jsonify({"message": "Pipeline reset successfully"})
 
 @app.route("/run_pipeline", methods=["POST"])
 def run_pipeline():
@@ -74,7 +80,7 @@ def get_config():
 def chat():
     data = request.json
     api_key = data.get("api_key")
-    os.environ["OPENAI_API_KEY"] = api_key
+    os.environ["OPENAI_API_KEY"]
     print("API Key Set.")
     user_message = data.get("message")
 

index.html

@@ -693,6 +693,26 @@
                 .style("color", "#333");
         }
 
+        function reset() {
+            fetch("/reset_pipeline", {
+                method: "POST",
+                headers: {
+                    "Content-Type": "application/json"
+                }
+            }).then(response => {
+                if (response.ok) {
+                    return response.json();
+                } else {
+                    throw new Error("Failed to reset pipeline");
+                }
+            }).then(data => {
+                console.log(data.message);
+                alert("Pipeline reset successfully");
+            }).catch(error => {
+                console.error("Error:", error);
+                alert("Error resetting pipeline");
+            });
+        }
         function createGraph(nodes, edges) {
             // 如果已有模拟实例则停止
             if (currentSimulation) currentSimulation.stop();
@@ -1250,9 +1270,9 @@
 
         }
 
-
+        reset();
         createGraph(nodes, edges);
-        // repaint();
+        repaint();
 
         function parseInput(input) {
             try {