Update app.py

app.py

@@ -57,7 +57,10 @@ def preprocess(text):
     
 def highlight_survived(s):
     return ['background-color: red']*len(s) if (s.Sexista == 1) else ['background-color: green']*len(s)
-                
+
+def color_survived(val):
+    color = 'red' if val==0 else 'yellow' if val==1 else 'green'
+    return f'background-color: {color}'                
 
 st.set_page_config(layout="wide")
 st.markdown('<style>body{background-color: Blue;}</style>',unsafe_allow_html=True)
@@ -158,7 +161,11 @@ def run():
                 df = pd.DataFrame(list(zip(tweet_list, flat_predictions)),columns =['Últimos '+ str(number_of_tweets)+' Tweets'+' de '+search_words, 'Sexista'])
                 df['Sexista']= np.where(df['Sexista']== 0, 'No Sexista', 'Sexista')
                 
-                st.dataframe(df.style.apply(highlight_survived, axis=1))
+                
+                st.table(df.reset_index(drop=True).head(20).style.applymap(color_survived, subset=['Sexista']))
+
+                
+                #st.dataframe(df.style.apply(highlight_survived, axis=1))
                 #st.table(df)
             #st.write(df)
 run()