Update app.py

app.py

@@ -281,12 +281,12 @@ def bloombergViz(att, val, numblocks, score_templates_df, flip=False):
     # numUnFilled = numblocks - numFilled
 
     res = ""
-    print("score templates df", score_templates_df)
-    for sentence in score_templates_df.loc[(score_templates_df['Attribute'] == att) & (score_templates_df['Stereotyped'] == 'yes')]:
-        print("sentence", sentence)
-        res += f"<div style='height:20px;width:20px;background-color:#555;display:inline-block' id='filled'><span class='tooltiptext' style='color:#FFF'>{sentence['Template']}</span></div> "
-    for sentence in score_templates_df.loc[(score_templates_df['Attribute'] == att) & (score_templates_df['Stereotyped'] == 'no')]:
-        res += f"<div style='height:20px;width:20px;background-color:#999;display:inline-block' id='empty'><span class='tooltiptext' style='color:#FFF'>{sentence['Template']}</span></div> "
+    dfy = score_templates_df.loc[(score_templates_df['Attribute'] == att) & (score_templates_df['Stereotyped'] == 'yes')]
+    dfn = score_templates_df.loc[(score_templates_df['Attribute'] == att) & (score_templates_df['Stereotyped'] == 'no')] 
+    for i in range(len(dfy.index))
+        res += f"<div style='height:20px;width:20px;background-color:#555;display:inline-block' id='filled'><span class='tooltiptext' style='color:#FFF'>{dfy.iloc[i, 3]}</span></div> "
+    for i in range(len(dfn.index))
+        res += f"<div style='height:20px;width:20px;background-color:#999;display:inline-block' id='empty'><span class='tooltiptext' style='color:#FFF'>{dfy.iloc[i, 3]}</span></div> "
     return res
     # if flip:
     #     return numFilled * unfilled + numUnFilled * filled;