Update app/dashboard.py

app/dashboard.py

@@ -1,49 +1,30 @@
 import streamlit as st
 import requests
 import plotly.graph_objects as go
+import plotly.express as px
+import pandas as pd
 from fpdf import FPDF
 import json
 import os
 
-# --- CONFIGURACIÓN DE URL (EL PUENTE HACIA TU BACKEND) ---
-# Cambiamos el localhost por tu URL real de Hugging Face
-BACKEND_URL = "https://denisijcu-vertex-risk-engine.hf.space"
-
-# --- PERSISTENCE CONFIG ---
-WATCHLIST_FILE = "watchlist.json"
+# --- CONFIGURACIÓN DE RUTAS Y RED ---
+BACKEND_URL = os.getenv("BACKEND_URL", "http://vertex-backend:8010")
+WATCHLIST_FILE = "app/watchlist.json"
 
 def load_watchlist():
-    if os.path.exists(WATCHLIST_FILE):
-        with open(WATCHLIST_FILE, "r") as f:
-            return json.load(f)
-    return ["INTC", "TSLA", "AAPL"]
+    try:
+        if os.path.exists(WATCHLIST_FILE):
+            with open(WATCHLIST_FILE, "r") as f:
+                return json.load(f)
+    except Exception: pass
+    return ["INTC", "TSLA", "AAPL", "SAVE"]
 
 def save_watchlist(watchlist):
+    os.makedirs(os.path.dirname(WATCHLIST_FILE), exist_ok=True)
     with open(WATCHLIST_FILE, "w") as f:
         json.dump(watchlist, f)
 
-# --- CONFIG ---
-st.set_page_config(
-    page_title="Vertex Risk Terminal | Némesis Engine",
-    page_icon="🛡️",
-    layout="wide"
-)
-
-if "watchlist" not in st.session_state:
-    st.session_state.watchlist = load_watchlist()
-
-# --- SIDEBAR: WATCHLIST & ADD ---
-st.sidebar.title("🏢 Stock Watchlist")
-new_ticker = st.sidebar.text_input("Add Company Ticker").upper()
-if st.sidebar.button("➕ Add"):
-    if new_ticker and new_ticker not in st.session_state.watchlist:
-        st.session_state.watchlist.append(new_ticker)
-        save_watchlist(st.session_state.watchlist)
-        st.sidebar.success(f"✅ {new_ticker} saved!")
-
-selected_ticker = st.sidebar.selectbox("Analyze Company", st.session_state.watchlist)
-
-# --- PDF GENERATOR ---
+# --- MOTOR DE REPORTES PDF ---
 class VertexReport(FPDF):
     def header(self):
         self.set_font('Arial', 'B', 15)
@@ -54,89 +35,155 @@ def generate_pdf(data):
     pdf = VertexReport()
     pdf.set_auto_page_break(auto=True, margin=15)
     pdf.add_page()
-    pdf.set_text_color(0, 0, 0) 
     pdf.set_font("Arial", 'B', 14)
     pdf.set_fill_color(151, 231, 225) 
     pdf.cell(0, 12, f"AUDIT REPORT: {data.get('ticker', 'N/A')}", 1, 1, 'C', fill=True)
     pdf.ln(5)
     pdf.set_font("Arial", 'B', 12)
     pdf.cell(0, 10, f"FINAL STATUS: {data.get('status', 'UNKNOWN')}", 0, 1)
+    semantic = data.get("semantic_analysis", {})
+    msg = str(semantic.get('summary', data.get('msg', 'No data'))).encode('latin-1', 'replace').decode('latin-1')
     pdf.ln(5)
     pdf.set_font("Arial", 'B', 11)
     pdf.cell(0, 10, "1. EXECUTIVE VERDICT", 0, 1)
     pdf.set_font("Arial", size=10)
-    msg = str(data.get('msg', 'No data')).encode('latin-1', 'replace').decode('latin-1')
     pdf.multi_cell(0, 8, msg)
     pdf.ln(5)
     pdf.set_font("Arial", 'B', 11)
     pdf.cell(0, 10, "2. FINANCIAL METRICS", 0, 1)
-    pdf.set_font("Arial", size=10)
-    z_score = data.get("numeric_analysis", {}).get("altman_z", 0)
+    z_score = data.get("numeric_analysis", {}).get("altman_z") or data.get("z_score", 0)
     pdf.cell(0, 8, f"- Altman Z-Score: {float(z_score):.2f}", 0, 1)
     return pdf.output(dest='S')
 
-# --- MAIN UI ---
+# --- CONFIGURACIÓN DE LA UI ---
+st.set_page_config(page_title="Vertex Risk Terminal | Némesis", page_icon="🛡️", layout="wide")
+
+if "watchlist" not in st.session_state:
+    st.session_state.watchlist = load_watchlist()
+
+# --- SIDEBAR ---
+st.sidebar.title("🏢 Stock Watchlist")
+new_ticker = st.sidebar.text_input("Add Ticker").upper()
+
+if st.sidebar.button("➕ Add"):
+    if new_ticker and new_ticker not in st.session_state.watchlist:
+        st.session_state.watchlist.append(new_ticker)
+        save_watchlist(st.session_state.watchlist)
+        st.sidebar.success(f"✅ {new_ticker} saved!")
+        st.rerun()
+
+selected_ticker = st.sidebar.selectbox("Analyze Company", st.session_state.watchlist)
+
+st.sidebar.divider()
+st.sidebar.subheader("📡 Bunker Status")
+def check_health(url):
+    try: return "🟢 ONLINE" if requests.get(url, timeout=2).status_code == 200 else "🔴 OFFLINE"
+    except: return "🔴 OFFLINE"
+
+st.sidebar.write(f"Backend Engine: {check_health(BACKEND_URL + '/docs')}")
+
+# --- CUERPO PRINCIPAL ---
 st.title("🛡️ Vertex Risk Terminal")
-st.caption("Enterprise-Level Analysis Platform | Némesis Engine v1.0")
+st.caption("Quantum Risk Analysis Platform | Enterprise Edition")
 
-tab1, tab2 = st.tabs(["📈 Stock Audit", "🔗 Web3 Audit (Rug Pull Detect)"])
+# REPARACIÓN DE TABS: Declaración única de las 4 pestañas
+tab1, tab2, tab3, tab4, tab5 = st.tabs(["📈 Stock Audit", "🔗 Web3 Audit", "🔍 Auditoría Individual", "📊 Comparativa Vertex", "⚙️ Settings"])
 
 with tab1:
-    run_audit = st.button("🚀 RUN FULL STOCK AUDIT", type="primary")
-    
-    if run_audit:
-        with st.spinner(f"🔍 Analyzing {selected_ticker}..."):
+    if st.button("🚀 RUN FULL STOCK AUDIT", type="primary", use_container_width=True):
+        with st.spinner(f"Auditing {selected_ticker} through Némesis Engine..."):
             try:
-                # Usamos la constante BACKEND_URL definida arriba
-                r = requests.get(f"{BACKEND_URL}/audit/{selected_ticker}", timeout=30)
+                r = requests.get(f"{BACKEND_URL}/audit/{selected_ticker}", timeout=25)
                 r.raise_for_status()
-                data = r.json()
-                if data.get("status") == "ERROR":
-                    st.error(f"❌ {data.get('msg', 'Unknown error')}")
-                else:
-                    st.session_state.last_audit = data
-                    st.rerun()
+                st.session_state.last_audit = r.json()
+                st.rerun()
             except Exception as e:
-                st.error(f"🔌 Error conectando al búnker: {e}")
+                st.error(f"🔌 Connection Failure: {e}")
 
     if "last_audit" in st.session_state:
-        data = st.session_state.last_audit
-        st.success("✅ Audit completed!")
-        col1, col2 = st.columns(2)
-        with col1:
-            status = data.get("status", "UNKNOWN")
-            st.metric("FINAL STATUS", f"🛡️ {status}")
-            st.info(data.get("msg", "No message"))
-            
-            z = data.get("numeric_analysis", {}).get("altman_z", 0)
+        res = st.session_state.last_audit
+        st.divider()
+        col_l, col_r = st.columns(2)
+        with col_l:
+            st.metric("FINAL STATUS", res.get("status", "UNKNOWN"))
+            z_val = res.get("numeric_analysis", {}).get("altman_z") or res.get("z_score", 0)
             fig = go.Figure(go.Indicator(
                 mode="gauge+number",
-                value=z,
-                gauge={'axis': {'range': [0, 5]}, 'steps': [
-                    {'range': [0, 1.8], 'color': "red"},
-                    {'range': [1.8, 3], 'color': "yellow"},
-                    {'range': [3, 5], 'color': "green"}]}
-            ))
+                value=float(z_val),
+                gauge={'axis': {'range': [0, 5]},
+                       'steps': [{'range': [0, 1.1], 'color': "lightcoral"},
+                                 {'range': [1.1, 2.9], 'color': "lightyellow"},
+                                 {'range': [2.9, 5], 'color': "lightgreen"}]}))
+            fig.update_layout(height=300)
             st.plotly_chart(fig, use_container_width=True)
-
-        with col2:
+        with col_r:
             st.subheader("🧠 Semantic Analysis")
-            st.write(data.get("semantic_analysis", {}).get("summary", "No summary available."))
+            sem = res.get("semantic_analysis", {})
+            st.info(sem.get("summary", res.get("msg", "No additional data.")))
+        st.download_button("📥 DOWNLOAD PDF REPORT", generate_pdf(res), f"Vertex_{selected_ticker}.pdf", "application/pdf", use_container_width=True)
 
 with tab2:
-    st.subheader("Smart Contract Security Scanner")
-    contract_address = st.text_input("Enter Ethereum Contract Address (0x...)")
-    
-    if st.button("🔍 SCAN SMART CONTRACT"):
-        if contract_address:
-            with st.spinner("Scanning..."):
+    st.subheader("🔗 Web3 Smart Contract Scanner")
+    contract = st.text_input("Dirección del Token (0x...)")
+    if st.button("🔍 SCAN WEB3 ASSET", use_container_width=True):
+        if contract:
+            with st.spinner("Escaneando seguridad..."):
                 try:
-                    # Corregido el error de concatenación que tenías
-                    r = requests.get(f"{BACKEND_URL}/audit_contract/{contract_address}", timeout=60)
-                    contract_data = r.json()
-                    st.warning(f"Audit Status: {contract_data['status']}")
-                    if contract_data.get("vulnerabilities"):
-                        for v in contract_data["vulnerabilities"]:
-                            st.write(f"- 🚩 {v['pattern']}: {v['risk']}")
-                except Exception as e:
-                    st.error(f"Connection Error: {e}")
+                    r = requests.get(f"{BACKEND_URL}/audit_contract/{contract}", timeout=120)
+                    res_w3 = r.json()
+                    st.divider()
+                    status_w3 = res_w3.get("status", "UNKNOWN")
+                    if status_w3 == "SAFE": st.success(f"✅ STATUS: {status_w3}")
+                    elif status_w3 == "DANGER": st.error(f"🚨 STATUS: {status_w3}")
+                    
+                    vulns = res_w3.get("vulnerabilities", [])
+                    if vulns:
+                        for v in vulns: st.error(f"**{v['description']}**")
+                    with st.expander("Ver Código Fuente"):
+                        st.code(res_w3.get("source_preview", ""), language='solidity')
+                except Exception as e: st.error(f"Error: {e}")
+
+with tab3:
+    st.subheader("🔍 Auditoría Individual")
+    st.write(f"Vigilancia activa sobre: **{selected_ticker}**")
+    st.info("Este módulo utiliza análisis heurístico para reportes rápidos.")
+
+with tab4:
+    st.subheader("📊 Comparativa de Salud Financiera")
+    comparison_list = st.multiselect("Compañías:", options=st.session_state.watchlist, default=st.session_state.watchlist[:3])
+    if st.button("📊 GENERAR COMPARATIVA", use_container_width=True):
+        comp_data = []
+        with st.spinner("Calculando ranking..."):
+            for t in comparison_list:
+                try:
+                    r = requests.get(f"{BACKEND_URL}/audit/{t}", timeout=10)
+                    if r.status_code == 200:
+                        res = r.json()
+                        z = res.get("numeric_analysis", {}).get("altman_z") or res.get("z_score", 0)
+                        comp_data.append({"Ticker": t, "Z-Score": float(z)})
+                except: continue
+        if comp_data:
+            df = pd.DataFrame(comp_data)
+            fig_bar = px.bar(df, x='Ticker', y='Z-Score', color='Z-Score', color_continuous_scale=['red', 'yellow', 'green'], range_y=[0, 5])
+            fig_bar.add_hline(y=1.1, line_dash="dash", line_color="red")
+            fig_bar.add_hline(y=2.9, line_dash="dash", line_color="green")
+            st.plotly_chart(fig_bar, use_container_width=True)
+
+with tab5:
+    st.header("⚙️ Configuración del Sistema")
+    st.info("Configura las credenciales de Telegram para que Némesis te envíe alertas automáticas.")
+    
+    # Cargar configuraciones actuales si existen
+    if "settings" not in st.session_state:
+        st.session_state.settings = {"bot_token": "", "chat_id": ""}
+
+    with st.form("settings_form"):
+        bot_token = st.text_input("Telegram Bot Token", value=st.session_state.settings["bot_token"], type="password", help="El token que te dio BotFather")
+        chat_id = st.text_input("Telegram Chat ID", value=st.session_state.settings["chat_id"], help="Tu ID de usuario o el del grupo")
+        
+        if st.form_submit_button("💾 Guardar Configuración"):
+            st.session_state.settings = {"bot_token": bot_token, "chat_id": chat_id}
+            # Aquí guardaríamos en un archivo que n8n vigile
+            with open("app/settings.json", "w") as f:
+                json.dump(st.session_state.settings, f)
+            st.success("✅ Configuración guardada. n8n ahora usará estas credenciales.")