Add Balance Sheet Overview table

Author: paty-oliveira

src/infra/yahoo.py

@@ -8,6 +8,8 @@
 
 
 class YahooFinance:
+    NUMBER_ROWS = 4
+
     @staticmethod
     def is_valid_ticker(ticker: str) -> bool:
         stock = yf.Ticker(ticker)
@@ -46,23 +48,9 @@ def get_balance_sheet(
         )
 
         try:
-            fields = [
-                "Current Assets",
-                "Current Liabilities",
-                "Total Assets",
-                "Total Liabilities Net Minority Interest",
-                "Total Capitalization",
-                "Working Capital",
-                "Invested Capital",
-                "Total Debt",
-                "Net Debt",
-                "Stockholders Equity",
-            ]
-
             df_balance_sheet = balance_sheet.transpose()
-            df_balance_sheet = df_balance_sheet.reindex(columns=fields, fill_value=0)
 
-            return df_balance_sheet.sort_index().tail(4)
+            return df_balance_sheet.sort_index().tail(YahooFinance.NUMBER_ROWS)
         except Exception as e:
             logging.error(f"Error fetching balance sheet for {stock}: {e}")
 
@@ -96,7 +84,7 @@ def get_income_statement(
             df_income_stmt = income_statement.transpose()
             df_income_stmt = df_income_stmt.reindex(columns=fields, fill_value=0)
 
-            return df_income_stmt.sort_index().tail(4)
+            return df_income_stmt.sort_index().tail(YahooFinance.NUMBER_ROWS)
 
         except Exception as e:
             logging.error(f"Error fetching income statement for {stock}: {e}")
@@ -122,7 +110,7 @@ def get_cashflow(
             df_cashflow = cashflow.transpose()
             df_cashflow = df_cashflow.reindex(columns=fields, fill_value=0)
 
-            return df_cashflow.sort_index().tail(4)
+            return df_cashflow.sort_index().tail(YahooFinance.NUMBER_ROWS)
 
         except Exception as e:
             logging.error(f"Error fetching cash flow for {stock}: {e}")

src/ui/layout.py

@@ -377,6 +377,13 @@ def render_balance_sheet(financial_data, financial_calculations):
             )
             st.plotly_chart(fig, use_container_width=True)
 
+    st.divider()
+
+    with st.container():
+        st.markdown("#### Balance Sheet Overview")
+        st.write("Hover over the table to download it as a CSV file")
+        st.dataframe(balance_sheet)
+
 
 def render_income_stmt(financial_data, *kwargs):
     income_stmt = financial_data.get_income_statement(