fixbug

shadowcz007 · Jul 28, 2024 · 2afd24f · 2afd24f
1 parent 437acd0
commit 2afd24f
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 12 deletions.
diff --git a/__init__.py b/__init__.py
@@ -619,29 +619,33 @@ async def mixlab_workflow_hander(request):
                 if 'admin' in data:
                     admin=data['admin']
 
-                ds=get_my_workflow_for_app(filename,category,admin)                
+                ds=get_my_workflow_for_app(filename,category,admin)
+                data=[]           
                 for json_data in ds:
                     # 不传给前端
-                    if 'output' in json_data:
-                        del json_data['output']
-                    if 'workflow' in json_data:
-                        del json_data['workflow']
+                    if 'output' in json_data['data']:
+                        del json_data['data']['output']
+                    if 'workflow' in json_data['data']:
+                        del json_data['data']['workflow']
+                    data.append(json_data)
 
                 result={
-                    'data':ds,
+                    'data':data,
                     'status':'success',
                 }
             elif data['task']=='list':
                 ds=get_workflows()
+                data=[]
                 for json_data in ds:
                     # 不传给前端
-                    if 'output' in json_data:
-                        del json_data['output']
-                    if 'workflow' in json_data:
-                        del json_data['workflow']
+                    if 'output' in json_data['data']:
+                        del json_data['data']['output']
+                    if 'workflow' in json_data['data']:
+                        del json_data['data']['workflow']
+                    data.append(json_data)
 
                 result={
-                    'data':ds,
+                    'data':data,
                     'status':'success',
                 }
     except Exception as e:

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-mixlab-nodes"
 description = "3D, ScreenShareNode & FloatingVideoNode, SpeechRecognition & SpeechSynthesis, GPT, LoadImagesFromLocal, Layers, Other Nodes, ..."
-version = "0.31.0"
+version = "0.31.1"
 license = "MIT"
 dependencies = ["numpy", "pyOpenSSL", "watchdog", "opencv-python-headless", "matplotlib", "openai", "simple-lama-inpainting", "clip-interrogator==0.6.0", "transformers>=4.36.0", "lark-parser", "imageio-ffmpeg", "rembg[gpu]", "omegaconf==2.3.0", "Pillow>=9.5.0", "einops==0.7.0", "trimesh>=4.0.5", "huggingface-hub", "scikit-image"]