JeCabrera commited on
Commit
2792969
·
verified ·
1 Parent(s): fdf8946

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -4
app.py CHANGED
@@ -76,15 +76,24 @@ def response(message, history):
76
  text_prompt = message["text"]
77
  files = message["files"]
78
 
79
- # Convertir imágenes a rutas de archivo o bytes antes de usarlas como prompts
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
- image_path = cache_pil_image(image) # Convertir a una ruta
87
- image_prompts.append(((image_path,), None)) # Usar la ruta en lugar de la imagen
 
 
 
 
 
 
 
 
 
88
 
89
  # Combinar texto e imágenes para el modelo
90
  prompts = [text_prompt] + image_prompts
 
76
  text_prompt = message["text"]
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
+ # 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