新增接口：支持传prompt_id的中断任务接口和/prompt接口

easyapi/api.py

@@ -1,5 +1,7 @@
 import os
 import json
+
+import nodes
 from server import PromptServer
 from aiohttp import web
 import execution
@@ -62,8 +64,59 @@ def register_routes():
 
         return web.json_response({"maxSize": maxSize})
 
+    @PromptServer.instance.routes.post("/easyapi/prompt")
+    async def post_prompt(request):
+        print("got prompt")
+        json_data = await request.json()
+        json_data = PromptServer.instance.trigger_on_prompt(json_data)
+        prompt_id = json_data["prompt_id"]
+        print("prompt_id={}".format(json_data["prompt_id"]))
+
+        if "number" in json_data:
+            number = float(json_data['number'])
+        else:
+            number = PromptServer.instance.number
+            if "front" in json_data:
+                if json_data['front']:
+                    number = -number
+
+            PromptServer.instance.number += 1
+
+        if "prompt" in json_data:
+            prompt = json_data["prompt"]
+            valid = execution.validate_prompt(prompt)
+            extra_data = {}
+            if "extra_data" in json_data:
+                extra_data = json_data["extra_data"]
+
+            if "client_id" in json_data:
+                extra_data["client_id"] = json_data["client_id"]
+            if valid[0]:
+                outputs_to_execute = valid[2]
+                PromptServer.instance.prompt_queue.put((number, prompt_id, prompt, extra_data, outputs_to_execute))
+                response = {"prompt_id": prompt_id, "number": number, "node_errors": valid[3]}
+                return web.json_response(response)
+            else:
+                print("invalid prompt:", valid[1])
+                return web.json_response({"error": valid[1], "node_errors": valid[3]}, status=400)
+        else:
+            return web.json_response({"error": "no prompt", "node_errors": []}, status=400)
+
+    @PromptServer.instance.routes.post("/easyapi/interrupt")
+    async def post_interrupt(request):
+        json_data = await request.json()
+        prompt_id = json_data["prompt_id"]
+        current_queue = PromptServer.instance.prompt_queue.get_current_queue()
+        queue_running = current_queue[0]
+        if queue_running is not None:
+            if queue_running[1] == prompt_id:
+                nodes.interrupt_processing()
+
+        delete_func = lambda a: a[1] == prompt_id
+        PromptServer.instance.prompt_queue.delete_queue_item(delete_func)
+        return web.Response(status=200)
+
 
 def init():
     reset_history_size(isStart=True)
     register_routes()
-