fix: add task log

This commit is contained in:
p
2026-01-28 21:08:37 +08:00
parent 710562deb8
commit cc8869aeea
4 changed files with 47 additions and 23 deletions
+14 -5
View File
@@ -13,6 +13,7 @@ import traceback
import logging
import sys
import time
# Configure logging
logging.basicConfig(level=logging.INFO, stream=sys.stdout)
@@ -635,11 +636,19 @@ Ensure the character's expression and pose reflect their personality: {personali
"""
log_task_event(session, task_id, f"Calling AI service for character {char.name}...")
image_bytes = ai.generate_image(
prompt,
aspect_ratio=char.project.aspect_ratio or "16:9",
resolution=char.project.resolution or "2K"
)
start_time = time.time()
try:
image_bytes = ai.generate_image(
prompt,
aspect_ratio=char.project.aspect_ratio or "16:9",
resolution=char.project.resolution or "2K"
)
elapsed = time.time() - start_time
log_task_event(session, task_id, f"AI generation finished in {elapsed:.2f}s. Image size: {len(image_bytes)} bytes.")
except Exception as e:
elapsed = time.time() - start_time
log_task_event(session, task_id, f"AI generation failed after {elapsed:.2f}s: {str(e)}")
raise e
relative_url = save_generated_image(session, char.project_id, "character", char.id, image_bytes)