Update app.py

app.py

@@ -2,90 +2,129 @@ import streamlit as st
 
 st.set_page_config(layout="wide")
 
-st.title("AI Model Auditing Strategy Simulator")
-
-st.markdown(
-"""
-Select actions taken by the **Model Owner** and the **Auditor**.  
-The dashboard will update to estimate:
-
-- Probability the model owner can cheat the audit
-- Cost of auditing
-- Level of openness required
-"""
-)
+st.title("AI Model Fairness Auditing")
 
 # -------------------------
 # Strategy Definitions
 # -------------------------
 
 model_owner_actions = {
-    "Fine-tuning / overfitting on auditing test": {
-        "cheating": 25,
-        "cost": 0,
-        "openness": -5
-    },
-    "Fine-tuning on auditing-like data distribution": {
-        "cheating": 20,
-        "cost": 0,
-        "openness": -3
-    },
-    "Training a special audit-behavior mode": {
-        "cheating": 30,
-        "cost": 5,
-        "openness": -10
-    },
-    "Withholding training details": {
-        "cheating": 15,
-        "cost": 0,
-        "openness": -15
-    }
+    "Fine-tuning / overfitting on auditing test": {"cheating": 25, "cost": 0, "openness": -5},
+    "Fine-tuning on auditing-like data distribution": {"cheating": 20, "cost": 0, "openness": -3},
+    "Training a special audit-behavior mode": {"cheating": 30, "cost": 5, "openness": -10},
+    "Withholding training details": {"cheating": 15, "cost": 0, "openness": -15}
 }
 
 auditor_actions = {
-    "Audit on one dataset": {
-        "cheating": 20,
-        "cost": 10,
-        "openness": 5
-    },
-    "Audit on multiple datasets": {
-        "cheating": -15,
-        "cost": 25,
-        "openness": 10
-    },
-    "Request internal model access": {
-        "cheating": -25,
-        "cost": 30,
-        "openness": 30
-    },
-    "Request training data documentation": {
-        "cheating": -10,
-        "cost": 5,
-        "openness": 20
-    }
+    "Audit on one dataset": {"cheating": 20, "cost": 10, "openness": 5},
+    "Audit on multiple datasets": {"cheating": -15, "cost": 25, "openness": 10},
+    "Request internal model access": {"cheating": -25, "cost": 30, "openness": 30},
+    "Request training data documentation": {"cheating": -10, "cost": 5, "openness": 20}
 }
 
+# -------------------------
+# Session State
+# -------------------------
+
+if "owner_selected" not in st.session_state:
+    st.session_state.owner_selected = []
+
+if "auditor_selected" not in st.session_state:
+    st.session_state.auditor_selected = []
+
+# -------------------------
+# Toggle functions
+# -------------------------
+
+def toggle_owner(action):
+    if action in st.session_state.owner_selected:
+        st.session_state.owner_selected.remove(action)
+    else:
+        st.session_state.owner_selected.append(action)
+
+def toggle_auditor(action):
+    if action in st.session_state.auditor_selected:
+        st.session_state.auditor_selected.remove(action)
+    else:
+        st.session_state.auditor_selected.append(action)
+
 # -------------------------
 # Layout
 # -------------------------
 
 left, right = st.columns(2)
 
+# -------------------------
+# Model Owner Panel
+# -------------------------
+
 with left:
-    st.header("Model Owner Actions")
+    st.header("Model Owner")
+
+    st.subheader("Selected Actions")
+
+    selected_box = st.container(border=True)
 
-    selected_owner = st.multiselect(
-        "Choose actions:",
-        list(model_owner_actions.keys())
-    )
+    with selected_box:
+        for action in st.session_state.owner_selected:
+            st.button(
+                action,
+                key=f"owner_selected_{action}",
+                on_click=toggle_owner,
+                args=(action,),
+                use_container_width=True
+            )
+
+    st.subheader("Available Actions")
+
+    available_box = st.container(border=True)
+
+    with available_box:
+        for action in model_owner_actions:
+            if action not in st.session_state.owner_selected:
+                st.button(
+                    action,
+                    key=f"owner_available_{action}",
+                    on_click=toggle_owner,
+                    args=(action,),
+                    use_container_width=True
+                )
+
+# -------------------------
+# Auditor Panel
+# -------------------------
 
 with right:
-    st.header("Auditor Actions")
+    st.header("Auditor")
+
+    st.subheader("Selected Actions")
+
+    selected_box = st.container(border=True)
+
+    with selected_box:
+        for action in st.session_state.auditor_selected:
+            st.button(
+                action,
+                key=f"auditor_selected_{action}",
+                on_click=toggle_auditor,
+                args=(action,),
+                use_container_width=True
+            )
 
-    selected_auditor = st.multiselect(
-        "Choose actions:",
-        list(auditor_actions.keys())
-    )
+    st.subheader("Available Actions")
+
+    available_box = st.container(border=True)
+
+    with available_box:
+        for action in auditor_actions:
+            if action not in st.session_state.auditor_selected:
+                st.button(
+                    action,
+                    key=f"auditor_available_{action}",
+                    on_click=toggle_auditor,
+                    args=(action,),
+                    use_container_width=True
+                )
 
 # -------------------------
 # Score Calculation
@@ -95,12 +134,12 @@ cheating_score = 50
 audit_cost = 0
 openness_required = 0
 
-for action in selected_owner:
+for action in st.session_state.owner_selected:
     cheating_score += model_owner_actions[action]["cheating"]
     audit_cost += model_owner_actions[action]["cost"]
     openness_required += model_owner_actions[action]["openness"]
 
-for action in selected_auditor:
+for action in st.session_state.auditor_selected:
     cheating_score += auditor_actions[action]["cheating"]
     audit_cost += auditor_actions[action]["cost"]
     openness_required += auditor_actions[action]["openness"]
@@ -113,27 +152,13 @@ openness_required = max(0, openness_required)
 # -------------------------
 
 st.divider()
+
 st.subheader("Audit Outcome Dashboard")
 
 col1, col2, col3 = st.columns(3)
 
-col1.metric(
-    "Chance of Cheating",
-    f"{cheating_score}%"
-)
-
-col2.metric(
-    "Audit Cost",
-    audit_cost
-)
-
-col3.metric(
-    "Openness Required",
-    openness_required
-)
-
-# -------------------------
-# Optional Visualization
-# -------------------------
+col1.metric("Chance of Cheating", f"{cheating_score}%")
+col2.metric("Audit Cost", audit_cost)
+col3.metric("Openness Required", openness_required)
 
 st.progress(cheating_score / 100)