Update app.py

app.py

@@ -6,8 +6,11 @@ import io
 import random
 
 url = "http://73.255.78.150:7909/sdapi/v1/img2img"
+session_seed, last_session_seed = None, None
 
-def set_seed_values(reset_seed, increment_seed, session_seed, last_session_seed):
+def generate_image(prompt, steps, reset_seed, increment_seed, image_file=None):
+    global session_seed, last_session_seed
+    
     if reset_seed:
         last_session_seed, session_seed = session_seed, -1 if session_seed != -1 else None
     elif not reset_seed and last_session_seed:
@@ -18,13 +21,6 @@ def set_seed_values(reset_seed, increment_seed, session_seed, last_session_seed)
     if increment_seed and session_seed != -1:
         session_seed += 1
 
-    return session_seed, last_session_seed
-
-def generate_image(prompt, steps, reset_seed, increment_seed, image_file=None):
-    global session_seed, last_session_seed
-    
-    session_seed, last_session_seed = set_seed_values(reset_seed, increment_seed, session_seed, last_session_seed)
-
     init_image_value = ""
     if image_file:
         image_data = image_file.read()
@@ -48,27 +44,17 @@ def generate_image(prompt, steps, reset_seed, increment_seed, image_file=None):
         ImageDraw.Draw(error_image).text((10, 10), f"Error: {response_data.get('error', 'Unknown error')}", fill=(255, 255, 0))
         return error_image
 
-def main_interface():
-    with gr.Blocks() as main_blocks:
-        with gr.Row():
-            img_output = gr.Image(label="Generated Image")
-        with gr.Row():
-            with gr.Column():
-                prompt_input = gr.Textbox(lines=2, placeholder="Imagine something...", label="Prompt")
-                steps_input = gr.Slider(minimum=15, maximum=50, step=15, default=5, label="Steps")
-                image_input = gr.File(label="Upload Image")
-            with gr.Column():
-                reset_seed_input = gr.Checkbox(label="Randomize Composition")
-                increment_seed_input = gr.Checkbox(label="New Composition")
-
-    iface = gr.Interface(
-        fn=generate_image,
-        inputs=[prompt_input, steps_input, reset_seed_input, increment_seed_input, image_input],
-        outputs=[img_output],
-        live=False,
-        layout=main_blocks
-    )
-    iface.launch()
-
-if __name__ == "__main__":
-    main_interface()
+iface = gr.Interface(
+    fn=generate_image,
+    inputs=[
+        gr.Textbox(placeholder="Imagine something...", label="Prompt"),
+        gr.Slider(minimum=15, maximum=50, step=15, default=5, label="Steps"),
+        gr.Checkbox(label="Randomize Composition"),
+        gr.Checkbox(label="New Composition"),
+        gr.File(label="Upload Image")
+    ],
+    outputs=gr.Image(label="Generated Image"),
+    live=False
+)
+
+iface.launch()