app.py

app.py

@@ -28,22 +28,16 @@ def update_imgbox(choices):
     choices_plus = extend_choices(choices)
     return [gr.Image(None, label=m, visible=(m != 'NA'), elem_id="custom_image") for m in choices_plus]
 
-def gen_fn(model_str, prompt, negative_prompt, cfg_scale, sampler, seed):
+def gen_fn(model_str, prompt):
     if model_str == 'NA':
         return None
-    # Use the provided seed or generate a random one if seed is -1
-    noise = seed if seed != -1 else randint(0, 99999999)
-    
-    # Prepare the full prompt with negative prompt and CFG scale
-    full_prompt = f'{prompt} --negative_prompt "{negative_prompt}" --cfg_scale {cfg_scale} --sampler {sampler} --seed {noise}'
-    
-    return models_load[model_str](full_prompt)
+    noise = str(randint(0, 99999999))
+    return models_load[model_str](f'{prompt} {noise}')
 
 def make_me():
     with gr.Row():
         with gr.Column(scale=4):
             txt_input = gr.Textbox(label='Your prompt:', lines=4, container=False, elem_id="custom_textbox", placeholder="Prompt", height=250)
-            negative_prompt = gr.Textbox(label='Negative Prompt:', lines=2, placeholder="What to avoid in the image", elem_id="custom_negative_prompt")
         
         with gr.Column(scale=1):
             gen_button = gr.Button('Generate images', elem_id="custom_gen_button")
@@ -59,20 +53,10 @@ def make_me():
             stop_button.click(on_stop_click, inputs=None, outputs=[gen_button, stop_button])
     
     with gr.Row():
-        output = [gr.Image(label=m, width=None, height=None, elem_id="custom_image", show_label=True, interactive=False, show_share_button=False) for m in default_models]
+        output = [gr.Image(label=m, width=512, max_height=768, elem_id="custom_image", show_label=True, interactive=False, show_share_button=False) for m in default_models]
         current_models = [gr.Textbox(m, visible=False) for m in default_models]
-        
-        # Additional inputs for CFG, sampler, seed, and negative prompt
-        cfg_scale = gr.Slider(minimum=1, maximum=20, step=0.1, value=7, label="CFG Scale", elem_id="custom_cfg_scale")
-        sampler = gr.Dropdown(choices=["DDIM", "PLMS", "Euler", "Heun"], value="DDIM", label="Sampler", elem_id="custom_sampler")
-        seed = gr.Number(value=-1, label="Seed (-1 for random)", elem_id="custom_seed")
-        
         for m, o in zip(current_models, output):
-            gen_event = gen_button.click(
-                gen_fn, 
-                [m, txt_input, negative_prompt, cfg_scale, sampler, seed], 
-                o
-            )
+            gen_event = gen_button.click(gen_fn, [m, txt_input], o)
             stop_button.click(on_stop_click, inputs=None, outputs=[gen_button, stop_button], cancels=[gen_event])
     
     with gr.Accordion('Model selection', elem_id="custom_accordion"):