jeysshon commited on
Commit
5754f17
verified
1 Parent(s): 58947a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
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
- # Procesa la imagen como un archivo binario
23
- respuesta = modelo.generate_content(imagen.read())
 
 
 
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(entrada_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! 馃槉")