Upload 133 files

app.py

@@ -19,6 +19,7 @@ from __future__ import annotations
 
 import gc
 import json as _json
+import logging
 import os
 import re
 import time
@@ -26,6 +27,8 @@ import threading
 from datetime import datetime
 from pathlib import Path
 
+logger = logging.getLogger(__name__)
+
 # ── Container environment fixes ──────────────────────────────────────
 # PyTorch 2.6+ calls getpass.getuser() to build a cache dir, which fails
 # in containers running as a UID with no /etc/passwd entry (e.g. UID 1000
@@ -1334,8 +1337,8 @@ def benchmark(
                 n_prompts=actual_n,
                 quantization=quantization,
             )
-        except Exception:
-            pass  # Telemetry is best-effort, never block benchmarks
+        except Exception as _tel_err:
+            logger.debug("Telemetry logging failed (best-effort): %s", _tel_err)
 
         # Store config so user can load this result into the Chat tab.
         # Keep the checkpoint on disk so loading doesn't require re-training.
@@ -1686,8 +1689,8 @@ def benchmark_multi_model(
                 n_prompts=actual_n,
                 quantization=quantization,
             )
-        except Exception:
-            pass  # Telemetry is best-effort
+        except Exception as _tel_err:
+            logger.debug("Telemetry logging failed (best-effort): %s", _tel_err)
 
         # Store config so user can load this result into the Chat tab.
         # Keep the checkpoint on disk so loading doesn't require re-training.
@@ -2087,11 +2090,11 @@ def obliterate(model_choice: str, method_choice: str,
 
     # Handle error
     if error_ref[0] is not None:
-        with _lock:
-            _state["status"] = "idle"
         err_msg = str(error_ref[0]) or repr(error_ref[0])
         log_lines.append(f"\nERROR: {err_msg}")
-        _state["log"] = log_lines
+        with _lock:
+            _state["status"] = "idle"
+            _state["log"] = log_lines
         yield f"**Error:** {err_msg}", "\n".join(log_lines), get_chat_header(), gr.update(), gr.update(), gr.update()
         return
 
@@ -2099,6 +2102,20 @@ def obliterate(model_choice: str, method_choice: str,
     # Wrapped in try/except to ensure status is never stuck on "obliterating".
     try:
         pipeline = pipeline_ref[0]
+        if pipeline is None:
+            # Worker thread completed without error but pipeline was never assigned
+            # (e.g. import failure caught internally, or early return in worker).
+            with _lock:
+                _state["status"] = "idle"
+            log_lines.append("\nERROR: Pipeline completed but produced no result.")
+            with _lock:
+                _state["log"] = log_lines
+            yield (
+                "**Error:** Obliteration finished but no pipeline was produced. "
+                "Check the log for details — this may indicate an import or configuration issue.",
+                "\n".join(log_lines), get_chat_header(), gr.update(), gr.update(), gr.update(),
+            )
+            return
         can_generate = pipeline._quality_metrics.get("coherence") is not None
 
         # ── Telemetry: log single obliteration to community leaderboard ──
@@ -2132,8 +2149,8 @@ def obliterate(model_choice: str, method_choice: str,
                 quantization=quantization,
             )
             maybe_send_pipeline_report(pipeline)
-        except Exception:
-            pass  # Telemetry is best-effort
+        except Exception as _tel_err:
+            logger.debug("Telemetry logging failed (best-effort): %s", _tel_err)
 
         # ── Session cache: register this obliteration for Chat tab switching ──
         global _last_obliterated_label
@@ -2298,7 +2315,8 @@ def obliterate(model_choice: str, method_choice: str,
             log_lines.append(f"LIBERATION COMPLETE in {_elapsed()} \u2014 model saved!")
         log_lines.append("=" * 50)
 
-        _state["log"] = log_lines
+        with _lock:
+            _state["log"] = log_lines
         if can_generate:
             status_msg = f"**{model_choice}** liberated with `{method}` in {_elapsed()}. Head to the **Chat** tab."
         else:
@@ -2324,11 +2342,11 @@ def obliterate(model_choice: str, method_choice: str,
 
     except Exception as e:
         # Ensure status never gets stuck on "obliterating"
-        with _lock:
-            _state["status"] = "idle"
         err_msg = str(e) or repr(e)
         log_lines.append(f"\nERROR (post-pipeline): {err_msg}")
-        _state["log"] = log_lines
+        with _lock:
+            _state["status"] = "idle"
+            _state["log"] = log_lines
         yield f"**Error:** {err_msg}", "\n".join(log_lines), get_chat_header(), gr.update(), gr.update(), gr.update()
 
 
@@ -3473,8 +3491,8 @@ def run_tourney(model_choice, selected_methods, dataset, quantization):
                 dataset=dataset_key,
                 quantization=quant,
             )
-        except Exception:
-            pass  # Telemetry is best-effort
+        except Exception as _tel_err:
+            logger.debug("Telemetry logging failed (best-effort): %s", _tel_err)
 
     if winner:
         bracket_md = render_bracket_html(result)
@@ -3988,8 +4006,38 @@ input[type="range"] { accent-color: #00ff41 !important; }
 
 _JS = """
 () => {
+    // ── Audible ping on completion ──────────────────────────────────
+    // Synthesize a short "ping" using Web Audio API — no audio files needed.
+    let _audioCtx = null;
+    function _playPing() {
+        try {
+            if (!_audioCtx) _audioCtx = new (window.AudioContext || window.webkitAudioContext)();
+            const osc = _audioCtx.createOscillator();
+            const gain = _audioCtx.createGain();
+            osc.connect(gain);
+            gain.connect(_audioCtx.destination);
+            osc.type = 'sine';
+            osc.frequency.setValueAtTime(880, _audioCtx.currentTime);       // A5
+            osc.frequency.setValueAtTime(1320, _audioCtx.currentTime + 0.08); // E6
+            gain.gain.setValueAtTime(0.3, _audioCtx.currentTime);
+            gain.gain.exponentialRampToValueAtTime(0.001, _audioCtx.currentTime + 0.4);
+            osc.start(_audioCtx.currentTime);
+            osc.stop(_audioCtx.currentTime + 0.4);
+        } catch(e) { /* Audio not available */ }
+    }
+
+    // Track which completion messages we've already pinged for
+    const _pingedMessages = new Set();
+    const _completionPatterns = [
+        'LIBERATION COMPLETE',
+        'BENCHMARK COMPLETE',
+        'Champion:',
+        'Tournament complete',
+    ];
+
     // Auto-scroll log box to bottom when content changes,
-    // and flash the log border red if an ERROR appears
+    // flash the log border red if an ERROR appears,
+    // and play a ping on completion events
     const observer = new MutationObserver(() => {
         document.querySelectorAll('.log-box textarea').forEach(el => {
             el.scrollTop = el.scrollHeight;
@@ -4000,6 +4048,16 @@ _JS = """
                 el.style.borderColor = '#00ff41';
                 el.style.boxShadow = 'none';
             }
+            // Check for completion patterns and ping once per unique message
+            if (el.value) {
+                for (const pattern of _completionPatterns) {
+                    if (el.value.includes(pattern) && !_pingedMessages.has(pattern + el.value.length)) {
+                        _pingedMessages.add(pattern + el.value.length);
+                        _playPing();
+                        break;
+                    }
+                }
+            }
         });
     });
     setTimeout(() => {