Update app.py

app.py

@@ -586,12 +586,38 @@ with gr.Blocks() as demo:
 
     with gr.Tab("Threat Readiness"):
         gr.Markdown("## Threat Readiness — color-coded commander brief (administrative only)")
+
+        # Hidden states to capture answers from other tabs
+        std_state = gr.State()
+        a_state = gr.State()
+        b_state = gr.State()
+        c_state = gr.State()
+
         threat_button = gr.Button("Evaluate Threat Readiness")
         threat_output = gr.Textbox(label="Threat Readiness & Admin Diagnostics", lines=28)
-        threat_button.click(fn=lambda *args: evaluate_threat_brief(*args),
-                            inputs=std_inputs + a_inputs + b_inputs + c_inputs,
-                            outputs=threat_output)
 
+        # Capture answers whenever reports are generated
+        def cache_std(*answers):
+            return list(answers)
+        std_button.click(cache_std, inputs=std_inputs, outputs=[std_state])
+
+        def cache_enh(*answers):
+            la, lb, lc = len(ENH_SECTION_A), len(ENH_SECTION_B), len(ENH_SECTION_C)
+            a_vals = answers[:la]
+            b_vals = answers[la:la+lb]
+            c_vals = answers[la+lb:la+lb+lc]
+            return a_vals, b_vals, c_vals
+        enh_button.click(cache_enh, inputs=a_inputs+b_inputs+c_inputs+[gate_input],
+                         outputs=[a_state, b_state, c_state])
+
+        def threat_runner(std_ans, a_ans, b_ans, c_ans):
+            # flatten all answers
+            all_vals = (std_ans or []) + (a_ans or []) + (b_ans or []) + (c_ans or [])
+            return evaluate_threat_brief(*all_vals)
+
+        threat_button.click(threat_runner, inputs=[std_state, a_state, b_state, c_state], outputs=threat_output)
+
+     
     gr.Markdown("**Safety note:** This tool is restricted to administrative/doctrinal outputs. It will refuse to provide any operational, tactical, or targeting information. If such content is requested or appears in model output, it will be redacted and replaced with a non-actionable administrative alternative.")
 
 if __name__ == "__main__":