Add Google Gemini integration and enhance post-mortem report generation UI

requirements.txt

@@ -1,3 +1,4 @@
 altair
 pandas
-streamlit
+streamlit
+google-genai

src/streamlit_app.py

@@ -1,40 +1,148 @@
-import altair as alt
-import numpy as np
-import pandas as pd
 import streamlit as st
+from google import genai
+import os
 
-"""
-# Welcome to Streamlit!
-
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
-
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
-
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+# --- Interfaz de Usuario con Streamlit ---
+st.set_page_config(
+    page_title="Generador de Post-Mortem (Gemini AI)",
+    layout="wide"
+)
+
+# 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.title("📄 Generador de Informes Post-Mortem con Gemini AI")
+st.markdown("Ingresa los detalles de la incidencia y la IA generará un informe técnico estructurado.")
+
+# --- Sidebar para la configuración de la API Key ---
+with st.sidebar:
+    st.header("⚙️ Configuración")
+    
+    # Campo de entrada para la 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!")
+    else:
+        st.warning("🚨 Por favor, ingresa tu API Key de Gemini.")
+
+
+# --- Función para generar el Post-Mortem ---
+def generar_post_mortem(api_key, tipo_alerta, sistema, fecha_hora, impacto, acciones):
+    """
+    Construye el prompt, inicializa el cliente con la API Key y genera el informe.
+    """
+    MODEL_NAME = "gemini-2.5-flash"
+    
+    # 1. Inicializar el cliente con la clave ingresada
+    try:
+        client = genai.Client(api_key=api_key)
+    except Exception as e:
+        return f"🚨 Error al inicializar el cliente de Gemini: {e}"
+
+    # 2. Definir la instrucción (System Instruction)
+    system_instruction = (
+        "Eres un analista técnico de IT. Tu tarea es generar un informe "
+        "Post-Mortem conciso y profesional, estructurado con las secciones: "
+        "'Resumen Ejecutivo', 'Qué Sucedió (Timeline)', 'Análisis de Causa Raíz', "
+        "'Impacto del Negocio' y 'Acciones Correctivas (Plan de Acción)'. "
+        "Usa Markdown (##) para los encabezados y un tono técnico y formal. "
+        "No añadas introducciones ni conclusiones fuera del formato solicitado."
+    )
+
+    # 3. Construir el contenido del prompt
+    prompt_content = f"""
+    Genera un informe Post-Mortem usando la siguiente información de la incidencia:
+    - **Tipo de Alerta/Problema Principal**: {tipo_alerta}
+    - **Sistema Afectado**: {sistema}
+    - **Fecha y Hora de Inicio**: {fecha_hora}
+    - **Impacto Reportado**: {impacto}
+    - **Acciones Inmediatas Tomadas**: {acciones}
+    """
+
+    try:
+        # Llamar a la API de Gemini
+        response = client.models.generate_content(
+            model=MODEL_NAME,
+            contents=prompt_content,
+            config=genai.types.GenerateContentConfig(
+                system_instruction=system_instruction
+            )
+        )
+        return response.text
+    except Exception as e:
+        # Muestra errores comunes de autenticación si la clave es incorrecta
+        return f"🚨 Error al generar el informe con Gemini (posible API Key inválida o límite excedido): {e}"
+
+
+# --- Formulario de entrada de datos principal ---
+with st.form("post_mortem_form"):
+    st.subheader("Detalles de la Incidencia")
+    
+    col1, col2 = st.columns(2)
+    with col1:
+        tipo_alerta = st.text_input(
+            "Tipo de Alerta / Título del Problema",
+            placeholder="Ej: High CPU usage en DB Server; Caída del servicio de Login"
+        )
+        sistema_afectado = st.text_input(
+            "Sistema Afectado",
+            placeholder="Ej: Servidor de Base de Datos; Microservicio de Autenticación"
+        )
+        
+    with col2:
+        fecha_hora = st.text_input(
+            "Fecha y Hora",
+            placeholder="Ej: 26/Sep/2025 21:45 PM UTC-3"
+        )
+        
+    st.subheader("Descripción y Consecuencias")
+    impacto_detalle = st.text_area(
+        "Impacto del Negocio y Observaciones",
+        placeholder="Ej: Latencia elevada afectó al 20% de los usuarios.",
+        height=150
+    )
+    
+    acciones_tomadas = st.text_area(
+        "Acciones Tomadas (Mitigación Inmediata)",
+        placeholder="Ej: Se escaló a Dev Team. Se reinició el servicio crítico.",
+        height=150
+    )
+
+    # Botón de envío
+    submitted = st.form_submit_button("🚀 Generar Informe Post-Mortem")
+
+# --- Lógica de procesamiento ---
+if submitted:
+    api_key = st.session_state.get('gemini_api_key')
+    
+    if not api_key:
+        st.error("❌ Por favor, ingresa tu API Key de Gemini en el panel lateral (sidebar) antes de continuar.")
+    elif not all([tipo_alerta, sistema_afectado, fecha_hora, impacto_detalle, acciones_tomadas]):
+        st.error("❌ Por favor, completa todos los campos del formulario para generar el informe.")
+    else:
+        with st.spinner("⏳ Generando informe técnico con Google Gemini..."):
+            informe = generar_post_mortem(
+                api_key,
+                tipo_alerta,
+                sistema_afectado,
+                fecha_hora,
+                impacto_detalle,
+                acciones_tomadas
+            )
+        
+        # Mostrar el resultado
+        st.divider()
+        st.subheader("✅ Informe Post-Mortem Generado")
+        st.markdown(informe)
+        st.download_button(
+            label="Descargar Informe (.md)",
+            data=informe,
+            file_name="post_mortem_incidencia.md",
+            mime="text/markdown"
+        )