Upload 3 files

.env

@@ -0,0 +1,3 @@
+OUTPUT_DIR=ComfyUI/output
+INPUT_DIR=ComfyUI/input
+COMF_PATH=ComfyUI/main.py

app.py

@@ -0,0 +1,165 @@
+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
+
+from pathlib import Path
+from PIL import Image
+import io
+
+URL="http://127.0.0.1"
+OUTPUT_DIR = config('OUTPUT_DIR')
+INPUT_DIR = config('INPUT_DIR')
+COMF_PATH = config('COMF_PATH')
+
+import torch
+
+import spaces
+
+print(f"Is CUDA available: {torch.cuda.is_available()}")
+print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+print(torch.version.cuda)
+device = torch.cuda.get_device_name(torch.cuda.current_device())
+print(device)
+
+
+def wait_for_image_with_prefix(folder, prefix):
+    def is_file_ready(file_path):
+        initial_size = os.path.getsize(file_path)
+        time.sleep(1)
+        return initial_size == os.path.getsize(file_path)
+
+
+    files = os.listdir(folder)
+    image_files = [f for f in files if f.lower().startswith(prefix.lower()) and
+                   f.lower().endswith(('.png', '.jpg', '.jpeg'))]
+
+    if image_files:
+        # Sort by modification time to get the latest file
+        image_files.sort(key=lambda x: os.path.getmtime(os.path.join(folder, x)), reverse=True)
+        latest_image = os.path.join(folder, image_files[0])
+
+        if is_file_ready(latest_image):
+            # Wait a bit more to ensure the file is completely written
+            time.sleep(3)
+            return latest_image
+
+    # If no matching file found, wait before checking again
+    return None
+
+
+def delete_image_file(file_path):
+    try:
+        if os.path.exists(file_path):
+            os.remove(file_path)
+            logger.debug(f"file {file_path} deleted")
+        else:
+            logger.debug(f"file {file_path} is not exist")
+    except Exception as e:
+        logger.debug(f"error {file_path}: {str(e)}")
+
+
+def start_queue(prompt_workflow, port):
+    p = {"prompt": prompt_workflow}
+    data = json.dumps(p).encode('utf-8')
+    requests.post(f"{URL}:{port}/prompt", data=data)
+
+
+def check_server_ready(port):
+    try:
+        response = requests.get(f"{URL}:{port}/history/123", timeout=5)
+        return response.status_code == 200
+    except requests.RequestException:
+        return False
+
+
+
+@spaces.GPU(duration=240)
+def generate_image(prompt, image, image2):
+    prefix_filename = str(random.randint(0, 999999))
+    prompt = prompt.replace('ComfyUI', prefix_filename)
+    prompt = json.loads(prompt)
+
+    image = Image.fromarray(image)
+    image.save(INPUT_DIR + '/input.png', format='PNG')
+    if image2 is not None:
+        image2 = Image.fromarray(image2)
+        image2.save(INPUT_DIR + '/input2.png', format='PNG')
+
+    process = None
+    new_port = str(random.randint(8123, 8200))
+
+    try:
+        # Запускаем скрипт как подпроцесс
+        process = subprocess.Popen([sys.executable, COMF_PATH, "--listen", "127.0.0.1", "--port", new_port])
+        logger.debug(f'Subprocess started with PID: {process.pid}')
+
+        # Ожидание запуска сервера
+        for _ in range(30):  # Максимум 20 секунд ожидания
+            if check_server_ready(new_port):
+                break
+            time.sleep(1)
+        else:
+            raise TimeoutError("Server did not start in time")
+
+        start_queue(prompt, new_port)
+
+        # Ожидание нового изображения
+        timeout = 220  # Максимальное время ожидания в секундах
+        start_time = time.time()
+        while time.time() - start_time < timeout:
+            latest_image = wait_for_image_with_prefix(OUTPUT_DIR, prefix_filename)
+            if latest_image:
+                logger.debug(f"file is: {latest_image}")
+                try:
+                    return Image.open(latest_image)
+                finally:
+                    delete_image_file(latest_image)
+                    delete_image_file(INPUT_DIR + '/input.png')
+                    if image2 is not None:
+                        delete_image_file(INPUT_DIR + '/input2.png')
+            time.sleep(1)
+
+        raise TimeoutError("New image was not generated in time")
+
+    except Exception as e:
+        logger.error(f"Error in generate_image: {e}")
+
+    finally:        
+        if process and process.poll() is None:
+            process.terminate()
+            logger.debug("process.terminate()")
+            try:
+                logger.debug("process.wait(timeout=5)")
+                process.wait(timeout=5)
+            except subprocess.TimeoutExpired:
+                logger.debug("process.kill()")
+                process.kill()
+
+
+
+if __name__ == "__main__":
+    demo = gr.Interface(fn=generate_image,
+                        inputs=[
+                            "text",
+                            gr.Image(image_mode='RGBA', type="numpy"),
+                            gr.Image(image_mode='RGBA', type="numpy")
+                        ],
+                        outputs=[
+                            gr.Image(type="numpy", image_mode='RGBA')
+                        ]
+                        )
+    demo.launch(debug=True)
+    logger.debug('demo.launch()')
+
+    logger.info("Основной скрипт завершил работу.")

requirements.txt

@@ -0,0 +1,122 @@
+absl-py==2.1.0
+accelerate
+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
+diffusers
+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
+gevent==24.2.1
+greenlet==3.0.3
+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_metadata==8.0.0
+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
+peft==0.11.1
+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
+torchaudio
+torchsde
+torchvision
+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
+websocket==0.2.1
+websocket-client==1.6.3
+websockets==11.0.3
+win32-setctime==1.1.0
+yarl==1.9.4
+zipp==3.19.2
+zope.event==5.0
+zope.interface==6.4.post2
+opencv-python
+insightface --user
+onnxruntime