Add training pipeline

app.py

@@ -4,16 +4,9 @@ Runs train_and_upgrade.py in the background, streams logs live,
 and exports the upgraded model to morpheuslord/rewrite on completion.
 """
 
-# ── Jinja2 / Gradio 4.44.0 compatibility patch ───────────────────────────────
-# Jinja2 >= 3.1.4 builds an LRU cache key that contains a dict (and sometimes
-# a list inside that dict), both of which are unhashable → TypeError on every
-# request.  Fix: wrap LRUCache.__getitem__ / __setitem__ / .get so that any
-# unhashable element is recursively converted to a hashable equivalent before
-# the key hits the underlying dict.  Key rules:
-#   dict  → frozenset of (k, v) pairs
-#   list  → tuple  (NOT list — list is still unhashable!)
-#   tuple → tuple with each element also made hashable
-#   other → unchanged
+# ── Patch 1: Jinja2 LRU cache ─────────────────────────────────────────────────
+# Jinja2 >= 3.1.4 puts environment globals (a dict) into the LRU cache key,
+# which is unhashable.  Convert any unhashable element to a hashable equivalent.
 def _patch_jinja2_lru_cache():
     import jinja2.utils as _ju
 
@@ -21,37 +14,64 @@ def _patch_jinja2_lru_cache():
         if isinstance(obj, dict):
             return frozenset((_make_hashable(k), _make_hashable(v)) for k, v in obj.items())
         if isinstance(obj, (list, tuple)):
-            return tuple(_make_hashable(i) for i in obj)   # always tuple, never list
+            return tuple(_make_hashable(i) for i in obj)  # always tuple, never list
         return obj
 
-    _orig_getitem = _ju.LRUCache.__getitem__
-    _orig_setitem = _ju.LRUCache.__setitem__
-    _orig_get     = _ju.LRUCache.get
+    _orig_gi = _ju.LRUCache.__getitem__
+    _orig_si = _ju.LRUCache.__setitem__
+    _orig_get = _ju.LRUCache.get
 
-    def _safe_getitem(self, key):
+    def _gi(self, key):
         try:
-            return _orig_getitem(self, key)
+            return _orig_gi(self, key)
         except TypeError:
-            return _orig_getitem(self, _make_hashable(key))
+            return _orig_gi(self, _make_hashable(key))
 
-    def _safe_setitem(self, key, value):
+    def _si(self, key, value):
         try:
-            _orig_setitem(self, key, value)
+            _orig_si(self, key, value)
         except TypeError:
-            _orig_setitem(self, _make_hashable(key), value)
+            _orig_si(self, _make_hashable(key), value)
 
-    def _safe_get(self, key, default=None):
+    def _get(self, key, default=None):
         try:
             return _orig_get(self, key, default)
         except TypeError:
             return _orig_get(self, _make_hashable(key), default)
 
-    _ju.LRUCache.__getitem__ = _safe_getitem
-    _ju.LRUCache.__setitem__ = _safe_setitem
-    _ju.LRUCache.get         = _safe_get
+    _ju.LRUCache.__getitem__ = _gi
+    _ju.LRUCache.__setitem__ = _si
+    _ju.LRUCache.get = _get
 
 _patch_jinja2_lru_cache()
 del _patch_jinja2_lru_cache
+
+# ── Patch 2: Starlette TemplateResponse API change ────────────────────────────
+# Gradio 4.44.0 calls: templates.TemplateResponse(name: str, context: dict)
+# Newer Starlette (0.29+) changed the signature to:
+#   TemplateResponse(request, name: str, context: dict)
+# so the context dict ends up as `name`, causing 'dict has no attribute split'.
+# Detect the old calling convention by checking if args[0] is a string, and
+# reorder arguments to match what newer Starlette expects.
+def _patch_starlette_template_response():
+    import starlette.templating as _st
+
+    _orig = _st.Jinja2Templates.TemplateResponse
+
+    def _compat(self, *args, **kwargs):
+        # Old API: (name: str, context: dict, ...)
+        # New API: (request, name: str, context: dict, ...)
+        if args and isinstance(args[0], str):
+            name = args[0]
+            context = args[1] if len(args) > 1 else {}
+            request = context.get("request")
+            return _orig(self, request, name, context, *args[2:], **kwargs)
+        return _orig(self, *args, **kwargs)
+
+    _st.Jinja2Templates.TemplateResponse = _compat
+
+_patch_starlette_template_response()
+del _patch_starlette_template_response
 # ─────────────────────────────────────────────────────────────────────────────
 
 # ── Compatibility patch ───────────────────────────────────────────────────────