fix(env): Ahora aceptamos variables de entorno para las API KEYS

src/app.py

@@ -1,3 +1,5 @@
+import os
+
 import streamlit as st
 from incident_crew import IncidentReporterCrew
 
@@ -10,7 +12,10 @@ st.set_page_config(
 
 # Inicializar el estado de la sesión para la API Key si no existe
 if 'gemini_api_key' not in st.session_state:
-    st.session_state['gemini_api_key'] = ''
+    st.session_state['gemini_api_key'] = os.getenv("GEMINI_API_KEY", "")
+
+if 'ipinfo_api_key' not in st.session_state:
+    st.session_state['ipinfo_api_key'] = os.getenv("IPINFO_API_KEY", "")
 
 
 st.title("📄 Generador IA de Informes Post-Mortem")
@@ -20,19 +25,32 @@ with st.sidebar:
     st.header("⚙️ Instrucciones")
 
     # Campo de entrada para la API Key
-    st.session_state['gemini_api_key'] = st.text_input(
-        "1. 🔑 Ingresa tu API Key de Gemini",
-        type="password",
-        value=st.session_state['gemini_api_key']
-    )
+    if not os.getenv("GEMINI_API_KEY"):
+        st.session_state['gemini_api_key'] = st.text_input(
+            "- 🔑 Ingresa tu API Key de Gemini",
+            type="password",
+            value=st.session_state['gemini_api_key']
+        )
     
     if st.session_state['gemini_api_key']:
-        st.success("API Key cargada. ¡Puedes generar el informe!")
+        st.success("Gemini API Key cargada!")
     else:
         st.warning("🚨 Por favor, ingresa tu API Key de Gemini.")
+    
+    if not os.getenv("IPINFO_API_KEY"):
+        st.session_state['ipinfo_api_key'] = st.text_input(
+            "- 🔑 Ingresa tu API Key de IPInfo",
+            type="password",
+            value=st.session_state['ipinfo_api_key']
+        )
+
+    if st.session_state['ipinfo_api_key']:
+        st.success("IPInfo API Key cargada!")
+    else:
+        st.warning("🚨 Por favor, ingresa tu API Key de IPInfo.")
 
-    st.write("2. 📄 Completa el formulario con los detalles de la incidencia")
-    st.write("3. 🚀 Presiona el botón y la IA generará un informe técnico estructurado post-mortem de tu incidente.")
+    st.write("- 📄 Completa el formulario con los detalles de la incidencia")
+    st.write("- 🚀 Presiona el botón y la IA generará un informe técnico estructurado post-mortem de tu incidente.")
 
 
 # --- Formulario de entrada de datos principal ---

src/custom_tools.py

@@ -1,8 +1,8 @@
-import os
 import requests
 from crewai.tools import BaseTool
 from typing import Type, List
 from pydantic import BaseModel, Field
+import streamlit as st
 
 # --- 1. IPInfo Geo Lookup Tool ---
 class IPInfoToolInput(BaseModel):
@@ -19,24 +19,27 @@ class IPInfoGeoLookup(BaseTool):
     args_schema: Type[BaseModel] = IPInfoToolInput
 
     def _run(self, target: List[str]) -> str:
-        api_key = os.getenv("IPINFO_APIKEY")
+        api_key = st.session_state['ipinfo_api_key']
         if not api_key:
             return "Error: La variable de entorno IPINFO_APIKEY no está configurada."
 
         results = []
+
         for ip_address in target:
             try:
                 url = f"https://api.ipinfo.io/lite/{ip_address}?token={api_key}"
                 response = requests.get(url)
-                response.raise_for_status()  # Lanza una excepción para respuestas de error (4xx o 5xx)
+                response.raise_for_status()
                 data = response.json()
                 # Formateamos la salida para que sea legible y útil para el agente
                 result_str = (
                     f"IP: {data.get('ip')}, Country: {data.get('country')}, ASN: {data.get('as_name')}"
                 )
                 results.append(result_str)
+
             except requests.exceptions.RequestException as e:
                 results.append(f"Error al consultar la IP {ip_address}: {e}")
+                
         return "\n".join(results)
 
 # --- 2. VirusTotal Scanner Tool ---