Update app.py

app.py

@@ -1,38 +1,54 @@
-import io
+import streamlit as st
 from PIL import Image
 import google.generativeai as genai
-import streamlit as st
+from gtts import gTTS
+import os
+import io
+
+# Configuración de la página de Streamlit
+st.set_page_config(page_title="🤖 ChatBot", layout="wide")
 
 # Configuración de la API key para Google Generative AI
 api_key = "AIzaSyDJZ3r6VRhRivR0pb96cBRg_VvGg_fXq5k"  # API key proporcionada
 
+def procesar_texto(texto):
+    genai.configure(api_key=api_key)
+    modelo = genai.GenerativeModel('gemini-1.5-pro-latest')
+    respuesta = modelo.generate_content(texto)
+    return respuesta.text
+
 def procesar_imagen(imagen):
     genai.configure(api_key=api_key)
     modelo = genai.GenerativeModel('gemini-1.5-pro-latest')
-    
-    # Convertir la imagen a formato compatible
+
+    # Asegúrate de que la imagen sea un objeto PIL.Image
+    if isinstance(imagen, Image.Image):
+        # Convertir la imagen a un formato compatible si es necesario
+        try:
+            # Aquí puedes convertir la imagen a un formato aceptable si la API requiere un formato específico
+            respuesta = modelo.generate_content(imagen)
+            return respuesta.text
+        except Exception as e:
+            return f"Error al procesar la imagen: {e}"
+    else:
+        return "La imagen no es del tipo esperado."
+
+def hablar_texto(texto):
     try:
-        # Asegúrate de que la imagen está en formato PIL.Image
-        if isinstance(imagen, bytes):
-            imagen = Image.open(io.BytesIO(imagen))
-        
-        # Convertir la imagen a un formato adecuado para la API
-        imagen_bytes = io.BytesIO()
-        imagen.save(imagen_bytes, format='PNG')
-        imagen_bytes.seek(0)
-        
-        # Procesar la imagen con la API
-        respuesta = modelo.generate_content(imagen_bytes)
-        return respuesta.text
+        tts = gTTS(text=texto, lang='es')
+        tts.save("respuesta.mp3")
+        os.system("mpg321 respuesta.mp3")  # Usa mpg321 para reproducir el archivo
     except Exception as e:
-        st.error(f"Error al procesar la imagen: {e}")
-        return "Hubo un error al procesar la imagen."
+        st.error(f"Error al generar el audio: {e}")
 
-# Ejemplo de uso en Streamlit
 st.title("🤖 ¡Bienvenido a ChatBot!")
 
 espacio_contenido_generado = st.empty()
 
+# Cargar y aplicar CSS personalizado
+with open("./style.css") as f:
+    st.markdown(f"<style>{f.read()}</style>", unsafe_allow_html=True)
+
 # Contenido principal
 st.write("¡Hola! Soy tu asistente virtual. ¿Cómo puedo ayudarte hoy? 😊")
 
@@ -57,7 +73,7 @@ with col2:
             imagen = Image.open(entrada_imagen)
             st.image(imagen, caption='Imagen subida.', use_column_width=True)
             with st.spinner("Procesando imagen..."):
-                respuesta = procesar_imagen(entrada_imagen.read())
+                respuesta = procesar_imagen(imagen)
             espacio_contenido_generado.write(f"**Respuesta de la imagen:** {respuesta}")
 
 st.write("¡Gracias por usar el ChatBot! 😊")