Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,74 @@
-import altair as alt
-import numpy as np
-import pandas as pd
 import streamlit as st
+import requests
+
+# Configuración
+TEXT_ENDPOINT = "https://lrs1lm2jji.execute-api.us-east-1.amazonaws.com/dev/generate"
+IMAGE_ENDPOINT = "https://lrs1lm2jji.execute-api.us-east-1.amazonaws.com/dev/generate-image"
+
+st.set_page_config(page_title="OpenAI App", layout="centered")
+
+st.title("🧠 Aplicación IA con OpenAI y Lambda")
+
+tab1, tab2 = st.tabs(["Generar Texto", "Generar Imagen"])
+
+with tab1:
+    st.subheader("✍️ Texto Generado por IA")
+    prompt = st.text_area("Escribe tu prompt aquí:", height=150)
+
+    if st.button("Generar Texto"):
+        if prompt.strip() == "":
+            st.warning("Escribe un prompt primero.")
+        else:
+            with st.spinner("Generando..."):
+                try:
+                    res = requests.post(TEXT_ENDPOINT, json={"prompt": prompt})
+                    res.raise_for_status()
+                    result = res.json()
+
+                    st.success(f"Respuesta generada con éxito: {res}")
+
+                    if "response" in result:
+                        st.text_area("Texto:", value=result["response"], height=300)
+                    elif "text" in result:
+                        st.text_area("Texto:", value=result["text"], height=300)
+
+                    if "imageUrl" in result:
+                        st.image(result["imageUrl"], caption="Imagen generada", use_column_width=True)
+                except Exception as e:
+                    st.error(f"Error al generar: {e}")
+
+
+with tab2:
+    st.subheader("🖼️ Imagen Generada por IA")
+    img_prompt = st.text_input("Describe la imagen que deseas:")
+
+    if st.button("Generar Imagen"):
+        if img_prompt.strip() == "":
+            st.warning("Escribe una descripción primero.")
+        else:
+            with st.spinner("Generando imagen..."):
+                try:
+                    st.info("⏳ Enviando solicitud para generar imagen...")
+                    st.write("🔍 Prompt enviado:", img_prompt)
+
+                    headers = {
+                        "Content-Type": "application/json",
+                        "User-Agent": "Python-Requests"
+                    }
+                    res = requests.post(IMAGE_ENDPOINT, json={"prompt": img_prompt}, headers=headers)
+                    st.write("📬 Código de respuesta:", res.status_code)
+                    st.write("📦 Respuesta cruda:", res.text)
+                    res.raise_for_status()
+                    data = res.json()
+                    st.write("🧩 JSON recibido:", data)
+                    url = data.get("image_url") or data.get("imageUrl") or data.get("url")
+
+                    if url:
+                        st.success("✅ Imagen generada:")
+                        st.image(url, use_container_width =True)
+                    else:
+                        st.error("⚠️ No se pudo obtener la URL de imagen.")
+                except Exception as e:
+                    st.error(f"❌ Error al generar imagen: {e}")
+
 
-"""
-# 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])),
-    ))