Fix duplicate progress handler in backend.py and add debug logs

parent 6c0c6b0b
......@@ -181,6 +181,15 @@ def handle_worker_message(message: Message, client_sock) -> None:
pending_results[progress_key] = message
job_progress_times[message.data.get('job_id')] = time.time()
print(f"Progress update for job {message.data.get('job_id')}: {message.data.get('progress', 0)}% - {message.data.get('message', '')}")
# Update job progress in database
from .database import update_job_progress
job_id = message.data.get('job_id')
progress = message.data.get('progress', 0)
progress_message = message.data.get('message', '')
if job_id:
print(f"DEBUG: calling update_job_progress job_id={job_id}, progress={progress}")
success = update_job_progress(job_id, progress, progress_message)
print(f"DEBUG: update_job_progress success = {success}")
elif message.msg_type == 'ping':
# Handle ping messages to keep connection alive and reset timeout
job_id = message.data.get('job_id')
......@@ -188,14 +197,6 @@ def handle_worker_message(message: Message, client_sock) -> None:
# Update progress timestamp to reset timeout
job_progress_times[job_id] = time.time()
print(f"PING received for job {job_id} - resetting timeout")
elif message.msg_type == 'progress':
# Update job progress in database
from .database import update_job_progress
job_id = message.data.get('job_id')
progress = message.data.get('progress', 0)
progress_message = message.data.get('message', '')
if job_id:
update_job_progress(job_id, progress, progress_message)
elif message.msg_type in ['analyze_response', 'train_response']:
# Store result for web to poll
pending_results[message.msg_id] = message
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment