Update app.py

app.py

@@ -1,5 +1,5 @@
-import pandas as pd
 import streamlit as st
+from transformers import pipeline
 from dotenv import load_dotenv
 import openai
 import os
@@ -9,13 +9,11 @@ import tempfile
 from google.cloud import texttospeech
 from google.cloud.speech import SpeechClient, RecognitionAudio, RecognitionConfig
 from streamlit_webrtc import webrtc_streamer, WebRtcMode, AudioProcessorBase
-import requests
 
 # Configuración de la clave API
 load_dotenv()
 openai.api_key = os.getenv("OPENAI_API_KEY")
-os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = "path/to/your/google-cloud-credentials.json"
-PEXELS_API_KEY = os.getenv("PEXELS_API_KEY")
+os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = "botidinamix-g.json"
 
 # Configuración de Streamlit
 st.set_page_config(page_title="Asistente Teológico", page_icon="📖")
@@ -76,6 +74,13 @@ 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(allow_output_mutation=True)
+def cargar_modelo():
+    return pipeline('text-to-image', model='CompVis/stable-diffusion-v1-4')
+
+modelo_generacion_imagenes = cargar_modelo()
+
 if page == "Chat Asistente":
     # Chat con el asistente
     st.subheader("🗣️ Chat con el Asistente")
@@ -144,31 +149,19 @@ 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 Pexels
+    # Función para generar una imagen alusiva usando Hugging Face
     def generar_imagen(frase):
-        headers = {
-            "Authorization": f"Bearer {PEXELS_API_KEY}"
-        }
-        params = {
-            "query": frase,
-            "per_page": 1
-        }
-        response = requests.get("https://api.pexels.com/v1/search", headers=headers, params=params)
-        if response.status_code == 200:
-            data = response.json()
-            st.write(data)  # Mostrar la respuesta de la API para depuración
-            if data["photos"]:
-                return data["photos"][0]["src"]["large"]
-            else:
-                st.error("No se encontraron imágenes")
-                return None
-        else:
-            st.error(f"Error al buscar la imagen: {response.status_code} - {response.text}")
+        try:
+            images = modelo_generacion_imagenes(frase, num_return_sequences=1)
+            image = images[0]["generated_image"]
+            return image
+        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 Pexels.")
+    st.markdown("Escribe un versículo o una palabra y obtén una imagen alusiva usando Hugging Face.")
 
     # Entrada de texto para el versículo o palabra
     versiculo_usuario = st.text_input("Escribe un versículo o una palabra:")
@@ -176,10 +169,10 @@ elif page == "Generador de Frases Bíblicas":
     if st.button("Generar"):
         if versiculo_usuario:
             with st.spinner("Generando imagen..."):
-                imagen_url = generar_imagen(versiculo_usuario)
-                if imagen_url:
+                imagen_generada = generar_imagen(versiculo_usuario)
+                if imagen_generada:
                     st.subheader("Imagen alusiva:")
-                    st.image(imagen_url)
+                    st.image(imagen_generada)
         else:
             st.warning("Por favor, ingresa un versículo o una palabra antes de generar.")