Update app.py

app.py

@@ -2,24 +2,26 @@ import streamlit as st
 
 st.set_page_config(layout="wide")
 
-st.title("AI Model Fairness Auditing")
+margin1, center, margin2 = st.columns([1,9,1])
+
+with center:
+
+    st.title("AI Model Fairness Auditing")
 
 # --------------------------------------------------
 # Strategy Definitions
 # --------------------------------------------------
 
 model_owner_actions = {
-    "Fine-tuning / overfitting on auditing test": {"cheating": 25, "cost": 0, "open": -5},
-    "Fine-tuning on auditing-like distribution": {"cheating": 20, "cost": 0, "open": -3},
-    "Training special audit behaviour": {"cheating": 30, "cost": 5, "open": -10},
-    "Withholding training details": {"cheating": 15, "cost": 0, "open": -15},
+    "Overfit to Audit Dataset": "overfit",
+    "Overfit to Audit Dataset-like Distribution": "overfitdist",
 }
 
 auditor_actions = {
-    "Audit on one dataset": {"cheating": 20, "cost": 10, "open": 5},
-    "Audit on multiple datasets": {"cheating": -15, "cost": 25, "open": 10},
-    "Request internal model access": {"cheating": -25, "cost": 30, "open": 30},
-    "Request training documentation": {"cheating": -10, "cost": 5, "open": 20},
+    "Audit with One Dataset": "auditone",
+    "Audit with Multiple Datasets": "auditmany",
+    "Use Private Auditing Datasets": "private",
+    "Audit Model Internals": "internals",
 }
 
 # --------------------------------------------------
@@ -135,12 +137,38 @@ with right:
 # Score Calculation
 # --------------------------------------------------
 
-auditor_cost = 0
-owner_cost = 0
-reliability = 50
-openness = 0
+auditor_active_list = []
+for action in st.session_state.auditor_selected:
+    auditor_active_list.append(auditor_actions[action])
+owner_active_list = []
+for action in st.session_state.owner_selected:
+    owner_active_list.append(owner_actions[action])
+
+
+
+auditor_cost = "🟢 Low"
+if "auditmany" in auditor_active_list:
+    auditor_cost = "🟡 Medium"
+if "internals" in auditor_active_list:
+    auditor_cost = "🔴 High"
+
+owner_cost = "None"
+if "overfit" in owner_active_list:
+    owner_cost = "🟢 Low"
+if "overfitdist" in owner_active_list:
+    owner_cost = "🟡 Medium"
+
+reliability
+openness
+
 
+owner_current = "none"
 for action in st.session_state.owner_selected:
+    if model_owner_actions[action]=="overfit" and owner_current = "none":
+        owner_current = "overfit"
+    if model_owner_actions[action]=="overfitdist" and owner_current = "none":
+        owner_current = "overfit"
+    
     auditor_cost += model_owner_actions[action]["cost"]
     owner_cost += 10  # simple baseline cost for owner strategies
     reliability -= model_owner_actions[action]["cheating"]