1.9.0

sdppp_python/apis.py

@@ -57,6 +57,25 @@ def registerComfyHTTPEndpoints(sdppp, PromptServer):
             return web.json_response({
                 'error': str(e)
             })
+
+    @PromptServer.instance.routes.post('/sdppp_upload2')
+    async def sdppp_upload(request):
+        post = await request.post()
+        image = post.get('image')
+        filename = post.get('filename')
+        if filename is None:
+            filename = image.filename or 'uploaded_image'
+        
+        # Read the file data from the FileField object
+        image_data = image.file.read()
+        
+        # 将filename传递给addImageCache作为name参数
+        result = await upload_image(sdppp, image_data, filename)
+        return web.json_response({
+            'name': result['name'], 
+            'subfolder': result['subfolder'], 
+            'type': result['type']
+        })
         
     
 def registerSDHTTPEndpoints(sdppp, app):
@@ -74,6 +93,7 @@ def registerSDHTTPEndpoints(sdppp, app):
         name = addImageCache(Image.open(BytesIO(image.file.read())))
         return {'name': name}
 
+
 def registerSocketEvents(sdppp, sio):
 
 

sdppp_python/sdppp.py

@@ -42,24 +42,6 @@ class SDPPP:
         registerComfyHTTPEndpoints(self, PromptServer)
         self.server_type = "comfy"
 
-
-    def attach_to_SD(self, app):
-        self.sio = socketio.AsyncServer(
-            async_mode='asgi', 
-            cors_allowed_origins="*",
-            max_http_buffer_size=524288000
-        )
-        app_sdppp = socketio.ASGIApp(
-            socketio_server=self.sio, 
-            socketio_path=''
-        )
-        app.mount('/sd-ppp', app_sdppp)
-        self.app = app
-        
-        self._registerSocketListeners()
-        registerSDHTTPEndpoints(self, app)
-        self.server_type = "a1111"
-
     def _registerSocketListeners(self):
         sio = self.sio
 
@@ -77,6 +59,8 @@ class SDPPP:
             with open(path.join(projectRoot, 'sdppp_python', 'version.txt'), 'r') as f:
                 api_level = f.read().strip()
             if api_level is not None and ('api_level' not in qsobj or qsobj['api_level'] != api_level):
+                if api_level is not None and 'api_level' in qsobj: 
+                    print(f'api_level: {api_level}, qsobj: {qsobj["api_level"]}')
                 raise socketio.exceptions.ConnectionRefusedError('version mismatch, please reinstall PS plugin')
 
         @sio.event

sdppp_python/version.txt

@@ -1 +1 @@
-504
+507