Update app/dashboard.py

app/dashboard.py

@@ -1,189 +1,142 @@
-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 RUTAS Y RED ---
-BACKEND_URL = os.getenv("BACKEND_URL", "http://vertex-backend:8010")
-WATCHLIST_FILE = "app/watchlist.json"
-
-def load_watchlist():
-    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)
-
-# --- MOTOR DE REPORTES PDF ---
-class VertexReport(FPDF):
-    def header(self):
-        self.set_font('Arial', 'B', 15)
-        self.cell(0, 10, 'VERTEX CODERS LLC - AUDIT REPORT', 0, 1, 'C')
-        self.ln(10)
-
-def generate_pdf(data):
-    pdf = VertexReport()
-    pdf.set_auto_page_break(auto=True, margin=15)
-    pdf.add_page()
-    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)
-    pdf.multi_cell(0, 8, msg)
-    pdf.ln(5)
-    pdf.set_font("Arial", 'B', 11)
-    pdf.cell(0, 10, "2. FINANCIAL METRICS", 0, 1)
-    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')
-
-# --- 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("Quantum Risk Analysis Platform | Enterprise Edition")
-
-# 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:
-    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:
-                r = requests.get(f"{BACKEND_URL}/audit/{selected_ticker}", timeout=25)
-                r.raise_for_status()
-                st.session_state.last_audit = r.json()
-                st.rerun()
-            except Exception as e:
-                st.error(f"🔌 Connection Failure: {e}")
-
-    if "last_audit" in st.session_state:
-        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=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 col_r:
-            st.subheader("🧠 Semantic Analysis")
-            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("🔗 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:
-                    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.")
+import streamlit as st
+import requests
+import plotly.graph_objects as go
+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"
+
+def load_watchlist():
+    if os.path.exists(WATCHLIST_FILE):
+        with open(WATCHLIST_FILE, "r") as f:
+            return json.load(f)
+    return ["INTC", "TSLA", "AAPL"]
+
+def save_watchlist(watchlist):
+    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 ---
+class VertexReport(FPDF):
+    def header(self):
+        self.set_font('Arial', 'B', 15)
+        self.cell(0, 10, 'VERTEX CODERS LLC - AUDIT REPORT', 0, 1, 'C')
+        self.ln(10)
+
+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)
+    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)
+    pdf.cell(0, 8, f"- Altman Z-Score: {float(z_score):.2f}", 0, 1)
+    return pdf.output(dest='S')
+
+# --- MAIN UI ---
+st.title("🛡️ Vertex Risk Terminal")
+st.caption("Enterprise-Level Analysis Platform | Némesis Engine v1.0")
+
+tab1, tab2 = st.tabs(["📈 Stock Audit", "🔗 Web3 Audit (Rug Pull Detect)"])
+
+with tab1:
+    run_audit = st.button("🚀 RUN FULL STOCK AUDIT", type="primary")
+    
+    if run_audit:
+        with st.spinner(f"🔍 Analyzing {selected_ticker}..."):
+            try:
+                # Usamos la constante BACKEND_URL definida arriba
+                r = requests.get(f"{BACKEND_URL}/audit/{selected_ticker}", timeout=30)
+                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()
+            except Exception as e:
+                st.error(f"🔌 Error conectando al búnker: {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)
+            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"}]}
+            ))
+            st.plotly_chart(fig, use_container_width=True)
+
+        with col2:
+            st.subheader("🧠 Semantic Analysis")
+            st.write(data.get("semantic_analysis", {}).get("summary", "No summary available."))
+
+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..."):
+                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}")