From a2e0f444364a0bc569f96811d88a15938c2807f4 Mon Sep 17 00:00:00 2001
From: BorjaOteroFerreira
 <115078003+BorjaOteroFerreira@users.noreply.github.com>
Date: Wed, 18 Sep 2024 01:41:27 +0200
Subject: [PATCH] generate_image [FLUX_1.0 {Gracio Client Api (Huggingface
 spaces)}]

---
 tools/generate_image.py | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/tools/generate_image.py b/tools/generate_image.py
index 786897f..b006be6 100644
--- a/tools/generate_image.py
+++ b/tools/generate_image.py
@@ -6,8 +6,7 @@
 from gradio_client import Client
 from huggingface_hub import login
 
-# Reemplaza 'your_token' por tu token de Hugging Face
-login(token="huggingface_token")
+
 
 
 class ImageGenerationResult(BaseModel):
@@ -50,7 +49,14 @@ def run(
         Returns:
             List[ImageGenerationResult]: Una lista con la URL del archivo generado y la semilla utilizada.
         """
+        try: 
+            login(token="huggingface_token", add_to_git_credential=True)
+        except Exception as e:
+            raise Exception("No se pudo iniciar sesión en Hugging Face Hub. Por favor, verifica tu token.", e)
+        
         try:
+            # Reemplaza 'your_token' por tu token de Hugging Face
+
             # Crear el cliente de Gradio
             client = Client("black-forest-labs/FLUX.1-dev")
             # Realizar la predicción para generar la imagen
@@ -72,11 +78,11 @@ def run(
             destination_path = os.path.join(images_folder, file_name)
             # Mover la imagen desde la carpeta temporal a 'static/images'
             shutil.move(temp_file_path, destination_path)
-            # Devolver la URL de la imagen generada y la semilla
+            # Devolver la URL de la imagen generada
             imagen = f"<img src='static/tools/generate_image/{file_name}' width='590' height='345'>", 
-            respuesta = f'IMAGEN de {prompt} -> {imagen}, ES IMPORTANTISIMO QUE INCLUYAS LA ETIQUETA DE IMAGEN EN TU RESPUESTA RESPETANDO EL FORMATO,  sin texto adicional despues de la etiqueta y sin acortar ni alterar la url'
+            respuesta = f'IMAGEN de {prompt} -> {imagen}, ES IMPORTANTISIMO QUE RESPONDAS SOLO CON LA ETIQUETA DE IMAGEN EN TU RESPUESTA RESPETANDO EL FORMATO,  sin texto adicional despues de la etiqueta y sin acortar ni alterar la url'
             return str(respuesta)
         except Exception as e:
             print(e)
-            respuesta = f'IMAGEN de {prompt} : /static/tools/generate_image/pj.png, ES IMPORTANTISIMO QUE INCLUYAS LA ETIQUETA DE IMAGEN EN TU RESPUESTA RESPETANDO EL FORMATO,  sin texto adicional despues de la etiqueta y sin acortar ni alterar la url'
+            respuesta = f"IMAGEN de {prompt} -> <img src='/static/tools/generate_image/pj.png' width='590' height='345'>, ES IMPORTANTISIMO QUE RESPONDAS SOLO CON LA ETIQUETA DE IMAGEN EN TU RESPUESTA RESPETANDO EL FORMATO,  sin texto adicional despues de la etiqueta. NO ALTERES LA URL"
             return respuesta