Upload 4 files

.env

@@ -0,0 +1,2 @@
+URL=http://127.0.0.1:8188/prompt
+OUTPUT_DIR=ComfyUI\\output

app.py

@@ -0,0 +1,103 @@
+import gradio as gr
+import numpy as np
+import random
+import multiprocessing
+import subprocess
+import sys
+import time
+import signal
+import json
+import os
+import requests
+
+from loguru import logger
+from decouple import config
+
+URL = config('URL')
+OUTPUT_DIR = config('OUTPUT_DIR')
+
+
+def get_latest_image(folder):
+    files = os.listdir(folder)
+    image_files = [f for f in files if f.lower().endswith(('.png', '.jpg', '.jpeg'))]
+    image_files.sort(key=lambda x: os.path.getmtime(os.path.join(folder, x)))
+    latest_image = os.path.join(folder, image_files[-1]) if image_files else None
+    return latest_image
+
+
+def start_queue(prompt_workflow):
+    p = {"prompt": prompt_workflow}
+    data = json.dumps(p).encode('utf-8')
+    requests.post(URL, data=data)
+
+def generate_image(prompt_text):
+    with open("workflow_api.json", "r", encoding="utf-8") as file_json:
+        prompt = json.load(file_json)
+
+        prompt["6"]["inputs"]["text"] = f"digital artwork of a {prompt_text}"
+
+    previous_image = get_latest_image(OUTPUT_DIR)
+
+    start_queue(prompt)
+
+
+    while True:
+        latest_image = get_latest_image(OUTPUT_DIR)
+        if latest_image != previous_image:
+            return latest_image
+
+        time.sleep(1)
+
+
+def run_main_script(exit_event):
+    process = subprocess.Popen([sys.executable, "C:\\Users\\ilya9\\comfyui_gradio\\ComfyUI\main.py"])
+    logger.info(f"Скрипт main.py запущен в отдельном процессе с PID: {process.pid}")
+
+    while not exit_event.is_set():
+        if process.poll() is not None:
+            logger.info("Процесс main.py завершился самостоятельно")
+            break
+        time.sleep(0.1)
+
+    if process.poll() is None:
+        logger.info("Завершаем процесс main.py")
+        process.terminate()
+        process.wait(timeout=5)
+        if process.poll() is None:
+            logger.info("Процесс main.py не завершился, применяем SIGKILL")
+            process.kill()
+
+
+def signal_handler(signum, frame):
+    logger.info(f"Получен сигнал {signum}, завершаем работу")
+    exit_event.set()
+
+
+if __name__ == "__main__":
+    exit_event = multiprocessing.Event()
+
+    # Устанавливаем обработчик сигналов
+    signal.signal(signal.SIGINT, signal_handler)
+    signal.signal(signal.SIGTERM, signal_handler)
+
+    # Создаем и запускаем процесс
+    process = multiprocessing.Process(target=run_main_script, args=(exit_event,))
+    process.start()
+
+    try:
+        demo = gr.Interface(fn=generate_image, inputs=["text"], outputs=["image"])
+        demo.launch(share=True)
+    finally:
+        logger.info("Ожидаем завершения дочернего процесса")
+        exit_event.set()
+        process.join(timeout=10)
+        if process.is_alive():
+            logger.info("Дочерний процесс не завершился, применяем terminate()")
+            process.terminate()
+            process.join(timeout=5)
+            if process.is_alive():
+                logger.info("Дочерний процесс все еще жив, применяем kill()")
+                process.kill()
+                process.join()
+
+    logger.info("Основной скрипт завершил работу.")

requirements.txt

@@ -0,0 +1,110 @@
+absl-py==2.1.0
+aiofiles==23.2.1
+aiohttp==3.9.5
+aiosignal==1.3.1
+altair==5.3.0
+annotated-types==0.7.0
+anyio==4.4.0
+async-timeout==4.0.3
+attrs==23.2.0
+certifi==2024.6.2
+cffi==1.16.0
+charset-normalizer==3.3.2
+click==8.1.7
+colorama==0.4.6
+contourpy==1.2.1
+cycler==0.12.1
+dnspython==2.6.1
+einops==0.8.0
+email_validator==2.2.0
+exceptiongroup==1.2.1
+fastapi==0.111.0
+fastapi-cli==0.0.4
+ffmpy==0.3.2
+filelock==3.15.4
+fonttools==4.53.0
+frozenlist==1.4.1
+fsspec==2024.6.1
+gradio==4.37.2
+gradio_client==1.0.2
+h11==0.14.0
+httpcore==1.0.5
+httptools==0.6.1
+httpx==0.27.0
+huggingface-hub==0.23.4
+idna==3.7
+importlib_resources==6.4.0
+intel-openmp==2021.4.0
+Jinja2==3.1.4
+jsonschema==4.22.0
+jsonschema-specifications==2023.12.1
+kiwisolver==1.4.5
+kornia==0.7.3
+kornia_rs==0.1.3
+loguru==0.7.2
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+matplotlib==3.9.0
+mdurl==0.1.2
+mediapipe==0.10.9
+mkl==2021.4.0
+mpmath==1.3.0
+multidict==6.0.5
+networkx==3.3
+numpy==1.26.4
+opencv-contrib-python==4.9.0.80
+orjson==3.10.5
+packaging==24.1
+pandas==2.2.2
+pillow==10.3.0
+psutil==6.0.0
+pycparser==2.22
+pydantic==2.7.4
+pydantic_core==2.18.4
+pydub==0.25.1
+Pygments==2.18.0
+pyparsing==3.1.2
+python-dateutil==2.9.0.post0
+python-decouple==3.8
+python-dotenv==1.0.1
+python-multipart==0.0.9
+pytz==2024.1
+PyYAML==6.0.1
+referencing==0.35.1
+regex==2024.5.15
+requests==2.32.3
+rich==13.7.1
+rpds-py==0.18.1
+ruff==0.5.0
+safetensors==0.4.3
+scipy==1.14.0
+semantic-version==2.10.0
+shellingham==1.5.4
+six==1.16.0
+sniffio==1.3.1
+sounddevice==0.4.6
+soundfile==0.12.1
+spandrel==0.3.4
+starlette==0.37.2
+sympy==1.12.1
+tbb==2021.13.0
+tokenizers==0.19.1
+tomlkit==0.12.0
+toolz==0.12.1
+torch==2.3.1+cu121
+torchaudio==2.3.1+cu121
+torchsde==0.2.6
+torchvision==0.18.1+cu121
+tqdm==4.66.4
+trampoline==0.1.2
+transformers==4.42.3
+typer==0.12.3
+typing_extensions==4.12.2
+tzdata==2024.1
+ujson==5.10.0
+urllib3==2.2.2
+uvicorn==0.30.1
+watchfiles==0.22.0
+websockets==11.0.3
+win32-setctime==1.1.0
+yarl==1.9.4

workflow_api.json

@@ -0,0 +1,119 @@
+{
+  "4": {
+    "inputs": {
+      "ckpt_name": "realisticVisionV51_v51VAE.safetensors"
+    },
+    "class_type": "CheckpointLoaderSimple",
+    "_meta": {
+      "title": "Load Checkpoint - BASE"
+    }
+  },
+  "5": {
+    "inputs": {
+      "width": 1024,
+      "height": 1024,
+      "batch_size": 1
+    },
+    "class_type": "EmptyLatentImage",
+    "_meta": {
+      "title": "Empty Latent Image"
+    }
+  },
+  "6": {
+    "inputs": {
+      "text": "evening sunset scenery blue sky nature, glass bottle with a galaxy in it",
+      "clip": [
+        "4",
+        1
+      ]
+    },
+    "class_type": "CLIPTextEncode",
+    "_meta": {
+      "title": "CLIP Text Encode (Prompt)"
+    }
+  },
+  "7": {
+    "inputs": {
+      "text": "text, watermark",
+      "clip": [
+        "4",
+        1
+      ]
+    },
+    "class_type": "CLIPTextEncode",
+    "_meta": {
+      "title": "CLIP Text Encode (Prompt)"
+    }
+  },
+  "17": {
+    "inputs": {
+      "samples": [
+        "51",
+        0
+      ],
+      "vae": [
+        "4",
+        2
+      ]
+    },
+    "class_type": "VAEDecode",
+    "_meta": {
+      "title": "VAE Decode"
+    }
+  },
+  "49": {
+    "inputs": {
+      "images": [
+        "17",
+        0
+      ]
+    },
+    "class_type": "PreviewImage",
+    "_meta": {
+      "title": "Preview Image"
+    }
+  },
+  "50": {
+    "inputs": {
+      "filename_prefix": "ComfyUI",
+      "images": [
+        "17",
+        0
+      ]
+    },
+    "class_type": "SaveImage",
+    "_meta": {
+      "title": "Save Image"
+    }
+  },
+  "51": {
+    "inputs": {
+      "seed": 649530258978181,
+      "steps": 20,
+      "cfg": 8,
+      "sampler_name": "euler",
+      "scheduler": "normal",
+      "denoise": 1,
+      "model": [
+        "4",
+        0
+      ],
+      "positive": [
+        "6",
+        0
+      ],
+      "negative": [
+        "7",
+        0
+      ],
+      "latent_image": [
+        "5",
+        0
+      ]
+    },
+    "class_type": "KSampler",
+    "_meta": {
+      "title": "KSampler"
+    }
+  }
+}