| from __future__ import annotations |
|
|
| import html |
| import time |
|
|
| from shared.api import extract_status_phase_label |
|
|
|
|
| def phase_label_from_status(status: str = "") -> str: |
| return extract_status_phase_label(status) |
|
|
|
|
| def phase_label_from_update(update=None, *, status: str = "", phase: str = "", raw_phase: str = "") -> str: |
| status_phase = phase_label_from_status(status or getattr(update, "status", "")) |
| raw_phase_text = str(raw_phase or getattr(update, "raw_phase", "") or phase or "").strip() |
| if len(status_phase) > 0: |
| return status_phase |
| return raw_phase_text |
|
|
|
|
| def _format_elapsed(seconds: float | None) -> str: |
| if seconds is None: |
| return "--" |
| total_seconds = int(round(float(seconds))) |
| if total_seconds < 0: |
| total_seconds = 0 |
| hours, rem = divmod(total_seconds, 3600) |
| minutes, seconds_only = divmod(rem, 60) |
| return f"{hours:02d}:{minutes:02d}:{seconds_only:02d}" if hours > 0 else f"{minutes:02d}:{seconds_only:02d}" |
|
|
|
|
| class ChunkCallbacks: |
| def __init__(self) -> None: |
| self.phase_label = "Queued in WanGP..." |
| self.status_text = "Queued in WanGP..." |
| self.current_step = None |
| self.total_steps = None |
| self._last_explicit_status_at = 0.0 |
|
|
| def on_status(self, status): |
| status_text = str(status or "").strip() |
| if len(status_text) == 0: |
| return |
| self.status_text = status_text |
| status_phase = phase_label_from_status(self.status_text) |
| if len(status_phase) > 0: |
| if status_phase != self.phase_label: |
| self.current_step = None |
| self.total_steps = None |
| self.phase_label = status_phase |
| self._last_explicit_status_at = time.time() |
|
|
| def on_progress(self, update): |
| incoming_status = str(getattr(update, "status", "") or "").strip() |
| incoming_phase = phase_label_from_update(update, status=incoming_status or self.status_text) |
| incoming_step = getattr(update, "current_step", None) |
| incoming_total = getattr(update, "total_steps", None) |
| if time.time() - self._last_explicit_status_at <= 1.0 and len(self.phase_label) > 0 and len(incoming_phase) > 0 and incoming_phase.lower() != self.phase_label.lower() and incoming_step is None: |
| return |
| if len(incoming_status) > 0: |
| self.status_text = incoming_status |
| if len(incoming_phase) > 0: |
| self.phase_label = incoming_phase |
| self.current_step = incoming_step |
| self.total_steps = incoming_total |
|
|
|
|
| def render_chunk_status_html(total_chunks: int, completed_chunks: int, current_chunk: int, phase_label: str, status_text: str, *, continued: bool = False, phase_current_step=None, phase_total_steps=None, elapsed_seconds: float | None = None, eta_seconds: float | None = None, prefer_status_phase: bool = False) -> str: |
| total_chunks = int(total_chunks) |
| completed_chunks = int(completed_chunks) |
| current_chunk = int(current_chunk) |
| if total_chunks > 0: |
| top_ratio = completed_chunks / total_chunks |
| chunks_text = f"{completed_chunks} / {total_chunks}" |
| else: |
| top_ratio = 0.0 |
| chunks_text = "- / -" |
| top_width = f"{100.0 * top_ratio:.2f}%" |
| raw_status_text = status_text.strip() |
| raw_phase_text = phase_label.strip() |
| if prefer_status_phase: |
| derived_phase = phase_label_from_status(raw_status_text) |
| if len(derived_phase) > 0: |
| raw_phase_text = derived_phase |
| phase_html = html.escape(raw_phase_text or "Queued in WanGP...") |
| status_html = html.escape(raw_status_text or raw_phase_text or "") |
| continued_suffix = " (Continued)" if continued else "" |
| has_phase_progress = phase_current_step is not None and phase_total_steps is not None and phase_total_steps > 0 |
| phase_ratio = float(phase_current_step) / float(phase_total_steps) if has_phase_progress else None |
| phase_width = f"{100.0 * phase_ratio:.2f}%" if phase_ratio is not None else "0%" |
| phase_suffix = f" ({phase_current_step} / {phase_total_steps})" if has_phase_progress else "" |
| elapsed_html = html.escape(_format_elapsed(elapsed_seconds)) |
| eta_html = html.escape(_format_elapsed(eta_seconds)) |
| normalized_phase = raw_phase_text.lower() |
| normalized_status = raw_status_text |
| show_status_line = (not prefer_status_phase) and len(normalized_status) > 0 and (len(normalized_phase) == 0 or normalized_phase not in normalized_status.lower()) |
| status_line_html = f"<div style='font-size:0.9em;color:#4b5563'>{status_html}</div>" if show_status_line else "" |
| return ( |
| "<div style='display:flex;flex-direction:column;gap:8px'>" |
| f"<div style='font-weight:600'>Chunks Processed: {chunks_text}{continued_suffix}</div>" |
| "<div style='height:12px;border-radius:999px;background:#d7dce3;overflow:hidden'>" |
| f"<div style='height:100%;width:{top_width};background:linear-gradient(90deg,#2f7de1,#5db0ff)'></div>" |
| "</div>" |
| f"<div style='font-size:0.95em'><b>Phase:</b> {phase_html}{phase_suffix}</div>" |
| "<div style='height:12px;border-radius:999px;background:#d7dce3;overflow:hidden'>" |
| f"<div style='height:100%;width:{phase_width};background:linear-gradient(90deg,#e37a2f,#ffb05d)'></div>" |
| "</div>" |
| f"<div style='font-size:0.9em;color:#4b5563'><b>Elapsed:</b> {elapsed_html} <span style='padding-left:12px'><b>ETA:</b> {eta_html}</span></div>" |
| f"{status_line_html}" |
| "</div>" |
| ) |
|
|
|
|
| def render_output_file_html(output_path: str) -> str: |
| value = html.escape(output_path, quote=False) |
| return ( |
| "<div style='display:flex;flex-direction:column;gap:6px'>" |
| "<div style='font-size:var(--block-label-text-size);font-weight:var(--block-label-text-weight);line-height:var(--line-sm)'>Output File</div>" |
| f"<textarea readonly onclick='this.select()' spellcheck='false' rows='1' " |
| "style='width:100%;min-height:35.64px;resize:none;overflow:hidden;padding:calc(8px * var(--wangp-ui-scale)) calc(12px * var(--wangp-ui-scale));" |
| "border:1px solid var(--input-border-color);border-radius:var(--input-radius);background:var(--input-background-fill);color:var(--body-text-color);" |
| "font:inherit;line-height:1.5;box-sizing:border-box'>" |
| f"{value}</textarea>" |
| "</div>" |
| ) |
|
|
|
|
|
|
