Update app.py

app.py

@@ -12,18 +12,14 @@ from gradio_client import Client, handle_file
 from huggingface_hub import login
 from gradio_imageslider import ImageSlider
 
-# Constantes
 MAX_SEED = np.iinfo(np.int32).max
 HF_TOKEN = os.environ.get("HF_TOKEN")
 HF_TOKEN_UPSCALER = os.environ.get("HF_TOKEN_UPSCALER")
 
-# Funciones
 def enable_lora(lora_add, basemodel):
-    """Habilita o deshabilita el modelo LORA"""
     return basemodel if not lora_add else lora_add
 
 async def generate_image(prompt, model, lora_word, width, height, scales, steps, seed):
-    """Genera una imagen a partir de un texto"""
     try:
         if seed == -1:
             seed = random.randint(0, MAX_SEED)
@@ -63,36 +59,31 @@ def get_upscale_finegrain(prompt, img_path, upscale_factor):
         print(f"Error upscale image: {e}")
         return None
 
-# Interfaz gráfica
 css = """
-#col-container{
-    margin: 0 auto;
-    max-width: 1024px;
-}
+#col-container{ margin: 0 auto; max-width: 1024px;}
 """
 
 with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
     with gr.Column(elem_id="col-container"):
-        gr.Markdown("Flux Upscaled +LORA")
         with gr.Row():
             with gr.Column(scale=3):
                 output_res = ImageSlider(label="Flux / Upscaled")
             with gr.Column(scale=2):
-                prompt = gr.Textbox(label="Prompt")
-                basemodel_choice = gr.Dropdown(label="Base Model", choices=["black-forest-labs/FLUX.1-schnell", "black-forest-labs/FLUX.1-DEV"], value="black-forest-labs/FLUX.1-schnell")
-                lora_model_choice = gr.Dropdown(label="LORA Model", choices=["Shakker-Labs/FLUX.1-dev-LoRA-add-details", "XLabs-AI/flux-RealismLora"], value="XLabs-AI/flux-RealismLora")
-                process_lora = gr.Checkbox(label="Process LORA")
-                process_upscale = gr.Checkbox(label="Process Upscale")
-                upscale_factor = gr.Radio(label="UpScale Factor", choices=[2, 4, 8], value=2)
+                prompt = gr.Textbox(label="Descripción de imágen")
+                basemodel_choice = gr.Dropdown(label="Modelo", choices=["black-forest-labs/FLUX.1-schnell", "black-forest-labs/FLUX.1-DEV"], value="black-forest-labs/FLUX.1-schnell")
+                lora_model_choice = gr.Dropdown(label="LORA Realismo", choices=["Shakker-Labs/FLUX.1-dev-LoRA-add-details", "XLabs-AI/flux-RealismLora"], value="XLabs-AI/flux-RealismLora")
+                process_lora = gr.Checkbox(label="Procesar LORA")
+                process_upscale = gr.Checkbox(label="Procesar Escalador")
+                upscale_factor = gr.Radio(label="Factor de Escala", choices=[2, 4, 8], value=2)
                 
-                with gr.Accordion(label="Advanced Options", open=False):
-                    width = gr.Slider(label="Width", minimum=512, maximum=1280, step=8, value=1280)
-                    height = gr.Slider(label="Height", minimum=512, maximum=1280, step=8, value=768)
-                    scales = gr.Slider(label="Guidance", minimum=3.5, maximum=7, step=0.1, value=3.5)
-                    steps = gr.Slider(label="Steps", minimum=1, maximum=100, step=1, value=24)
-                    seed = gr.Slider(label="Seeds", minimum=-1, maximum=MAX_SEED, step=1, value=-1)
+                with gr.Accordion(label="Opciones Avanzadas", open=False):
+                    width = gr.Slider(label="Ancho", minimum=512, maximum=1280, step=8, value=1280)
+                    height = gr.Slider(label="Alto", minimum=512, maximum=1280, step=8, value=768)
+                    scales = gr.Slider(label="Escalas", minimum=3.5, maximum=7, step=0.1, value=3.5)
+                    steps = gr.Slider(label="Pasos", minimum=1, maximum=100, step=1, value=24)
+                    seed = gr.Slider(label="Semillas", minimum=-1, maximum=MAX_SEED, step=1, value=-1)
             
-                submit_btn = gr.Button("Submit", scale=1)
+                submit_btn = gr.Button("Crear", scale=1)
                 submit_btn.click(
                     fn=lambda: None,
                     inputs=None,