Freak-ppa commited on
Commit
00e7bd6
·
verified ·
1 Parent(s): d0014dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -17,7 +17,7 @@ from pathlib import Path
17
  from PIL import Image
18
  import io
19
 
20
- URL = config('URL')
21
  OUTPUT_DIR = config('OUTPUT_DIR')
22
  INPUT_DIR = config('INPUT_DIR')
23
  COMF_PATH = config('COMF_PATH')
@@ -72,12 +72,12 @@ def delete_image_file(file_path):
72
  def start_queue(prompt_workflow, port):
73
  p = {"prompt": prompt_workflow}
74
  data = json.dumps(p).encode('utf-8')
75
- requests.post(URL+port+'/prompt', data=data)
76
 
77
 
78
  def check_server_ready(port):
79
  try:
80
- response = requests.get(f"{URL}{port}/history/123", timeout=5)
81
  return response.status_code == 200
82
  except requests.RequestException:
83
  return False
@@ -94,7 +94,7 @@ def generate_image(prompt, image):
94
  image.save(INPUT_DIR + '/input.png', format='PNG')
95
 
96
  process = None
97
- new_port = str(random.randint(8123, 15000))
98
 
99
  try:
100
  # Запускаем скрипт как подпроцесс
@@ -102,7 +102,7 @@ def generate_image(prompt, image):
102
  logger.debug(f'Subprocess started with PID: {process.pid}')
103
 
104
  # Ожидание запуска сервера
105
- for _ in range(20): # Максимум 20 секунд ожидания
106
  if check_server_ready(new_port):
107
  break
108
  time.sleep(1)
 
17
  from PIL import Image
18
  import io
19
 
20
+ URL="http://127.0.0.1"
21
  OUTPUT_DIR = config('OUTPUT_DIR')
22
  INPUT_DIR = config('INPUT_DIR')
23
  COMF_PATH = config('COMF_PATH')
 
72
  def start_queue(prompt_workflow, port):
73
  p = {"prompt": prompt_workflow}
74
  data = json.dumps(p).encode('utf-8')
75
+ requests.post(f"{URL}:{port}/prompt", data=data)
76
 
77
 
78
  def check_server_ready(port):
79
  try:
80
+ response = requests.get(f"{URL}:{port}/history/123", timeout=5)
81
  return response.status_code == 200
82
  except requests.RequestException:
83
  return False
 
94
  image.save(INPUT_DIR + '/input.png', format='PNG')
95
 
96
  process = None
97
+ new_port = str(random.randint(8123, 8200))
98
 
99
  try:
100
  # Запускаем скрипт как подпроцесс
 
102
  logger.debug(f'Subprocess started with PID: {process.pid}')
103
 
104
  # Ожидание запуска сервера
105
+ for _ in range(30): # Максимум 20 секунд ожидания
106
  if check_server_ready(new_port):
107
  break
108
  time.sleep(1)