feat: add tradingview charts

df.py

@@ -2,6 +2,7 @@ import pandas as pd
 import pytz
 import cloudscraper
 import yfinance as yf
+from rich import print
 
 from typing import List
 from types import SimpleNamespace
@@ -168,9 +169,21 @@ def fetch_etf_volumes(funds: List[str], start_time=None):
 
 
 def fetch_asset_price(ticker: str, start_time=None):
-    price = yf.download(ticker, interval="1d", period="max", start=start_time)["Close"]
-    price = extract_date_index(price)
-    price.rename(columns={ticker: "Price"}, inplace=True)
+    price = yf.download(ticker, interval="1d", period="max", start=start_time)
+    price = price.droplevel("Ticker", axis=1)
+    price.columns = [c[0] if isinstance(c, list) else c for c in price.columns]
+    price.rename(
+        columns={
+            "Price_Date": "Date",
+            "Open": "open",
+            "High": "high",
+            "Low": "low",
+            "Close": "close",
+            "Volume": "volume",
+        },
+        inplace=True,
+    )
+    price.reset_index(names="Date", inplace=True)
     print(price.head())
 
     return price
@@ -203,8 +216,39 @@ def fetch(asset):
         }
     )
 
+    tdv = f"""
+<!-- TradingView Widget BEGIN -->
+<div class="tradingview-widget-container">
+  <div class="tradingview-widget-container__widget"></div>
+  <div class="tradingview-widget-copyright"><a href="https://www.tradingview.com/" rel="noopener nofollow" target="_blank"><span class="blue-text">Track all markets on TradingView</span></a></div>
+  <script type="text/javascript" src="https://s3.tradingview.com/external-embedding/embed-widget-advanced-chart.js" async>
+  {{
+    "width": "100%",
+    "height": "640",
+    "symbol": "INDEX:{asset}USD",
+    "interval": "D",
+    "timezone": "Etc/UTC",
+    "theme": "dark",
+    "style": "1",
+    "locale": "en",
+    "withdateranges": true,
+    "hide_side_toolbar": false,
+    "allow_symbol_change": true,
+    "details": false,
+    "calendar": false,
+    "studies": [
+      "STD;RSI"
+    ],
+    "support_host": "https://www.tradingview.com"
+  }}
+  </script>
+</div>
+<!-- TradingView Widget END -->
+            """
+
     return SimpleNamespace(
         url=etf_url,
+        tdv=tdv,
         etf_flow=etf_flow,
         etf_volumes=etf_volumes,
         price=price,

pyproject.toml

@@ -10,6 +10,7 @@ dependencies = [
     "lxml>=5.3.1",
     "pandas>=2.2.3",
     "pygwalker>=0.4.9.14",
+    "rich>=13.9.4",
     "streamlit>=1.43.2",
     "vega>=4.1.0",
     "workalendar>=17.0.0",

requirements.txt

@@ -1,9 +0,0 @@
-streamlit
-pandas
-lxml
-yfinance[nospam,repair]==0.2.54
-altair
-vega
-workalendar
-pygwalker
-cloudscraper

streamlit_app.py

@@ -3,7 +3,7 @@
 import pandas as pd
 
 import streamlit as st
-from streamlit.components.v1 import iframe
+from streamlit.components.v1 import iframe, html
 import altair as alt
 
 from pygwalker.api.streamlit import StreamlitRenderer, init_streamlit_comm
@@ -44,7 +44,7 @@ def gen_charts(asset, chart_size={"width": 560, "height": 150}):
                     axis=alt.Axis(tickCount={"interval": "month", "step": 1}),
                     title="",
                 ),
-                y=alt.Y("Price:Q").scale(zero=False),
+                y=alt.Y("close:Q").scale(zero=False),
                 color=alt.value("crimson"),
             )
         )
@@ -275,11 +275,14 @@ def app():
             st.altair_chart(btc_charts, use_container_width=True)
         with eth_chart_col:
             st.altair_chart(eth_charts, use_container_width=True)
+
         # Display iframes
         btc_col, eth_col = st.columns(2)
         with btc_col:
+            html(btc.tdv, height=640)
             iframe(btc.url, height=1200, scrolling=True)
         with eth_col:
+            html(eth.tdv, height=640)
             iframe(eth.url, height=1200, scrolling=True)
     with single_view:
         asset = st.selectbox(
@@ -288,27 +291,29 @@ def app():
         )
         charts = asset_charts(asset, chart_size={"width": "container", "height": 300})
         st.altair_chart(charts, use_container_width=True)
+        match asset:
+            case "BTC":
+                html(btc.tdv, height=640)
+            case "ETH":
+                html(eth.tdv, height=640)
         iframe(fetch_asset(asset).url, height=1200, scrolling=True)
     with flow_tab:
         btc_flow, eth_flow = btc.etf_flow, eth.etf_flow
         btc_flow["Asset"] = "BTC"
         eth_flow["Asset"] = "ETH"
         df = pd.concat([btc_flow, eth_flow])
-        df.Date = df.Date.astype(str)
         StreamlitRenderer(df).explorer()
     with volume_tab:
         btc_volume, eth_volume = btc.etf_volumes, eth.etf_volumes
         btc_volume["Asset"] = "BTC"
         eth_volume["Asset"] = "ETH"
         df = pd.concat([btc_volume, eth_volume])
-        df.Date = df.Date.astype(str)
         StreamlitRenderer(df).explorer()
     with price_tab:
         btc_price, eth_price = btc.price, eth.price
         btc_price["Asset"] = "BTC"
         eth_price["Asset"] = "ETH"
         df = pd.concat([btc_price, eth_price])
-        df.Date = df.Date.astype(str)
         StreamlitRenderer(df).explorer()