Pull Spaces decorators up

app.py

@@ -434,6 +434,259 @@ def get_rules_json():
     return json.dumps(rule_loader.get_all_rules())
 
 
+def load_chat_context(case_id):
+    if not case_id:
+        default_ctx = "You are TraceScene AI assistant. You help insurers and investigating officers analyze accident cases, traffic rules, and insurance clauses. Answer concisely and accurately.\n\n"
+        # Load traffic rules as general context
+        ensure_init()
+        rules_data = rule_loader.get_all_rules()
+        rules_text = ""
+        for cat in rules_data.get("categories", []):
+            rules_text += f"\nCategory: {cat.get('name', '')}\n"
+            for r in cat.get("rules", []):
+                rules_text += f"  - {r.get('id', '')}: {r.get('title', '')} (Severity: {r.get('severity', '')})\n"
+        ctx = default_ctx + "TRAFFIC RULES:\n" + rules_text
+        return ctx, "*General mode: traffic rules loaded. Ask any question!*"
+
+    ensure_init()
+    case = run_async(db.get_case(int(case_id)))
+    if not case:
+        return "", f"❌ Case {int(case_id)} not found."
+
+    analyses = run_async(db.get_analyses_by_case(int(case_id)))
+    parties = run_async(db.get_parties_by_case(int(case_id)))
+    violations = run_async(db.get_violations_by_case(int(case_id)))
+    fault = run_async(db.get_fault_analysis(int(case_id)))
+    rules_data = rule_loader.get_all_rules()
+
+    ctx = f"""You are TraceScene AI assistant analyzing Case #{case.get('case_number', '')}.
+Location: {case.get('location', 'Unknown')}
+Date: {case.get('incident_date', 'Unknown')}
+Officer: {case.get('officer_name', 'Unknown')}
+Status: {case.get('status', 'Unknown')}
+
+SCENE ANALYSES:\n"""
+    for a in analyses:
+        ctx += f"\n--- Photo Analysis ---\n{a.get('raw_analysis', '')}\n"
+
+    if parties:
+        ctx += "\nPARTIES IDENTIFIED:\n"
+        for p in parties:
+            ctx += f"  - {p.get('label', '')}: {p.get('vehicle_type', '')} {p.get('vehicle_color', '')} — {p.get('vehicle_description', '')}\n"
+
+    if violations:
+        ctx += "\nVIOLATIONS FOUND:\n"
+        for v in violations:
+            ctx += f"  - {v.get('rule_title', '')} (Severity: {v.get('severity', '')}, Confidence: {v.get('confidence', 0):.0%})\n"
+
+    if fault:
+        ctx += f"\nFAULT ANALYSIS:\n  Primary Fault: {fault.get('primary_fault_party', 'N/A')}\n  Confidence: {fault.get('overall_confidence', 0):.0%}\n  Summary: {fault.get('analysis_summary', '')}\n"
+
+    # Append traffic rules
+    rules_text = ""
+    for cat in rules_data.get("categories", []):
+        rules_text += f"\nCategory: {cat.get('name', '')}\n"
+        for r in cat.get("rules", []):
+            rules_text += f"  - {r.get('id', '')}: {r.get('title', '')} (Severity: {r.get('severity', '')})\n"
+    ctx += "\nTRAFFIC RULES:\n" + rules_text
+
+    return ctx, f"✅ Case **{case.get('case_number', '')}** loaded with {len(analyses)} analyses, {len(violations)} violations."
+
+
+@spaces.GPU(duration=120)
+def chat_respond(user_message, history, system_ctx):
+    if not user_message or not user_message.strip():
+        return history, "", system_ctx
+    ensure_init()
+    if not inference_engine.is_loaded:
+        inference_engine.load_model()
+    try:
+        # Use the vision model's text generation capability for chat
+        chat_prompt = f"""You are TraceScene AI assistant helping with accident analysis.
+
+CONTEXT:
+{system_ctx}
+
+USER QUESTION: {user_message.strip()}
+
+Provide a concise, helpful answer based on the context above."""
+        # Create a small blank image for the vision model
+        from PIL import Image as PILImg
+        blank = PILImg.new('RGB', (64, 64), color=(0, 0, 0))
+        response = gpu_run_inference(blank, chat_prompt)
+    except Exception as e:
+        response = f"Error: {e}"
+    history = history or []
+    history.append((user_message.strip(), response))
+    return history, "", system_ctx
+
+
+def generate_animation_fn(case_id):
+    if not case_id:
+        return "<p style='color:red;'>Enter a Case ID.</p>"
+    ensure_init()
+    analyses = run_async(db.get_analyses_by_case(int(case_id)))
+    if not analyses:
+        return "<p style='color:red;'>No analyses found. Run analysis first.</p>"
+
+    # Parse scene details from the first analysis
+    raw = analyses[0].get("raw_analysis", "")
+
+    def extract_field(text, field):
+        import re
+        pattern = rf"{re.escape(field)}:\s*(.+)"
+        m = re.search(pattern, text, re.IGNORECASE)
+        return m.group(1).strip() if m else "Unknown"
+
+    road_type = extract_field(raw, "Road Type")
+    num_vehicles = extract_field(raw, "Vehicles Involved")
+    v1_pos = extract_field(raw, "Vehicle 1 Position")
+    v1_tyre = extract_field(raw, "Vehicle 1 Tyre Direction")
+    impact = extract_field(raw, "Area of Impact")
+    category = extract_field(raw, "Accident Category")
+    v1_make = extract_field(raw, "Vehicle 1 Make/Model")
+
+    # Check for Vehicle 2
+    v2_pos = extract_field(raw, "Vehicle 2 Position")
+    v2_tyre = extract_field(raw, "Vehicle 2 Tyre Direction")
+    v2_make = extract_field(raw, "Vehicle 2 Make/Model")
+    has_v2 = v2_make != "Unknown"
+
+    # Determine colors from extracted make
+    import re as re_mod
+    def extract_color(make_str):
+        colors = ["Red", "Blue", "White", "Black", "Silver", "Grey", "Green", "Yellow", "Brown", "Orange"]
+        for c in colors:
+            if c.lower() in make_str.lower():
+                return c.lower()
+        return "#3b82f6"
+
+    v1_color = extract_color(v1_make)
+    v2_color = extract_color(v2_make) if has_v2 else "#ef4444"
+
+    # Severity affects animation speed
+    speed_map = {"mild": 1.5, "medium": 2.5, "critical": 4.0}
+    anim_speed = speed_map.get(category.lower(), 2.5)
+
+    # Road layout
+    road_is_intersection = "intersection" in road_type.lower()
+    road_is_highway = "highway" in road_type.lower()
+
+    num_v = 1
+    try:
+        num_v = int(num_vehicles)
+    except:
+        pass
+    # Unique ID to force Gradio to re-render on each click (enables replay)
+    import random
+    uid = random.randint(10000, 99999)
+    # Determine animation duration based on severity
+    dur = "3s" if category.lower() == "mild" else "2s" if category.lower() == "medium" else "1.5s"
+    sev_color = "#22c55e" if category.lower() == "mild" else "#f59e0b" if category.lower() == "medium" else "#ef4444"
+
+    # Build SVG road
+    if road_is_intersection:
+        road_svg = '''
+        <rect x="0" y="160" width="700" height="100" fill="#555"/>
+        <rect x="300" y="0" width="100" height="420" fill="#555"/>
+        <line x1="0" y1="210" x2="300" y2="210" stroke="#fbbf24" stroke-width="2" stroke-dasharray="20,15"/>
+        <line x1="400" y1="210" x2="700" y2="210" stroke="#fbbf24" stroke-width="2" stroke-dasharray="20,15"/>
+        <line x1="350" y1="0" x2="350" y2="160" stroke="#fbbf24" stroke-width="2" stroke-dasharray="20,15"/>
+        <line x1="350" y1="260" x2="350" y2="420" stroke="#fbbf24" stroke-width="2" stroke-dasharray="20,15"/>
+        '''
+    else:
+        road_svg = '''
+        <rect x="0" y="150" width="700" height="120" fill="#555" rx="2"/>
+        <line x1="0" y1="210" x2="700" y2="210" stroke="#fbbf24" stroke-width="2" stroke-dasharray="20,15"/>
+        <line x1="0" y1="150" x2="700" y2="150" stroke="white" stroke-width="2"/>
+        <line x1="0" y1="270" x2="700" y2="270" stroke="white" stroke-width="2"/>
+        '''
+
+    # Vehicle 2 SVG (if present)
+    v2_svg = ""
+    if has_v2:
+        if road_is_intersection:
+            v2_svg = f'''<g>
+              <animateTransform attributeName="transform" type="translate" from="0,0" to="0,135" dur="{dur}" fill="freeze"/>
+              <rect x="325" y="60" width="50" height="26" rx="5" fill="{v2_color}" stroke="#fff" stroke-width="1"/>
+              <text x="350" y="78" fill="white" font-size="10" font-weight="bold" text-anchor="middle">V2</text>
+            </g>'''
+        else:
+            v2_svg = f'''<g>
+              <animateTransform attributeName="transform" type="translate" from="0,0" to="-200,0" dur="{dur}" fill="freeze"/>
+              <rect x="560" y="215" width="50" height="26" rx="5" fill="{v2_color}" stroke="#fff" stroke-width="1"/>
+              <text x="585" y="233" fill="white" font-size="10" font-weight="bold" text-anchor="middle">V2</text>
+            </g>'''
+
+    html = f'''
+<div style="text-align:center; font-family: Inter, Arial, sans-serif;">
+<svg id="anim_{uid}" width="700" height="420" viewBox="0 0 700 420" xmlns="http://www.w3.org/2000/svg" style="border:1px solid #444; border-radius:10px; background:#1a1a2e;">
+  <defs>
+    <radialGradient id="glow_{uid}" cx="50%" cy="50%" r="50%">
+      <stop offset="0%" stop-color="#fbbf24" stop-opacity="0.8"/>
+      <stop offset="100%" stop-color="#fbbf24" stop-opacity="0"/>
+    </radialGradient>
+  </defs>
+
+  {road_svg}
+
+  <!-- Vehicle 1 -->
+  <g>
+    <animateTransform attributeName="transform" type="translate" from="0,0" to="200,0" dur="{dur}" fill="freeze"/>
+    <rect x="80" y="190" width="50" height="26" rx="5" fill="{v1_color}" stroke="#fff" stroke-width="1"/>
+    <text x="105" y="207" fill="white" font-size="10" font-weight="bold" text-anchor="middle">V1</text>
+  </g>
+
+  {v2_svg}
+
+  <!-- Impact flash -->
+  <circle cx="340" cy="210" r="0" fill="url(#glow_{uid})">
+    <animate attributeName="r" values="0;0;0;0;0;0;0;45;55;0" dur="{dur}" fill="freeze"/>
+    <animate attributeName="opacity" values="0;0;0;0;0;0;0;1;0.5;0" dur="{dur}" fill="freeze"/>
+  </circle>
+
+  <!-- Debris -->
+  <circle cx="340" cy="210" r="3" fill="#fbbf24" opacity="0">
+    <animate attributeName="opacity" values="0;0;0;0;0;0;0;1;0" dur="{dur}" fill="freeze"/>
+    <animate attributeName="cx" values="340;340;340;340;340;340;340;310;290" dur="{dur}" fill="freeze"/>
+    <animate attributeName="cy" values="210;210;210;210;210;210;210;185;170" dur="{dur}" fill="freeze"/>
+  </circle>
+  <circle cx="340" cy="210" r="2" fill="#ef4444" opacity="0">
+    <animate attributeName="opacity" values="0;0;0;0;0;0;0;1;0" dur="{dur}" fill="freeze"/>
+    <animate attributeName="cx" values="340;340;340;340;340;340;340;370;395" dur="{dur}" fill="freeze"/>
+    <animate attributeName="cy" values="210;210;210;210;210;210;210;190;175" dur="{dur}" fill="freeze"/>
+  </circle>
+  <circle cx="340" cy="210" r="3" fill="#e2e8f0" opacity="0">
+    <animate attributeName="opacity" values="0;0;0;0;0;0;0;1;0" dur="{dur}" fill="freeze"/>
+    <animate attributeName="cx" values="340;340;340;340;340;340;340;320;305" dur="{dur}" fill="freeze"/>
+    <animate attributeName="cy" values="210;210;210;210;210;210;210;235;255" dur="{dur}" fill="freeze"/>
+  </circle>
+  <circle cx="340" cy="210" r="2" fill="#f97316" opacity="0">
+    <animate attributeName="opacity" values="0;0;0;0;0;0;0;1;0" dur="{dur}" fill="freeze"/>
+    <animate attributeName="cx" values="340;340;340;340;340;340;340;365;385" dur="{dur}" fill="freeze"/>
+    <animate attributeName="cy" values="210;210;210;210;210;210;210;230;250" dur="{dur}" fill="freeze"/>
+  </circle>
+
+  <!-- Collision label -->
+  <text x="350" y="145" fill="#ef4444" font-size="18" font-weight="bold" text-anchor="middle" opacity="0" font-family="Inter, Arial, sans-serif">
+    COLLISION
+    <animate attributeName="opacity" values="0;0;0;0;0;0;0;1;1" dur="{dur}" fill="freeze"/>
+  </text>
+
+  <!-- HUD -->
+  <rect x="10" y="350" width="680" height="60" rx="8" fill="rgba(0,0,0,0.6)"/>
+  <text x="20" y="375" fill="#e2e8f0" font-size="12" font-family="Inter, Arial, sans-serif">{v1_make[:35]}</text>
+  <text x="20" y="398" fill="#e2e8f0" font-size="12" font-family="Inter, Arial, sans-serif">{"" if not has_v2 else v2_make[:35]}{"Single vehicle accident" if not has_v2 else ""}</text>
+  <text x="680" y="375" fill="{sev_color}" font-size="14" font-weight="bold" text-anchor="end" font-family="Inter, Arial, sans-serif">{category.upper()}</text>
+  <text x="680" y="398" fill="#94a3b8" font-size="11" text-anchor="end" font-family="Inter, Arial, sans-serif">Impact: {impact} | Road: {road_type}</text>
+</svg>
+<div style="margin-top:8px; color:#94a3b8; font-size:12px;">
+  Vehicles: {num_v} | Animation auto-plays on load
+</div>
+</div>
+'''
+    return html
+
 # ── Build Gradio App ──────────────────────────────────────────────────
 
 CUSTOM_CSS = """
@@ -548,90 +801,7 @@ with gr.Blocks(
             # State for context
             chat_system_ctx = gr.State(value="You are TraceScene AI assistant. You help insurers and investigating officers analyze accident cases, traffic rules, and insurance clauses. Answer concisely and accurately based on the context provided.")
 
-            def load_chat_context(case_id):
-                if not case_id:
-                    default_ctx = "You are TraceScene AI assistant. You help insurers and investigating officers analyze accident cases, traffic rules, and insurance clauses. Answer concisely and accurately.\n\n"
-                    # Load traffic rules as general context
-                    ensure_init()
-                    rules_data = rule_loader.get_all_rules()
-                    rules_text = ""
-                    for cat in rules_data.get("categories", []):
-                        rules_text += f"\nCategory: {cat.get('name', '')}\n"
-                        for r in cat.get("rules", []):
-                            rules_text += f"  - {r.get('id', '')}: {r.get('title', '')} (Severity: {r.get('severity', '')})\n"
-                    ctx = default_ctx + "TRAFFIC RULES:\n" + rules_text
-                    return ctx, "*General mode: traffic rules loaded. Ask any question!*"
-
-                ensure_init()
-                case = run_async(db.get_case(int(case_id)))
-                if not case:
-                    return "", f"❌ Case {int(case_id)} not found."
-
-                analyses = run_async(db.get_analyses_by_case(int(case_id)))
-                parties = run_async(db.get_parties_by_case(int(case_id)))
-                violations = run_async(db.get_violations_by_case(int(case_id)))
-                fault = run_async(db.get_fault_analysis(int(case_id)))
-                rules_data = rule_loader.get_all_rules()
-
-                ctx = f"""You are TraceScene AI assistant analyzing Case #{case.get('case_number', '')}.
-Location: {case.get('location', 'Unknown')}
-Date: {case.get('incident_date', 'Unknown')}
-Officer: {case.get('officer_name', 'Unknown')}
-Status: {case.get('status', 'Unknown')}
-
-SCENE ANALYSES:\n"""
-                for a in analyses:
-                    ctx += f"\n--- Photo Analysis ---\n{a.get('raw_analysis', '')}\n"
-
-                if parties:
-                    ctx += "\nPARTIES IDENTIFIED:\n"
-                    for p in parties:
-                        ctx += f"  - {p.get('label', '')}: {p.get('vehicle_type', '')} {p.get('vehicle_color', '')} — {p.get('vehicle_description', '')}\n"
-
-                if violations:
-                    ctx += "\nVIOLATIONS FOUND:\n"
-                    for v in violations:
-                        ctx += f"  - {v.get('rule_title', '')} (Severity: {v.get('severity', '')}, Confidence: {v.get('confidence', 0):.0%})\n"
-
-                if fault:
-                    ctx += f"\nFAULT ANALYSIS:\n  Primary Fault: {fault.get('primary_fault_party', 'N/A')}\n  Confidence: {fault.get('overall_confidence', 0):.0%}\n  Summary: {fault.get('analysis_summary', '')}\n"
-
-                # Append traffic rules
-                rules_text = ""
-                for cat in rules_data.get("categories", []):
-                    rules_text += f"\nCategory: {cat.get('name', '')}\n"
-                    for r in cat.get("rules", []):
-                        rules_text += f"  - {r.get('id', '')}: {r.get('title', '')} (Severity: {r.get('severity', '')})\n"
-                ctx += "\nTRAFFIC RULES:\n" + rules_text
-
-                return ctx, f"✅ Case **{case.get('case_number', '')}** loaded with {len(analyses)} analyses, {len(violations)} violations."
-
-            @spaces.GPU(duration=120)
-            def chat_respond(user_message, history, system_ctx):
-                if not user_message or not user_message.strip():
-                    return history, "", system_ctx
-                ensure_init()
-                if not inference_engine.is_loaded:
-                    inference_engine.load_model()
-                try:
-                    # Use the vision model's text generation capability for chat
-                    chat_prompt = f"""You are TraceScene AI assistant helping with accident analysis.
 
-CONTEXT:
-{system_ctx}
-
-USER QUESTION: {user_message.strip()}
-
-Provide a concise, helpful answer based on the context above."""
-                    # Create a small blank image for the vision model
-                    from PIL import Image as PILImg
-                    blank = PILImg.new('RGB', (64, 64), color=(0, 0, 0))
-                    response = gpu_run_inference(blank, chat_prompt)
-                except Exception as e:
-                    response = f"Error: {e}"
-                history = history or []
-                history.append((user_message.strip(), response))
-                return history, "", system_ctx
 
             chat_load_btn.click(load_chat_context, inputs=[chat_case_id], outputs=[chat_system_ctx, chat_context_status])
             chat_send_btn.click(chat_respond, inputs=[chat_input, chatbot, chat_system_ctx], outputs=[chatbot, chat_input, chat_system_ctx], api_name="chat")
@@ -645,176 +815,6 @@ Provide a concise, helpful answer based on the context above."""
             anim_btn = gr.Button("Generate Animation", variant="primary")
             anim_output = gr.HTML(label="Animation View")
 
-            def generate_animation_fn(case_id):
-                if not case_id:
-                    return "<p style='color:red;'>Enter a Case ID.</p>"
-                ensure_init()
-                analyses = run_async(db.get_analyses_by_case(int(case_id)))
-                if not analyses:
-                    return "<p style='color:red;'>No analyses found. Run analysis first.</p>"
-
-                # Parse scene details from the first analysis
-                raw = analyses[0].get("raw_analysis", "")
-
-                def extract_field(text, field):
-                    import re
-                    pattern = rf"{re.escape(field)}:\s*(.+)"
-                    m = re.search(pattern, text, re.IGNORECASE)
-                    return m.group(1).strip() if m else "Unknown"
-
-                road_type = extract_field(raw, "Road Type")
-                num_vehicles = extract_field(raw, "Vehicles Involved")
-                v1_pos = extract_field(raw, "Vehicle 1 Position")
-                v1_tyre = extract_field(raw, "Vehicle 1 Tyre Direction")
-                impact = extract_field(raw, "Area of Impact")
-                category = extract_field(raw, "Accident Category")
-                v1_make = extract_field(raw, "Vehicle 1 Make/Model")
-
-                # Check for Vehicle 2
-                v2_pos = extract_field(raw, "Vehicle 2 Position")
-                v2_tyre = extract_field(raw, "Vehicle 2 Tyre Direction")
-                v2_make = extract_field(raw, "Vehicle 2 Make/Model")
-                has_v2 = v2_make != "Unknown"
-
-                # Determine colors from extracted make
-                import re as re_mod
-                def extract_color(make_str):
-                    colors = ["Red", "Blue", "White", "Black", "Silver", "Grey", "Green", "Yellow", "Brown", "Orange"]
-                    for c in colors:
-                        if c.lower() in make_str.lower():
-                            return c.lower()
-                    return "#3b82f6"
-
-                v1_color = extract_color(v1_make)
-                v2_color = extract_color(v2_make) if has_v2 else "#ef4444"
-
-                # Severity affects animation speed
-                speed_map = {"mild": 1.5, "medium": 2.5, "critical": 4.0}
-                anim_speed = speed_map.get(category.lower(), 2.5)
-
-                # Road layout
-                road_is_intersection = "intersection" in road_type.lower()
-                road_is_highway = "highway" in road_type.lower()
-
-                num_v = 1
-                try:
-                    num_v = int(num_vehicles)
-                except:
-                    pass
-                # Unique ID to force Gradio to re-render on each click (enables replay)
-                import random
-                uid = random.randint(10000, 99999)
-                # Determine animation duration based on severity
-                dur = "3s" if category.lower() == "mild" else "2s" if category.lower() == "medium" else "1.5s"
-                sev_color = "#22c55e" if category.lower() == "mild" else "#f59e0b" if category.lower() == "medium" else "#ef4444"
-
-                # Build SVG road
-                if road_is_intersection:
-                    road_svg = '''
-                    <rect x="0" y="160" width="700" height="100" fill="#555"/>
-                    <rect x="300" y="0" width="100" height="420" fill="#555"/>
-                    <line x1="0" y1="210" x2="300" y2="210" stroke="#fbbf24" stroke-width="2" stroke-dasharray="20,15"/>
-                    <line x1="400" y1="210" x2="700" y2="210" stroke="#fbbf24" stroke-width="2" stroke-dasharray="20,15"/>
-                    <line x1="350" y1="0" x2="350" y2="160" stroke="#fbbf24" stroke-width="2" stroke-dasharray="20,15"/>
-                    <line x1="350" y1="260" x2="350" y2="420" stroke="#fbbf24" stroke-width="2" stroke-dasharray="20,15"/>
-                    '''
-                else:
-                    road_svg = '''
-                    <rect x="0" y="150" width="700" height="120" fill="#555" rx="2"/>
-                    <line x1="0" y1="210" x2="700" y2="210" stroke="#fbbf24" stroke-width="2" stroke-dasharray="20,15"/>
-                    <line x1="0" y1="150" x2="700" y2="150" stroke="white" stroke-width="2"/>
-                    <line x1="0" y1="270" x2="700" y2="270" stroke="white" stroke-width="2"/>
-                    '''
-
-                # Vehicle 2 SVG (if present)
-                v2_svg = ""
-                if has_v2:
-                    if road_is_intersection:
-                        v2_svg = f'''<g>
-                          <animateTransform attributeName="transform" type="translate" from="0,0" to="0,135" dur="{dur}" fill="freeze"/>
-                          <rect x="325" y="60" width="50" height="26" rx="5" fill="{v2_color}" stroke="#fff" stroke-width="1"/>
-                          <text x="350" y="78" fill="white" font-size="10" font-weight="bold" text-anchor="middle">V2</text>
-                        </g>'''
-                    else:
-                        v2_svg = f'''<g>
-                          <animateTransform attributeName="transform" type="translate" from="0,0" to="-200,0" dur="{dur}" fill="freeze"/>
-                          <rect x="560" y="215" width="50" height="26" rx="5" fill="{v2_color}" stroke="#fff" stroke-width="1"/>
-                          <text x="585" y="233" fill="white" font-size="10" font-weight="bold" text-anchor="middle">V2</text>
-                        </g>'''
-
-                html = f'''
-<div style="text-align:center; font-family: Inter, Arial, sans-serif;">
-<svg id="anim_{uid}" width="700" height="420" viewBox="0 0 700 420" xmlns="http://www.w3.org/2000/svg" style="border:1px solid #444; border-radius:10px; background:#1a1a2e;">
-  <defs>
-    <radialGradient id="glow_{uid}" cx="50%" cy="50%" r="50%">
-      <stop offset="0%" stop-color="#fbbf24" stop-opacity="0.8"/>
-      <stop offset="100%" stop-color="#fbbf24" stop-opacity="0"/>
-    </radialGradient>
-  </defs>
-
-  {road_svg}
-
-  <!-- Vehicle 1 -->
-  <g>
-    <animateTransform attributeName="transform" type="translate" from="0,0" to="200,0" dur="{dur}" fill="freeze"/>
-    <rect x="80" y="190" width="50" height="26" rx="5" fill="{v1_color}" stroke="#fff" stroke-width="1"/>
-    <text x="105" y="207" fill="white" font-size="10" font-weight="bold" text-anchor="middle">V1</text>
-  </g>
-
-  {v2_svg}
-
-  <!-- Impact flash -->
-  <circle cx="340" cy="210" r="0" fill="url(#glow_{uid})">
-    <animate attributeName="r" values="0;0;0;0;0;0;0;45;55;0" dur="{dur}" fill="freeze"/>
-    <animate attributeName="opacity" values="0;0;0;0;0;0;0;1;0.5;0" dur="{dur}" fill="freeze"/>
-  </circle>
-
-  <!-- Debris -->
-  <circle cx="340" cy="210" r="3" fill="#fbbf24" opacity="0">
-    <animate attributeName="opacity" values="0;0;0;0;0;0;0;1;0" dur="{dur}" fill="freeze"/>
-    <animate attributeName="cx" values="340;340;340;340;340;340;340;310;290" dur="{dur}" fill="freeze"/>
-    <animate attributeName="cy" values="210;210;210;210;210;210;210;185;170" dur="{dur}" fill="freeze"/>
-  </circle>
-  <circle cx="340" cy="210" r="2" fill="#ef4444" opacity="0">
-    <animate attributeName="opacity" values="0;0;0;0;0;0;0;1;0" dur="{dur}" fill="freeze"/>
-    <animate attributeName="cx" values="340;340;340;340;340;340;340;370;395" dur="{dur}" fill="freeze"/>
-    <animate attributeName="cy" values="210;210;210;210;210;210;210;190;175" dur="{dur}" fill="freeze"/>
-  </circle>
-  <circle cx="340" cy="210" r="3" fill="#e2e8f0" opacity="0">
-    <animate attributeName="opacity" values="0;0;0;0;0;0;0;1;0" dur="{dur}" fill="freeze"/>
-    <animate attributeName="cx" values="340;340;340;340;340;340;340;320;305" dur="{dur}" fill="freeze"/>
-    <animate attributeName="cy" values="210;210;210;210;210;210;210;235;255" dur="{dur}" fill="freeze"/>
-  </circle>
-  <circle cx="340" cy="210" r="2" fill="#f97316" opacity="0">
-    <animate attributeName="opacity" values="0;0;0;0;0;0;0;1;0" dur="{dur}" fill="freeze"/>
-    <animate attributeName="cx" values="340;340;340;340;340;340;340;365;385" dur="{dur}" fill="freeze"/>
-    <animate attributeName="cy" values="210;210;210;210;210;210;210;230;250" dur="{dur}" fill="freeze"/>
-  </circle>
-
-  <!-- Collision label -->
-  <text x="350" y="145" fill="#ef4444" font-size="18" font-weight="bold" text-anchor="middle" opacity="0" font-family="Inter, Arial, sans-serif">
-    COLLISION
-    <animate attributeName="opacity" values="0;0;0;0;0;0;0;1;1" dur="{dur}" fill="freeze"/>
-  </text>
-
-  <!-- HUD -->
-  <rect x="10" y="350" width="680" height="60" rx="8" fill="rgba(0,0,0,0.6)"/>
-  <text x="20" y="375" fill="#e2e8f0" font-size="12" font-family="Inter, Arial, sans-serif">{v1_make[:35]}</text>
-  <text x="20" y="398" fill="#e2e8f0" font-size="12" font-family="Inter, Arial, sans-serif">{"" if not has_v2 else v2_make[:35]}{"Single vehicle accident" if not has_v2 else ""}</text>
-  <text x="680" y="375" fill="{sev_color}" font-size="14" font-weight="bold" text-anchor="end" font-family="Inter, Arial, sans-serif">{category.upper()}</text>
-  <text x="680" y="398" fill="#94a3b8" font-size="11" text-anchor="end" font-family="Inter, Arial, sans-serif">Impact: {impact} | Road: {road_type}</text>
-</svg>
-<div style="margin-top:8px; color:#94a3b8; font-size:12px;">
-  Vehicles: {num_v} | Animation auto-plays on load
-</div>
-</div>
-'''
-                return html
-
-            anim_btn.click(generate_animation_fn, inputs=[anim_case_id], outputs=[anim_output])
-
-
-
             anim_btn.click(generate_animation_fn, inputs=[anim_case_id], outputs=[anim_output])
 
         # Hidden API-only endpoints (for @gradio/client from custom frontend)