Adapt to API changes

api.py

@@ -52,3 +52,14 @@ class TradeAssistantAPI:
  return body
  except JSONDecodeError:
  raise Exception(response.text)
+
+ def get_portfolio(self):
+ url = f"{self.url}/portfolio"
+ response = requests.get(url, headers=self.base_headers)
+ try:
+ body = response.json()
+ if "message" in body:
+ raise Exception(body["message"])
+ return body
+ except JSONDecodeError:
+ raise Exception(response.text)

app.py

@@ -40,6 +40,16 @@ except Exception as e:
  st.error(e)
  st.expander("Details").write(recommendation_res)
 
+try:
+ portfolio_res = api.get_portfolio()
+ portfolio = json.loads(portfolio_res["portfolio"])
+ portfolio = pd.DataFrame(portfolio)
+ st.write("## Portfolio")
+ st.dataframe(portfolio)
+except Exception as e:
+ st.error(e)
+ st.expander("Details").write(portfolio_res)
+
 st.sidebar.write("## Actions")
 schedule_training_res = None
 if st.sidebar.button("Train model"):

style.py

@@ -1,5 +1,5 @@
 def recommendation_table_styler(styler):
- styler = styler.background_gradient(subset=["metric"], cmap="viridis_r")
- styler = styler.background_gradient(subset=["pred_change"], cmap="viridis")
- styler = styler.format({"pred_change": "{:.2%}", "metric": "{:.2f}"})
+ styler = styler.background_gradient(subset=["proba_strong_up"], cmap="viridis_r")
+ styler = styler.background_gradient(subset=["actual"], cmap="viridis")
+ styler = styler.format({"proba_strong_up": "{:.2%}", "actual": "{:.2%}"})
  return styler