Update app.py

app.py

@@ -13,8 +13,8 @@ with center:
 # --------------------------------------------------
 
 model_owner_actions = {
-    "Overfit to Audit Dataset": "overfit",
-    "Overfit to Audit Dataset-like Distribution": "overfitdist",
+    "Overfit to One Audit Dataset": "overfit",
+    "Overfit to Audit Dataset-like Distributions": "overfitdist",
 }
 
 auditor_actions = {
@@ -150,7 +150,7 @@ auditor_cost = "🟢 Low"
 if "auditmany" in auditor_active_list:
     auditor_cost = "🟡 Medium"
 if "internals" in auditor_active_list:
-    auditor_cost = "🔴 High"
+    auditor_cost = "🔴 Very High"
 
 owner_cost = "None"
 if "overfit" in owner_active_list:
@@ -158,47 +158,25 @@ if "overfit" in owner_active_list:
 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"]
-    openness += model_owner_actions[action]["open"]
-
-for action in st.session_state.auditor_selected:
-    auditor_cost += auditor_actions[action]["cost"]
-    reliability -= auditor_actions[action]["cheating"]
-    openness += auditor_actions[action]["open"]
-
-reliability = max(0, min(100, reliability))
-openness = max(0, openness)
-
-# --------------------------------------------------
-# Convert to Levels
-# --------------------------------------------------
-
-def level(score):
-    if score < 30:
-        return "Low"
-    elif score < 60:
-        return "Medium"
-    else:
-        return "High"
+reliability = "🔴 Low"
+if "auditone" in auditor_active_list and len(owner_active_list)==0:
+    reliability = "🟡 Medium"
+if "auditmany" in auditor_active_list and len(owner_active_list)==0:
+    reliability = "🟢 High"
+if "auditmany" in auditor_active_list and "overfit" in owner_active_list:
+    reliability = "🟡 Medium"
+if "private" in auditor_active_list and "overfitdist" not in owner_active_list:
+    reliability = "🟡 Medium"
+if "private" in auditor_active_list and "auditmany" in auditor_active_list and "overfitdist" not in owner_active_list:
+    reliability = "🟢 High"
+if "private" in auditor_active_list and "auditmany" in auditor_active_list and "overfitdist" in owner_active_list:
+    reliability = "🟡 Medium"
+if "internals" in auditor_active_list:
+    reliability = "🟢 High"
 
-indicator_map = {
-    "Low": "🟢 Low",
-    "Medium": "🟡 Medium",
-    "High": "🔴 High"
-}
+openness = "None"
+if "internals" in auditor_active_list:
+    openness = "🟢 High (Model Weight Sharing)"
 
 # --------------------------------------------------
 # Dashboard
@@ -209,12 +187,12 @@ st.divider()
 c1, c2, c3, c4 = st.columns(4)
 
 c1.metric(
-    "Cost for Auditor",
+    "Auditign Cost",
     indicator_map[level(auditor_cost)]
 )
 
 c2.metric(
-    "Cost for Model Owner",
+    "Cost to Cheat for Model Owner",
     indicator_map[level(owner_cost)]
 )