Upload app.py

app.py

@@ -3,15 +3,15 @@ from flask_cors import CORS
 import openai
 import sys
 import os
-from methods.self_RAG_demo import pipeline, graph
+from methods.self_RAG_demo import pipeline
 from citekit.utils.utils import parse_html_config
 import copy
+from parser import *
 
 app = Flask(__name__)
 CORS(app)  # 允许跨域请求
 
 original_pipeline = copy.deepcopy(pipeline)  # 保存原始 pipeline
-original_graph = copy.deepcopy(graph)  # 保存原始 graph
 
 @app.route("/")
 def index():
@@ -21,8 +21,6 @@ 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"])
@@ -40,7 +38,7 @@ def run_pipeline():
     
 @app.route("/get_nodes", methods=["POST"])
 def get_nodes(*args, **kwargs):
-    graph.update()
+    graph = PipelineGraph(pipeline)
     try:
         return jsonify(graph.get_json()) 
     except Exception as e:
@@ -84,9 +82,9 @@ def get_config():
 def chat():
     data = request.json
     api_key = data.get("api_key")
-    os.environ["OPENAI_API_KEY"]
-    print("API Key Set.")
     user_message = data.get("message")
+    if os.environ.get("OPENAI_API_KEY"):
+        api_key = os.environ.get("OPENAI_API_KEY")
 
     if not api_key or not user_message:
         return jsonify({"error": "API Key and message are required"}), 400