Update app.py

app.py

@@ -26,7 +26,7 @@ LORA_OPTIONS = {
 DEFAULT_MODEL = "black-forest-labs/FLUX.1-dev"
 
 # Función principal
-def query(lora_name, prompt, steps=20, cfg_scale=3.5, randomize_seed=True, seed=-1, width=1024, height=1024):
+def query(lora_name, prompt, steps, cfg_scale, randomize_seed, seed, width, height):
     if not prompt:
         return None
     
@@ -52,7 +52,7 @@ def query(lora_name, prompt, steps=20, cfg_scale=3.5, randomize_seed=True, seed=
         if response.status_code == 200:
             try:
                 image = Image.open(io.BytesIO(response.content))
-                return image, seed
+                return image, str(seed)
             except Exception as e:
                 raise gr.Error(f"Error al procesar la imagen: {e}")
         elif response.status_code == 503:
@@ -69,7 +69,7 @@ with gr.Blocks() as app:
     gr.Markdown("<center><h1>Generador de Imágenes</h1></center>")
     
     with gr.Row():
-        lora_selector = gr.Dropdown(label="Modelo", choices=list(LORA_OPTIONS.keys()), value="")
+        lora_selector = gr.Dropdown(label="Modelo", choices=list(LORA_OPTIONS.keys()), value="", allow_custom_value=True)
         text_prompt = gr.Textbox(label="Prompt", placeholder="Describe tu imagen")
     
     with gr.Row():
@@ -78,6 +78,7 @@ with gr.Blocks() as app:
         steps = gr.Slider(label="Pasos", minimum=1, maximum=100, step=1, value=20)
         cfg = gr.Slider(label="CFG Scale", minimum=1, maximum=20, step=0.5, value=3.5)
         randomize_seed = gr.Checkbox(label="Randomizar semilla", value=True)
+        seed = gr.Textbox(label="Semilla", placeholder="-1 para aleatorio", value="-1")
 
     image_output = gr.Image(type="pil", label="Resultado")
     seed_output = gr.Textbox(label="Semilla utilizada", interactive=False)
@@ -85,8 +86,8 @@ with gr.Blocks() as app:
 
     generate_button.click(
         query,
-        inputs=[lora_selector, text_prompt, steps, cfg, randomize_seed, -1, width, height],
+        inputs=[lora_selector, text_prompt, steps, cfg, randomize_seed, seed, width, height],
         outputs=[image_output, seed_output]
     )
 
-app.launch(share=True)
+app.launch(share=True)