Update app.py

app.py

@@ -63,9 +63,10 @@ def improve_prompt(prompt):
         instruction = "Mejora mi prompt para texto a imagen en inglés con estilo, cinematografía, cámaras, atmósfera e iluminación para la mejor calidad, de máximo 200 palabras."
         formatted_prompt = f"{instruction}: {prompt}"
         response = llm_client.text_generation(formatted_prompt, max_new_tokens=200)
-        improved_text = ""
-        for r in response:
-            improved_text += r["generated_text"]
+        
+        # Aquí asumimos que la respuesta es un string completo, no una lista de objetos.
+        improved_text = response  # Se espera que 'response' sea una cadena de texto.
+
         return improved_text.strip()
     except Exception as e:
         return f"Error mejorando el prompt: {e}"
@@ -81,22 +82,24 @@ with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
                 output_res = ImageSlider(label="Flux / Upscaled")
             with gr.Column(scale=2):
                 prompt = gr.Textbox(label="Descripción de imágen")
-                improved_prompt = gr.Textbox(label="Prompt Mejorado", interactive=False)
-                improve_btn = gr.Button("Mejora mi prompt")
                 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)
+
+                improved_prompt = gr.Textbox(label="Prompt Mejorado", interactive=False)
+
+                improve_btn = gr.Button("Mejora mi prompt")
                 improve_btn.click(fn=improve_prompt, inputs=[prompt], outputs=improved_prompt)
-               
+                
                 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="Escalado", minimum=1, maximum=20, step=1, value=10)
-                    steps = gr.Slider(label="Pasos", minimum=1, maximum=100, step=1, value=20)
+                    steps = gr.Slider(label="Pasos", minimum=1, maximum 100, step=1, value=20)
                     seed = gr.Number(label="Semilla", value=-1)
-
+    
                 btn = gr.Button("Generar")
                 btn.click(fn=gen, inputs=[improved_prompt, basemodel_choice, width, height, scales, steps, seed, upscale_factor, process_upscale, lora_model_choice, process_lora], outputs=output_res)
     demo.launch()