Fix formatting

app.py

@@ -7,7 +7,7 @@ import streamlit as st
 from tzlocal import get_localzone
 
 from api import TradeAssistantAPI
-from style import recommendation_table_styler
+from style import portfolio_table_styler, recommendation_table_styler
 
 st.title("📊 Trading Assistant")
 
@@ -46,7 +46,7 @@ try:
     portfolio = portfolio[list(portfolio.keys())[0]]
     portfolio = pd.DataFrame(portfolio)
     st.write("## Portfolio")
-    st.dataframe(portfolio)
+    st.dataframe(portfolio.style.pipe(portfolio_table_styler))
 except Exception as e:
     st.error(e)
     st.expander("Details").write(portfolio)

style.py

@@ -1,12 +1,26 @@
 def recommendation_table_styler(styler):
-    for col in [
+    proba_cols = [
         "proba_strong_up",
         "proba_weak_up",
         "proba_same",
         "proba_weak_down",
         "proba_strong_down",
-    ]:
+    ]
+    for col in proba_cols:
         styler = styler.background_gradient(subset=[col], cmap="viridis_r")
-        styler = styler.format({col: "{:.2%}"})
+    styler = styler.format({col: "{:.2%}" for col in proba_cols})
     styler = styler.background_gradient(subset=["actual"], cmap="viridis")
     return styler
+
+
+def portfolio_table_styler(styler):
+    proba_cols = [
+        "strong up",
+        "weak up",
+        "same",
+        "weak down",
+        "strong down",
+    ]
+    styler = styler.background_gradient(subset=proba_cols, cmap="viridis")
+    styler = styler.format({col: "{:.2f}" for col in proba_cols})
+    return styler