Update app.py

app.py

@@ -7,6 +7,7 @@ import time
 import base64
 import tempfile
 import random
+from transformers import pipeline
 from google.cloud import texttospeech
 from google.cloud.speech import SpeechClient, RecognitionAudio, RecognitionConfig
 from streamlit_webrtc import webrtc_streamer, WebRtcMode, AudioProcessorBase
@@ -76,16 +77,29 @@ 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 clasificación de imágenes
+clasificador = pipeline("zero-shot-image-classification")
+
 # Función para buscar imágenes en Pexels
 def buscar_imagenes(query):
     search_url = "https://api.pexels.com/v1/search"
-    headers = {"Authorization":"sWMvMKk7Mvv5o6Q90XGMfP2vliUiagc9rESBAO9DVZQdmlcla0LMJ8O3"}
+    headers = {"Authorization": "sWMvMKk7Mvv5o6Q90XGMfP2vliUiagc9rESBAO9DVZQdmlcla0LMJ8O3"}
     params = {"query": query, "per_page": 10}
     response = requests.get(search_url, headers=headers, params=params)
     response.raise_for_status()
     search_results = response.json()
     images = [photo["src"]["original"] for photo in search_results["photos"]]
-    return random.choice(images) if images else None
+    return images
+
+# Función para verificar la adecuación de la imagen
+def verificar_imagen(image_url, label):
+    image_data = requests.get(image_url).content
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as temp_file:
+        temp_file.write(image_data)
+        temp_file_path = temp_file.name
+    results = clasificador(temp_file_path, candidate_labels=[label])
+    os.remove(temp_file_path)
+    return results[0]['label'] == label
 
 # Función para generar reflexión y consulta de imagen con OpenAI
 def generar_reflexion_y_consulta(keyword):
@@ -103,6 +117,18 @@ def generar_reflexion_y_consulta(keyword):
     consulta_imagen = partes[1].strip() if len(partes) > 1 else keyword
     return reflexion, consulta_imagen
 
+# Función para obtener un audio aleatorio de la carpeta "reflexiones"
+def obtener_audio_aleatorio():
+    carpeta_reflexiones = "reflexiones"
+    archivos = os.listdir(carpeta_reflexiones)
+    archivos_mp3 = [archivo for archivo in archivos if archivo.endswith(".mp3")]
+    if archivos_mp3:
+        audio_seleccionado = random.choice(archivos_mp3)
+        with open(os.path.join(carpeta_reflexiones, audio_seleccionado), "rb") as audio_file:
+            audio_bytes = audio_file.read()
+        return base64.b64encode(audio_bytes).decode("utf-8")
+    return None
+
 if page == "Chat Asistente":
     # Chat con el asistente
     st.subheader("🗣️ Chat con el Asistente")
@@ -175,8 +201,12 @@ elif page == "Generador de Frases Bíblicas":
     def generar_imagen_y_reflexion(keyword):
         try:
             reflexion, consulta_imagen = generar_reflexion_y_consulta(keyword)
-            image_url = buscar_imagenes(consulta_imagen)
-            return image_url, reflexion
+            image_urls = buscar_imagenes(consulta_imagen)
+            for image_url in image_urls:
+                if verificar_imagen(image_url, keyword):
+                    return image_url, reflexion
+            st.warning("No se encontraron imágenes adecuadas. Intenta nuevamente.")
+            return None, reflexion
         except Exception as e:
             st.error(f"Error al generar la imagen o reflexión: {e}")
             return None, None
@@ -197,6 +227,16 @@ elif page == "Generador de Frases Bíblicas":
                     st.image(imagen_url)
                     st.subheader("Reflexión:")
                     st.markdown(reflexion)
+
+                    # Obtener y reproducir un audio aleatorio
+                    audio_base64 = obtener_audio_aleatorio()
+                    if audio_base64:
+                        audio_html = f"""
+                            <audio autoplay>
+                                <source src="data:audio/mp3;base64,{audio_base64}" type="audio/mp3">
+                            </audio>
+                        """
+                        st.markdown(audio_html, unsafe_allow_html=True)
         else:
             st.warning("Por favor, ingresa un versículo o una palabra antes de generar.")