Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from diffusers import StableDiffusionPipeline
+import requests
 from dotenv import load_dotenv
 import openai
 import os
@@ -13,6 +13,7 @@ from streamlit_webrtc import webrtc_streamer, WebRtcMode, AudioProcessorBase
 # Configuración de la clave API
 load_dotenv()
 openai.api_key = os.getenv("OPENAI_API_KEY")
+pexels_api_key = os.getenv("sWMvMKk7Mvv5o6Q90XGMfP2vliUiagc9rESBAO9DVZQdmlcla0LMJ8O3")
 os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = "botidinamix-g.json"
 
 # Configuración de Streamlit
@@ -74,12 +75,15 @@ st.markdown("Bienvenido al Asistente Teológico, donde puedes preguntar sobre in
 st.sidebar.title("Navegación")
 page = st.sidebar.selectbox("Selecciona una página", ["Chat Asistente", "Gestión de Pedidos", "Generador de Frases Bíblicas"])
 
-# Cargar el modelo de Hugging Face
-@st.cache_resource
-def cargar_modelo():
-    return StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
-
-modelo_generacion_imagenes = cargar_modelo()
+# Función para buscar imágenes en Pexels
+def buscar_imagenes(query):
+    search_url = "https://api.pexels.com/v1/search"
+    headers = {"Authorization": pexels_api_key}
+    params = {"query": query, "per_page": 1}
+    response = requests.get(search_url, headers=headers, params=params)
+    response.raise_for_status()
+    search_results = response.json()
+    return search_results["photos"][0]["src"]["original"]
 
 if page == "Chat Asistente":
     # Chat con el asistente
@@ -149,29 +153,29 @@ elif page == "Gestión de Pedidos":
     calculo_pedido_agent.calcular_total(st.session_state)
 
 elif page == "Generador de Frases Bíblicas":
-    # Función para generar una imagen alusiva usando Hugging Face
+    # Función para generar una imagen alusiva usando Pexels
     def generar_imagen(frase):
         try:
-            image = modelo_generacion_imagenes(frase, num_inference_steps=50, guidance_scale=7.5).images[0]
-            return image
+            image_url = buscar_imagenes(frase)
+            return image_url
         except Exception as e:
             st.error(f"Error al generar la imagen: {e}")
             return None
 
     # Encabezado
     st.subheader("📜 Generador de Frases Bíblicas")
-    st.markdown("Escribe un versículo o una palabra y obtén una imagen alusiva usando Hugging Face.")
+    st.markdown("Escribe un versículo o una palabra y obtén una imagen alusiva usando Pexels.")
 
     # Entrada de texto para el versículo o palabra
     versiculo_usuario = st.text_input("Escribe un versículo o una palabra:")
 
     if st.button("Generar"):
         if versiculo_usuario:
-            with st.spinner("Generando imagen..."):
-                imagen_generada = generar_imagen(versiculo_usuario)
-                if imagen_generada:
+            with st.spinner("Buscando imagen..."):
+                imagen_url = generar_imagen(versiculo_usuario)
+                if imagen_url:
                     st.subheader("Imagen alusiva:")
-                    st.image(imagen_generada)
+                    st.image(imagen_url)
         else:
             st.warning("Por favor, ingresa un versículo o una palabra antes de generar.")