yeq6x commited on
Commit
6446714
·
1 Parent(s): 9fcc748
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -60,6 +60,7 @@ def update_queue_status(message):
60
  def process_task(task):
61
  try:
62
  client_id = session.get("sid")
 
63
  task.is_processing = True
64
  # ファイルデータをPIL Imageに変換
65
  image = Image.open(io.BytesIO(task.file_data))
@@ -71,7 +72,7 @@ def process_task(task):
71
 
72
  # 画像処理ロジックを呼び出す
73
  sotai_image, sketch_image = process_image_as_base64(image, task.mode, task.weight1, task.weight2)
74
-
75
  # キャンセルチェック
76
  if task.cancel_flag:
77
  return
@@ -82,6 +83,7 @@ def process_task(task):
82
  'sketch_image': sketch_image
83
  }, to=client_id)
84
  except Exception as e:
 
85
  if not task.cancel_flag:
86
  socketio.emit('task_error', {'task_id': task.task_id, 'error': str(e)}, to=client_id)
87
  finally:
 
60
  def process_task(task):
61
  try:
62
  client_id = session.get("sid")
63
+ print(f"client_id: {client_id}")
64
  task.is_processing = True
65
  # ファイルデータをPIL Imageに変換
66
  image = Image.open(io.BytesIO(task.file_data))
 
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
 
83
  'sketch_image': sketch_image
84
  }, to=client_id)
85
  except Exception as e:
86
+ print(f"Task error: {str(e)}")
87
  if not task.cancel_flag:
88
  socketio.emit('task_error', {'task_id': task.task_id, 'error': str(e)}, to=client_id)
89
  finally: