Spaces:

Quant0
/

AFI

Sleeping

AFI / app.py

Create app.py

b876318 verified over 1 year ago

10.9 kB

# ============================================================
# app.py — Interfaz de usuario (Gradio) y visualizaciones (Plotly)
# ------------------------------------------------------------
# - Capa de presentación:
#   * UI con Gradio (componentes y callbacks).
#   * Plot interactivo de la evolución de la correlación (Plotly).
#   * Iframes de bitcointreasuries.
#   * Botones de exportación (delegan en exports.py).
# - La lógica y datos están en:
#   analytics.py, scraping.py, matching.py, data_sources.py, exports.py, config.py
# - Sin estado persistente; la caché en memoria vive en data_sources.py.
# ============================================================

from __future__ import annotations

# --- Librerías UI & Visualización
import gradio as gr
import plotly.graph_objects as go

# --- Núcleo de negocio y datos (módulos del proyecto)
from analytics import evolucion_correlacion
from matching import find_company_online
from scraping import generate_tickers_from_bitcointreasuries
from exports import exportar_correlacion, descargar_resumen_holding_excel
from config import TICKERS_TO_IDS, GRADIO_SHARE
from data_sources import get_data  # precarga opcional de BTC-USD

# ============================================================
# Helpers específicos de la UI
# ============================================================

def get_iframe_stock_vs_btc(ticker: str) -> str:
    """
    HTML del iframe 'Stock vs Bitcoin' para un ticker, usando el mapeo TICKERS_TO_IDS.
    """
    try:
        entity_id = TICKERS_TO_IDS[ticker]
        return f"""
        https://bitcointreasuries.net/embed?component=StockVsBitcoinChart&entityId={entity_id}
        """
    except KeyError:
        return f"No hay datos de holding de Bitcoin para {ticker}."


def get_iframe_holdings(ticker: str) -> str:
    """
    HTML del iframe 'Bitcoin Holdings' para un ticker, usando TICKERS_TO_IDS.
    """
    try:
        entity_id = TICKERS_TO_IDS[ticker]
        return f"""
        https://bitcointreasuries.net/embed?component=BitcoinHoldingsChart&entityId={entity_id}
        """
    except KeyError:
        return f"No hay datos de holding de Bitcoin para {ticker}."


def plot_evolucion_correlacion_gradio(
    ticker_company: str, step: int, window_months: int
):
    """
    Genera un gráfico Plotly con la evolución de la correlación entre `ticker_company` y BTC-USD.
    """
    serie = evolucion_correlacion(ticker_company, step=step, window_months=window_months)

    if serie is None or serie.empty:
        return None, f"No hay datos de correlación disponibles para {ticker_company}."

    # --- Configuración del gráfico Plotly (estética y controles)
    fig = go.Figure()

    fig.add_trace(
        go.Scatter(
            x=serie.index,
            y=serie.values,
            mode="lines",
            name="Correlación",
            line=dict(color="#2263eb", width=2),
            fill="tozeroy",
            fillcolor="rgba(34,99,235,0.1)",
            hovertemplate="Fecha: %{x|%Y-%m-%d}<br>Correlación: %{y:.4f}<extra></extra>",
        )
    )

    # Layout y controles de rango
    fig.update_layout(
        title_text=f"<b>Evolución de la correlación entre {ticker_company} y BTC-USD<b>",
        title_x=0,
        height=500,
        template="plotly_white",
        margin=dict(l=20, r=20, t=60, b=20),
        showlegend=False,
        xaxis=go.layout.XAxis(
            rangeslider=dict(
                visible=True,
                thickness=0.08,
                bgcolor="rgba(50, 0, 128, 0.1)",
                bordercolor="rgba(50, 0, 128, 0.2)",
                borderwidth=1,
                range=[serie.index.min(), serie.index.max()],
            ),
            rangeselector=dict(
                buttons=list(
                    [
                        dict(count=1, label="1m", step="month", stepmode="backward"),
                        dict(count=3, label="3m", step="month", stepmode="backward"),
                        dict(count=6, label="6m", step="month", stepmode="backward"),
                        dict(step="year", count=1, label="YTD", stepmode="todate"),
                        dict(count=1, label="1y", step="year", stepmode="backward"),
                        dict(step="all", label="All"),
                    ]
                ),
                bgcolor="rgba(247,247,247,255)",
                bordercolor="rgba(0,0,0,0)",
                borderwidth=0,
                activecolor="rgba(230,233,255,255)",
                font=dict(color="#000000", size=10),
                x=0.9,
                y=1.1,
                xanchor="right",
                yanchor="top",
            ),
            type="date",
            domain=[0, 1],
            rangeslider_yaxis_rangemode="auto",
            showgrid=False,
            gridcolor="#e0e0e0",
            zeroline=False,
        ),
        yaxis=go.layout.YAxis(
            title_text="Correlación",
            side="left",
            showgrid=True,
            gridcolor="#e0e0e0",
            zeroline=False,
            tickformat=".2f",
            range=[-1.0, 1.0],
        ),
        hovermode="x unified",
        paper_bgcolor="rgba(255,255,255,1)",
        plot_bgcolor="rgba(255,255,255,1)",
    )

    # Anotación en el máximo
    max_value = float(serie.max())
    max_index = serie.idxmax()
    date_str = max_index.date() if hasattr(max_index, "date") else str(max_index)
    fig.add_annotation(
        x=max_index,
        y=max_value,
        text=f"Corr_max: {max_value:.4f}<br>Fecha: {date_str}",
        showarrow=True,
        arrowhead=2,
        ax=0,
        ay=-40,
        bgcolor="rgba(255, 255, 255, 0.8)",
        bordercolor="#c7c7c7",
        borderwidth=1,
        borderpad=4,
        font=dict(size=10, color="#333"),
    )

    return fig, "Gráfico de correlación generado."


# ============================================================
# Datos iniciales para la UI
# ============================================================

# Lista inicial del Dropdown desde bitcointreasuries (equivale a tu generar_tickers()).
tickers_dropdown = generate_tickers_from_bitcointreasuries()

# Precarga rápida de BTC-USD en caché con yfinance (opcional, sin clave).
get_data("BTC-USD", silent=True)


# ============================================================
# Construcción de la interfaz (Gradio Blocks)
# ============================================================

with gr.Blocks(title="Análisis de Correlación") as demo:
    # --- Cabecera y branding
    gr.Image(
        "https://cdn.prod.website-files.com/67174dcb1d43b06a0eae317e/672bbbc7a2b88045e5661994_logo-afi.svg",
        show_label=False,
    )
    gr.Markdown("# Analizador de Correlación de Bitcoin")
    gr.Markdown("Explora la correlación de diferentes activos con Bitcoin y sus holdings.")

    # --- Buscador de empresa/ticker
    with gr.Row():
        input_company = gr.Dropdown(
            label="Buscar por Nombre (se usarán los primeros 6 caracteres)",
            choices=tickers_dropdown,
            value="MicroStrategy",
            allow_custom_value=True,
            interactive=True,
        )
        texto_info1 = gr.Textbox(label="Nombre Completo:", interactive=False)
        # Estado del TICKER resuelto (lo usamos en los callbacks)
        texto_info2 = gr.Textbox(label="Ticker:", interactive=False)

        # Resolver (Nombre, Ticker) al cambiar la selección
        input_company.change(
            fn=lambda texto: find_company_online(texto[:6]),
            inputs=input_company,
            outputs=[texto_info1, texto_info2],
        )
        # Resolver también al cargar la app para mostrar valores iniciales
        demo.load(
            fn=lambda texto: find_company_online(texto[:6]),
            inputs=input_company,
            outputs=[texto_info1, texto_info2],
        )

    # --- Tabs de la aplicación
    with gr.Tab("Evolución de Correlación"):
        with gr.Row():
            input_step = gr.Slider(
                minimum=1,
                maximum=30,
                value=1,
                step=1,
                label="Indica cada cuántos días se calcula la correlación",
            )
            input_window = gr.Slider(
                minimum=1,
                maximum=60,
                value=12,
                step=1,
                label="Ventana rolling (meses)",
            )

        btn_plot = gr.Button("Mostrar Gráfico")
        output_plot = gr.Plot(label="Gráfico de Evolución")
        btn_excel = gr.Button("Descargar en Excel")
        output_file = gr.File(label="Archivo Excel")
        output_msg = gr.Label()

        # Graficar (usa el ticker resuelto en texto_info2)
        btn_plot.click(
            fn=plot_evolucion_correlacion_gradio,
            inputs=[texto_info2, input_step, input_window],
            outputs=[output_plot, output_msg],
        )

        # Exportar a Excel (serie de correlación)
        btn_excel.click(
            fn=exportar_correlacion,
            inputs=[texto_info2, input_step, input_window],
            outputs=[output_file, output_msg],
        )

    with gr.Tab("Stock vs BTC"):
        btn_stock_btc = gr.Button("Cargar Gráfico")
        output_iframe_stock = gr.HTML(label="Stock vs Bitcoin Chart")
        btn_stock_btc.click(
            fn=get_iframe_stock_vs_btc, inputs=[texto_info2], outputs=[output_iframe_stock]
        )

    with gr.Tab("Holding de Bitcoin"):
        btn_holdings = gr.Button("Cargar Gráfico")
        output_iframe_holdings = gr.HTML(label="Bitcoin Holdings Chart")
        btn_holdings.click(
            fn=get_iframe_holdings, inputs=[texto_info2], outputs=[output_iframe_holdings]
        )

    with gr.Tab("Exportar Resumen Holding"):
        input_template_url = gr.Textbox(
            label="URL de la plantilla de Excel (Google Drive)",
            value=(
                "https://docs.google.com/spreadsheets/d/1Ad3bbYQAAZO6ljkaCGTGSw5aG_FQF7Lg/"
                "edit?usp=sharing&ouid=109222615757657219494&rtpof=true&sd=true"
            ),
            placeholder="Ej: https://docs.google.com/spreadsheets/d/ID_DEL_ARCHIVO/...",
        )
        btn_descargar_resumen = gr.Button("Descargar Resumen a Excel")
        output_resumen_file = gr.File(label="Archivo de Resumen de Holdings")
        output_resumen_msg = gr.Label()

        btn_descargar_resumen.click(
            fn=descargar_resumen_holding_excel,
            inputs=[input_template_url],
            outputs=[output_resumen_file, output_resumen_msg],
        )

# ============================================================
# Lanzamiento de la app
# ============================================================

if __name__ == "__main__":
    # GRADIO_SHARE controla si se expone URL pública (útil para demos/HF Spaces)
    demo.launch(share=GRADIO_SHARE)

	# ============================================================
	# app.py — Interfaz de usuario (Gradio) y visualizaciones (Plotly)
	# ------------------------------------------------------------
	# - Capa de presentación:
	# * UI con Gradio (componentes y callbacks).
	# * Plot interactivo de la evolución de la correlación (Plotly).
	# * Iframes de bitcointreasuries.
	# * Botones de exportación (delegan en exports.py).
	# - La lógica y datos están en:
	# analytics.py, scraping.py, matching.py, data_sources.py, exports.py, config.py
	# - Sin estado persistente; la caché en memoria vive en data_sources.py.
	# ============================================================

	from __future__ import annotations

	# --- Librerías UI & Visualización
	import gradio as gr
	import plotly.graph_objects as go

	# --- Núcleo de negocio y datos (módulos del proyecto)
	from analytics import evolucion_correlacion
	from matching import find_company_online
	from scraping import generate_tickers_from_bitcointreasuries
	from exports import exportar_correlacion, descargar_resumen_holding_excel
	from config import TICKERS_TO_IDS, GRADIO_SHARE
	from data_sources import get_data # precarga opcional de BTC-USD

	# ============================================================
	# Helpers específicos de la UI
	# ============================================================

	def get_iframe_stock_vs_btc(ticker: str) -> str:
	"""
	HTML del iframe 'Stock vs Bitcoin' para un ticker, usando el mapeo TICKERS_TO_IDS.
	"""
	try:
	entity_id = TICKERS_TO_IDS[ticker]
	return f"""
	https://bitcointreasuries.net/embed?component=StockVsBitcoinChart&entityId={entity_id}
	"""
	except KeyError:
	return f"No hay datos de holding de Bitcoin para {ticker}."


	def get_iframe_holdings(ticker: str) -> str:
	"""
	HTML del iframe 'Bitcoin Holdings' para un ticker, usando TICKERS_TO_IDS.
	"""
	try:
	entity_id = TICKERS_TO_IDS[ticker]
	return f"""
	https://bitcointreasuries.net/embed?component=BitcoinHoldingsChart&entityId={entity_id}
	"""
	except KeyError:
	return f"No hay datos de holding de Bitcoin para {ticker}."


	def plot_evolucion_correlacion_gradio(
	ticker_company: str, step: int, window_months: int
	):
	"""
	Genera un gráfico Plotly con la evolución de la correlación entre `ticker_company` y BTC-USD.
	"""
	serie = evolucion_correlacion(ticker_company, step=step, window_months=window_months)

	if serie is None or serie.empty:
	return None, f"No hay datos de correlación disponibles para {ticker_company}."

	# --- Configuración del gráfico Plotly (estética y controles)
	fig = go.Figure()

	fig.add_trace(
	go.Scatter(
	x=serie.index,
	y=serie.values,
	mode="lines",
	name="Correlación",
	line=dict(color="#2263eb", width=2),
	fill="tozeroy",
	fillcolor="rgba(34,99,235,0.1)",
	hovertemplate="Fecha: %{x\|%Y-%m-%d}<br>Correlación: %{y:.4f}<extra></extra>",
	)
	)

	# Layout y controles de rango
	fig.update_layout(
	title_text=f"<b>Evolución de la correlación entre {ticker_company} y BTC-USD<b>",
	title_x=0,
	height=500,
	template="plotly_white",
	margin=dict(l=20, r=20, t=60, b=20),
	showlegend=False,
	xaxis=go.layout.XAxis(
	rangeslider=dict(
	visible=True,
	thickness=0.08,
	bgcolor="rgba(50, 0, 128, 0.1)",
	bordercolor="rgba(50, 0, 128, 0.2)",
	borderwidth=1,
	range=[serie.index.min(), serie.index.max()],
	),
	rangeselector=dict(
	buttons=list(
	[
	dict(count=1, label="1m", step="month", stepmode="backward"),
	dict(count=3, label="3m", step="month", stepmode="backward"),
	dict(count=6, label="6m", step="month", stepmode="backward"),
	dict(step="year", count=1, label="YTD", stepmode="todate"),
	dict(count=1, label="1y", step="year", stepmode="backward"),
	dict(step="all", label="All"),
	]
	),
	bgcolor="rgba(247,247,247,255)",
	bordercolor="rgba(0,0,0,0)",
	borderwidth=0,
	activecolor="rgba(230,233,255,255)",
	font=dict(color="#000000", size=10),
	x=0.9,
	y=1.1,
	xanchor="right",
	yanchor="top",
	),
	type="date",
	domain=[0, 1],
	rangeslider_yaxis_rangemode="auto",
	showgrid=False,
	gridcolor="#e0e0e0",
	zeroline=False,
	),
	yaxis=go.layout.YAxis(
	title_text="Correlación",
	side="left",
	showgrid=True,
	gridcolor="#e0e0e0",
	zeroline=False,
	tickformat=".2f",
	range=[-1.0, 1.0],
	),
	hovermode="x unified",
	paper_bgcolor="rgba(255,255,255,1)",
	plot_bgcolor="rgba(255,255,255,1)",
	)

	# Anotación en el máximo
	max_value = float(serie.max())
	max_index = serie.idxmax()
	date_str = max_index.date() if hasattr(max_index, "date") else str(max_index)
	fig.add_annotation(
	x=max_index,
	y=max_value,
	text=f"Corr_max: {max_value:.4f}<br>Fecha: {date_str}",
	showarrow=True,
	arrowhead=2,
	ax=0,
	ay=-40,
	bgcolor="rgba(255, 255, 255, 0.8)",
	bordercolor="#c7c7c7",
	borderwidth=1,
	borderpad=4,
	font=dict(size=10, color="#333"),
	)

	return fig, "Gráfico de correlación generado."


	# ============================================================
	# Datos iniciales para la UI
	# ============================================================

	# Lista inicial del Dropdown desde bitcointreasuries (equivale a tu generar_tickers()).
	tickers_dropdown = generate_tickers_from_bitcointreasuries()

	# Precarga rápida de BTC-USD en caché con yfinance (opcional, sin clave).
	get_data("BTC-USD", silent=True)


	# ============================================================
	# Construcción de la interfaz (Gradio Blocks)
	# ============================================================

	with gr.Blocks(title="Análisis de Correlación") as demo:
	# --- Cabecera y branding
	gr.Image(
	"https://cdn.prod.website-files.com/67174dcb1d43b06a0eae317e/672bbbc7a2b88045e5661994_logo-afi.svg",
	show_label=False,
	)
	gr.Markdown("# Analizador de Correlación de Bitcoin")
	gr.Markdown("Explora la correlación de diferentes activos con Bitcoin y sus holdings.")

	# --- Buscador de empresa/ticker
	with gr.Row():
	input_company = gr.Dropdown(
	label="Buscar por Nombre (se usarán los primeros 6 caracteres)",
	choices=tickers_dropdown,
	value="MicroStrategy",
	allow_custom_value=True,
	interactive=True,
	)
	texto_info1 = gr.Textbox(label="Nombre Completo:", interactive=False)
	# Estado del TICKER resuelto (lo usamos en los callbacks)
	texto_info2 = gr.Textbox(label="Ticker:", interactive=False)

	# Resolver (Nombre, Ticker) al cambiar la selección
	input_company.change(
	fn=lambda texto: find_company_online(texto[:6]),
	inputs=input_company,
	outputs=[texto_info1, texto_info2],
	)
	# Resolver también al cargar la app para mostrar valores iniciales
	demo.load(
	fn=lambda texto: find_company_online(texto[:6]),
	inputs=input_company,
	outputs=[texto_info1, texto_info2],
	)

	# --- Tabs de la aplicación
	with gr.Tab("Evolución de Correlación"):
	with gr.Row():
	input_step = gr.Slider(
	minimum=1,
	maximum=30,
	value=1,
	step=1,
	label="Indica cada cuántos días se calcula la correlación",
	)
	input_window = gr.Slider(
	minimum=1,
	maximum=60,
	value=12,
	step=1,
	label="Ventana rolling (meses)",
	)

	btn_plot = gr.Button("Mostrar Gráfico")
	output_plot = gr.Plot(label="Gráfico de Evolución")
	btn_excel = gr.Button("Descargar en Excel")
	output_file = gr.File(label="Archivo Excel")
	output_msg = gr.Label()

	# Graficar (usa el ticker resuelto en texto_info2)
	btn_plot.click(
	fn=plot_evolucion_correlacion_gradio,
	inputs=[texto_info2, input_step, input_window],
	outputs=[output_plot, output_msg],
	)

	# Exportar a Excel (serie de correlación)
	btn_excel.click(
	fn=exportar_correlacion,
	inputs=[texto_info2, input_step, input_window],
	outputs=[output_file, output_msg],
	)

	with gr.Tab("Stock vs BTC"):
	btn_stock_btc = gr.Button("Cargar Gráfico")
	output_iframe_stock = gr.HTML(label="Stock vs Bitcoin Chart")
	btn_stock_btc.click(
	fn=get_iframe_stock_vs_btc, inputs=[texto_info2], outputs=[output_iframe_stock]
	)

	with gr.Tab("Holding de Bitcoin"):
	btn_holdings = gr.Button("Cargar Gráfico")
	output_iframe_holdings = gr.HTML(label="Bitcoin Holdings Chart")
	btn_holdings.click(
	fn=get_iframe_holdings, inputs=[texto_info2], outputs=[output_iframe_holdings]
	)

	with gr.Tab("Exportar Resumen Holding"):
	input_template_url = gr.Textbox(
	label="URL de la plantilla de Excel (Google Drive)",
	value=(
	"https://docs.google.com/spreadsheets/d/1Ad3bbYQAAZO6ljkaCGTGSw5aG_FQF7Lg/"
	"edit?usp=sharing&ouid=109222615757657219494&rtpof=true&sd=true"
	),
	placeholder="Ej: https://docs.google.com/spreadsheets/d/ID_DEL_ARCHIVO/...",
	)
	btn_descargar_resumen = gr.Button("Descargar Resumen a Excel")
	output_resumen_file = gr.File(label="Archivo de Resumen de Holdings")
	output_resumen_msg = gr.Label()

	btn_descargar_resumen.click(
	fn=descargar_resumen_holding_excel,
	inputs=[input_template_url],
	outputs=[output_resumen_file, output_resumen_msg],
	)

	# ============================================================
	# Lanzamiento de la app
	# ============================================================

	if __name__ == "__main__":
	# GRADIO_SHARE controla si se expone URL pública (útil para demos/HF Spaces)
	demo.launch(share=GRADIO_SHARE)