JeCabrera commited on
Commit
4ca591f
·
verified ·
1 Parent(s): b99f3a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -3
app.py CHANGED
@@ -77,14 +77,23 @@ def response(message, history):
77
  files = message["files"]
78
 
79
  # Procesar imágenes cargadas
80
- image_prompts = []
81
  if files:
82
  for file in files:
83
  image = Image.open(file).convert('RGB')
84
  image_preview = preprocess_image(image)
85
  if image_preview:
86
- cache_pil_image(image)
87
- image_prompts.append(image_preview) # Incluir en la lista de prompts
 
 
 
 
 
 
 
 
 
88
 
89
  # Combinar texto e imágenes para el modelo
90
  prompts = [text_prompt] + image_prompts
 
77
  files = message["files"]
78
 
79
  # Procesar imágenes cargadas
80
+ image_prompts = [preprocess_image(Image.open(file).convert('RGB')) for file in files] if files else []
81
  if files:
82
  for file in files:
83
  image = Image.open(file).convert('RGB')
84
  image_preview = preprocess_image(image)
85
  if image_preview:
86
+ # Guardar la imagen y obtener la ruta
87
+ image_path = cache_pil_image(image)
88
+ # Leer la imagen en formato binario para enviarla como Blob
89
+ with open(image_path, "rb") as img_file:
90
+ img_data = img_file.read()
91
+ # Crear un diccionario con los datos binarios y su tipo MIME
92
+ image_prompt = {
93
+ "mime_type": "image/jpeg",
94
+ "data": img_data
95
+ }
96
+ image_prompts.append(image_prompt)
97
 
98
  # Combinar texto e imágenes para el modelo
99
  prompts = [text_prompt] + image_prompts