Update app.py

app.py

@@ -24,7 +24,7 @@ COMF_PATH = config('COMF_PATH')
 
 import torch
 
-import spaces
+#import spaces
 
 print(f"Is CUDA available: {torch.cuda.is_available()}")
 print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
@@ -69,24 +69,22 @@ def delete_image_file(file_path):
         logger.debug(f"error {file_path}: {str(e)}")
 
 
-def start_queue(prompt_workflow):
+def start_queue(prompt_workflow, port):
     p = {"prompt": prompt_workflow}
     data = json.dumps(p).encode('utf-8')
-    requests.post(URL, data=data)
+    requests.post(URL+port+'/prompt', data=data)
 
 
-def check_server_ready():
+def check_server_ready(port):
     try:
-        response = requests.get(f"http://127.0.0.1:8188/history/123", timeout=5)
+        response = requests.get(f"{URL}{port}/history/123", timeout=5)
         return response.status_code == 200
     except requests.RequestException:
         return False
 
 
-queue_reqs = set()
-process_started = False
 
-@spaces.GPU(duration=240)
+#@spaces.GPU(duration=240)
 def generate_image(prompt, image):
     prefix_filename = str(random.randint(0, 999999))
     prompt = prompt.replace('ComfyUI', prefix_filename)
@@ -95,25 +93,23 @@ def generate_image(prompt, image):
     image = Image.fromarray(image)
     image.save(INPUT_DIR + '/input.png', format='PNG')
 
-    queue_reqs.add(prefix_filename)
     process = None
-    global process_started
+    new_port = str(random.randint(8123, 15000))
+
     try:
-        if not process_started:
-            process_started = True
-            # Запускаем скрипт как подпроцесс
-            process = subprocess.Popen([sys.executable, COMF_PATH, "--listen", "127.0.0.1"])
-            logger.debug(f'Subprocess started with PID: {process.pid}')
+        # Запускаем скрипт как подпроцесс
+        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(20):  # Максимум 20 секунд ожидания
-            if check_server_ready():
+            if check_server_ready(new_port):
                 break
             time.sleep(1)
         else:
             raise TimeoutError("Server did not start in time")
 
-        start_queue(prompt)
+        start_queue(prompt, new_port)
 
         # Ожидание нового изображения
         timeout = 220  # Максимальное время ожидания в секундах
@@ -134,15 +130,14 @@ def generate_image(prompt, image):
         logger.error(f"Error in generate_image: {e}")
         return None
 
-    finally:
-        queue_reqs.remove(prefix_filename)
-        if len(queue_reqs) == 0 and process and process.poll() is None:
+    finally:        
+        if process and process.poll() is None:
             process.terminate()
             try:
                 process.wait(timeout=5)
             except subprocess.TimeoutExpired:
                 process.kill()
-            process_started = False
+
 
 
 if __name__ == "__main__":