Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,61 @@
-import altair as alt
-import numpy as np
-import pandas as pd
-import streamlit as st
+# app/streamlit_app.py
 
-"""
-# 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])),
-    ))
+import os
+import streamlit as st
+from agents import Runner
+from agent.linkedin_agent import create_agent
+
+st.set_page_config(page_title="Buscador de Perfiles LinkedIn", layout="centered")
+
+# Verificamos que la clave esté presente en entorno
+if not os.getenv("OPENAI_API_KEY"):
+    st.error("❌ Por favor define la variable de entorno OPENAI_API_KEY.")
+    st.stop()
+
+# Instanciamos una sola vez el agente al arrancar la app
+agent = create_agent()
+
+st.title("🔍 Agente de Búsqueda de Perfiles LinkedIn")
+st.write("Ingresa la descripción de la oferta y la cantidad de perfiles que deseas encontrar.")
+
+# Contenedor para historial de chat
+if "history" not in st.session_state:
+    st.session_state.history = []
+
+# Formulario para que el usuario ingrese oferta + número de perfiles
+with st.form(key="oferta_form", clear_on_submit=False):
+    oferta = st.text_area("📰 Descripción de la oferta de empleo:", height=150)
+    num_perfiles = st.number_input(
+        "🔢 Cantidad de perfiles a buscar:", 
+        min_value=1, max_value=20, value=5
+    )
+    enviar = st.form_submit_button(label="Enviar al agente")
+
+# Cuando el usuario hace clic en “Enviar”
+if enviar and oferta:
+    # Agregamos el mensaje del usuario al historial
+    st.session_state.history.append(("usuario", oferta))
+
+    # Construimos el prompt tal como espera el agente: oferta + N
+    prompt = f"Oferta: {oferta}\nNúmero perfiles: {num_perfiles}"
+
+    # Ejecutamos el agente de forma síncrona
+    # Runner.run_sync se encarga de bloquear hasta obtener la salida final
+    resultado = Runner.run_sync(agent, prompt)
+    respuesta = resultado.final_output
+
+    # Agregamos la respuesta del agente al historial
+    st.session_state.history.append(("agente", respuesta))
+
+# Mostramos el “chat” en orden cronológico
+st.markdown("""---""")
+st.markdown("### 💬 Historial de Chat")
+for quien, texto in st.session_state.history:
+    if quien == "usuario":
+        st.markdown(f"**Tú:** {texto}")
+    else:
+        st.markdown(f"**Agente:** {texto}")
+
+# Al pie, sugerimos al usuario limpiar el historial si desea volver a empezar
+if st.button("🔄 Limpiar historial"):
+    st.session_state.history = []