Update app.py

app.py

@@ -867,7 +867,20 @@ def handle(user_msg: str, files: list, yield_update) -> str:
                 total_duration_ms=_duration_ms
             )
 
-            return _sanitize_text(final_report)
+            # Append code traceability section
+            # "Every finding traces back to specific lines of generated Python code"
+            traceability_section = (
+                f"\n\n---\n\n"
+                f"<details>\n"
+                f"<summary>📜 <strong>View Analysis Code</strong> (click to expand)</summary>\n\n"
+                f"The findings in this report were generated by the following Python code, "
+                f"executed in a sandboxed environment:\n\n"
+                f"```python\n{analysis_script}\n```\n\n"
+                f"**Session ID:** `{session_id}`\n\n"
+                f"</details>"
+            )
+
+            return _sanitize_text(final_report) + traceability_section
         else:
             chat_input = redacted_in if (PHI_MODE and not ALLOW_EXTERNAL_PHI) else safe_in
             prompt = f"{GENERAL_CONVERSATION_PROMPT}\n\nUser: {chat_input}\nAssistant:"