Restructure Tabs and Add Header

Author: paty-oliveira

src/ui/components/container.py

@@ -1,4 +1,3 @@
-from .footer import render as footer
 from .homepage import render as homepage
 from .stock import render as stock
 from ..data_providers.FinancialDataProviderFactory import FinancialDataProviderFactory
@@ -12,7 +11,6 @@ def render():
         ticker_name = ticker.get_ticker()
         data_provider = FinancialDataProviderFactory(ticker_name).get_provider()
         stock(data_provider)
-        footer()
 
     else:
         homepage()

src/ui/components/header.py

@@ -0,0 +1,7 @@
+import streamlit as st
+
+
+def render(_data_provider):
+    data = _data_provider.get_stock_info()
+
+    st.markdown(f"### {data['symbol']} - {data['longName']}", unsafe_allow_html=True)

src/ui/components/stock.py

@@ -1,20 +1,9 @@
-import streamlit_shadcn_ui as ui
-
-from .balance_sheet import render as balance_sheet
-from .cashflow import render as cashflow
-from .income_stmt import render as income_stmt
-from .overview import render as overview
-
-tab_content = {
-    "Overview": overview,
-    "Balance Sheet": balance_sheet,
-    "Income Statement": income_stmt,
-    "CashFlow": cashflow,
-}
+from .footer import render as footer
+from .header import render as header
+from .tabs import render as tabs
 
 
 def render(_data_provider):
-    options = list(tab_content.keys())
-    current_tab = ui.tabs(options, default_value=options[0], key="current_tab")
-
-    return tab_content[current_tab](_data_provider)
+    header(_data_provider)
+    tabs(_data_provider)
+    footer()

src/ui/components/tabs.py

@@ -0,0 +1,22 @@
+import streamlit as st
+
+from .balance_sheet import render as balance_sheet
+from .cashflow import render as cashflow
+from .income_stmt import render as income_stmt
+from .overview import render as overview
+
+tab_content = {
+    "Overview": overview,
+    "Balance Sheet": balance_sheet,
+    "Income Statement": income_stmt,
+    "CashFlow": cashflow,
+}
+
+
+def render(_data_provider):
+    options = list(tab_content.keys())
+
+    tabs = st.tabs(options)
+    for tab, (tab_name, content) in zip(tabs, tab_content.items()):
+        with tab:
+            content(_data_provider)