Update app.py

app.py

@@ -306,7 +306,9 @@ def text_to_image(text, size, position):
 
     return image
 
-# Modify the main Gradio interface
+
+# ... (previous imports and setup remain the same)
+
 with gr.Blocks(css=css) as Kolors:
     with gr.Row():
         with gr.Column(elem_id="col-left"):
@@ -318,7 +320,7 @@ with gr.Blocks(css=css) as Kolors:
                 )
             with gr.Row():
                 image_input_type = gr.Radio(["Upload Image", "Generate Text Image"], label="Input Type", value="Upload Image")
-            
+
             with gr.Row():
                 image = gr.Image(label="Image", type="pil", visible=True)
                 with gr.Column(visible=False) as text_image_inputs:
@@ -330,26 +332,7 @@ with gr.Blocks(css=css) as Kolors:
                         value="middle-center"
                     )
                     generate_text_image = gr.Button("Generate Text Image")
-                    
-
-    def toggle_image_input(choice):
-        return {
-            image: gr.update(visible=choice == "Upload Image"),
-            text_image_inputs: gr.update(visible=choice == "Generate Text Image")
-        }
-
-    image_input_type.change(toggle_image_input, image_input_type, [image, text_image_inputs])
 
-    def generate_and_use_text_image(text, size, position):
-        text_image = text_to_image(text, size, position)
-        return text_image
-
-    generate_text_image.click(
-        generate_and_use_text_image,
-        inputs=[text_input, font_size, text_position],
-        outputs=image
-    )
-    
             with gr.Accordion("Advanced Settings", open=False):
                 negative_prompt = gr.Textbox(
                     label="Negative prompt",
@@ -407,11 +390,29 @@ with gr.Blocks(css=css) as Kolors:
                 canny_button = gr.Button("Canny", elem_id="button")
                 depth_button = gr.Button("Depth", elem_id="button")
                 pose_button = gr.Button("Pose", elem_id="button")
-            
+
         with gr.Column(elem_id="col-right"):
             result = gr.Gallery(label="Result", show_label=False, columns=2)
             seed_used = gr.Number(label="Seed Used")
-    
+
+    def toggle_image_input(choice):
+        return {
+            image: gr.update(visible=choice == "Upload Image"),
+            text_image_inputs: gr.update(visible=choice == "Generate Text Image")
+        }
+
+    image_input_type.change(toggle_image_input, image_input_type, [image, text_image_inputs])
+
+    def generate_and_use_text_image(text, size, position):
+        text_image = text_to_image(text, size, position)
+        return text_image
+
+    generate_text_image.click(
+        generate_and_use_text_image,
+        inputs=[text_input, font_size, text_position],
+        outputs=image
+    )
+
     with gr.Row():
         gr.Examples(
                 fn = infer_canny,
@@ -428,7 +429,7 @@ with gr.Blocks(css=css) as Kolors:
                 outputs = [result, seed_used],
                 label = "Depth"
             )
-        
+
     with gr.Row():
         gr.Examples(
                 fn = infer_pose,
@@ -456,4 +457,4 @@ with gr.Blocks(css=css) as Kolors:
         outputs = [result, seed_used]
     )
 
-Kolors.queue().launch(debug=True)                    
+Kolors.queue().launch(debug=True)