Update app.py
Browse files
app.py
CHANGED
@@ -3,6 +3,7 @@ from PIL import Image
|
|
3 |
import google.generativeai as genai
|
4 |
from gtts import gTTS
|
5 |
import os
|
|
|
6 |
|
7 |
# Configuraci贸n de la p谩gina de Streamlit
|
8 |
st.set_page_config(page_title="馃 ChatBot", layout="wide")
|
@@ -19,8 +20,11 @@ def procesar_texto(texto):
|
|
19 |
def procesar_imagen(imagen):
|
20 |
genai.configure(api_key=api_key)
|
21 |
modelo = genai.GenerativeModel('gemini-1.5-pro-latest')
|
22 |
-
#
|
23 |
-
|
|
|
|
|
|
|
24 |
return respuesta.text
|
25 |
|
26 |
def hablar_texto(texto):
|
@@ -63,7 +67,7 @@ with col2:
|
|
63 |
imagen = Image.open(entrada_imagen)
|
64 |
st.image(imagen, caption='Imagen subida.', use_column_width=True)
|
65 |
with st.spinner("Procesando imagen..."):
|
66 |
-
respuesta = procesar_imagen(
|
67 |
espacio_contenido_generado.write(f"**Respuesta de la imagen:** {respuesta}")
|
68 |
|
69 |
st.write("隆Gracias por usar el ChatBot! 馃槉")
|
|
|
3 |
import google.generativeai as genai
|
4 |
from gtts import gTTS
|
5 |
import os
|
6 |
+
import io
|
7 |
|
8 |
# Configuraci贸n de la p谩gina de Streamlit
|
9 |
st.set_page_config(page_title="馃 ChatBot", layout="wide")
|
|
|
20 |
def procesar_imagen(imagen):
|
21 |
genai.configure(api_key=api_key)
|
22 |
modelo = genai.GenerativeModel('gemini-1.5-pro-latest')
|
23 |
+
# Convertir la imagen a un formato adecuado para el modelo
|
24 |
+
imagen_bytes = io.BytesIO()
|
25 |
+
imagen.save(imagen_bytes, format='PNG')
|
26 |
+
imagen_bytes.seek(0)
|
27 |
+
respuesta = modelo.generate_content(imagen_bytes)
|
28 |
return respuesta.text
|
29 |
|
30 |
def hablar_texto(texto):
|
|
|
67 |
imagen = Image.open(entrada_imagen)
|
68 |
st.image(imagen, caption='Imagen subida.', use_column_width=True)
|
69 |
with st.spinner("Procesando imagen..."):
|
70 |
+
respuesta = procesar_imagen(imagen)
|
71 |
espacio_contenido_generado.write(f"**Respuesta de la imagen:** {respuesta}")
|
72 |
|
73 |
st.write("隆Gracias por usar el ChatBot! 馃槉")
|