Harden TemplateResponse shim for mapping-context calls

app.py

@@ -6,6 +6,7 @@ import time
 from concurrent.futures import ThreadPoolExecutor, as_completed
 from dataclasses import dataclass
 from io import BytesIO
+from collections.abc import Mapping
 from typing import Dict, List, Optional, Tuple
 
 import gradio as gr
@@ -49,8 +50,14 @@ try:
     _original_template_response = Jinja2Templates.TemplateResponse
 
     def _safe_template_response(self, name, context=None, *args, **kwargs):
-        if isinstance(context, dict):
+        if context is None:
+            context = kwargs.pop("context", None)
+        if context is None:
+            context = {}
+        elif isinstance(context, Mapping):
             context = dict(context)
+
+        if isinstance(context, dict):
             context.setdefault(
                 "config",
                 {