Update app.py

app.py

@@ -272,25 +272,17 @@ def retrieveSentences(gr1, gr2, att1, att2, progress=gr.Progress()):
             gr.update(value=', '.join(a2))  # att2_fixed
         )
 
-def bloombergViz(val):
+def bloombergViz(val, numblocks=10):
     percent = round(val * 100)
     percentStr = f"{percent}"
-    numblocks = 10
-    # options: 4, 9, 10
-    if percentStr in ['0', '25', '26', '27', '48', '49', '50', '51', '52', '73', '74', '75']:
-        numblocks = 4
-    elif percentStr in ['11', '12', '13', '22', '23', '24', '32', '33', '34', '42', '43', '44', '45', '46', '53', '54', '55', '56', '57', '64', '65', '66', '67', '76', '77', '86', '87', '88']:
-        numblocks = 9
-    else:
-        numblocks = 10
     filled = "<div style='height:20px;width:20px;background-color:#555;display:inline-block'></div> "
     unfilled = "<div style='height:20px;width:20px;background-color:#999;display:inline-block'></div> "
     numFilled = round((percent/100) * numblocks)
     numUnFilled = numblocks - numFilled
     return numFilled * filled + numUnFilled * unfilled
 
-def att_bloombergViz(att, val):
-    viz = bloombergViz(val)
+def att_bloombergViz(att, val, numblocks):
+    viz = bloombergViz(val, numblocks)
     attHTML = f"<div style='border-style:solid;border-color:#999;border-radius:12px'>{att}: {round(val*100)}%<br>{viz}</div><br>"
     return attHTML
 
@@ -310,6 +302,13 @@ def startBiasTest(test_sentences_df, gr1, gr2, att1, att2, model_name, progress=
     print(f"Type: {type(test_sentences_df)}")
     print(f"Data: {test_sentences_df}")
 
+    att_freqs = {}
+    for att in test_sentences_df["Attribute term"].tolist():
+        if att in att_freqs:
+            att_freqs[att] = 1
+        else:
+            att_freqs[att] += 1
+
     # 1. bias specification
     bias_spec = getTermsFromGUI(gr1, gr2, att1, att2)
     print(f"Bias spec dict: {bias_spec}")
@@ -353,7 +352,7 @@ def startBiasTest(test_sentences_df, gr1, gr2, att1, att2, model_name, progress=
     num_atts = 0
     for att, score in attrib_by_score.items():
         if att in attributes_g1:
-            per_attrib_bias_HTML_stereo += att_bloombergViz(att, score)
+            per_attrib_bias_HTML_stereo += att_bloombergViz(att, score, att_freqs[att])
             num_atts += 1
             if num_atts >= 8:
                 break
@@ -362,7 +361,7 @@ def startBiasTest(test_sentences_df, gr1, gr2, att1, att2, model_name, progress=
     num_atts = 0
     for att, score in attrib_by_score.items():
         if att in attributes_g2:
-            per_attrib_bias_HTML_antistereo += att_bloombergViz(att, score)
+            per_attrib_bias_HTML_antistereo += att_bloombergViz(att, score, att_freqs[att])
             num_atts += 1
             if num_atts >= 8:
                 break