yeq6x commited on
Commit
e9f2510
·
1 Parent(s): e7c4eb4
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -65,6 +65,7 @@ def process_task(task):
65
  # ファイルデータをPIL Imageに変換
66
  image = Image.open(io.BytesIO(task.file_data))
67
  image = ensure_rgb(image)
 
68
 
69
  # キャンセルチェック
70
  if task.cancel_flag:
@@ -72,10 +73,12 @@ def process_task(task):
72
 
73
  # 画像処理ロジックを呼び出す
74
  sotai_image, sketch_image = process_image_as_base64(image, task.mode, task.weight1, task.weight2)
 
75
 
76
  # キャンセルチェック
77
  if task.cancel_flag:
78
  return
 
79
 
80
  socketio.emit('task_complete', {
81
  'task_id': task.task_id,
@@ -97,7 +100,7 @@ def process_task(task):
97
  client_ip = task.client_ip
98
  tasks_per_client[client_ip] = tasks_per_client.get(client_ip, 0) - 1
99
 
100
- update_queue_status('Task completed or cancelled')
101
 
102
  def worker():
103
  while True:
 
65
  # ファイルデータをPIL Imageに変換
66
  image = Image.open(io.BytesIO(task.file_data))
67
  image = ensure_rgb(image)
68
+ update_queue_status('Task 1')
69
 
70
  # キャンセルチェック
71
  if task.cancel_flag:
 
73
 
74
  # 画像処理ロジックを呼び出す
75
  sotai_image, sketch_image = process_image_as_base64(image, task.mode, task.weight1, task.weight2)
76
+ update_queue_status('Task 2')
77
 
78
  # キャンセルチェック
79
  if task.cancel_flag:
80
  return
81
+ update_queue_status('Task 3')
82
 
83
  socketio.emit('task_complete', {
84
  'task_id': task.task_id,
 
100
  client_ip = task.client_ip
101
  tasks_per_client[client_ip] = tasks_per_client.get(client_ip, 0) - 1
102
 
103
+ update_queue_status('Task completed or cancelled4')
104
 
105
  def worker():
106
  while True: