Add image rotation to controlnet

app.py

@@ -463,7 +463,7 @@ controlnet_pipe.scheduler = UniPCMultistepScheduler.from_config(controlnet_pipe.
 controlnet_pipe.enable_model_cpu_offload()
 
 
-def controlnet_edges(canny_input_prompt, input_image, input_low_threshold, input_high_threshold, input_invert, canny_input_seed):
+def controlnet_edges(canny_input_prompt, input_image, input_low_threshold, input_high_threshold, input_invert, canny_input_seed, canny_input_rotate):
     np_image = np.array(input_image)
 
     output_image = input_image
@@ -480,6 +480,11 @@ def controlnet_edges(canny_input_prompt, input_image, input_low_threshold, input
     canny_2 = Image.fromarray(canny_1)
     canny_1 = Image.fromarray(canny_1)
 
+    if int(canny_input_rotate) > 0:
+      canny_rotation = 360 - int(canny_input_rotate)
+      canny_2 = canny_2.rotate(canny_rotation, resample=Image.BICUBIC)
+      canny_1 = canny_1.rotate(canny_rotation, resample=Image.BICUBIC)
+
     input_width, input_height = canny_2.size
 
     limit_size = 768
@@ -533,6 +538,7 @@ with gr.Blocks() as canny_blocks_interface:
       canny_input_high_threshold = gr.inputs.Slider(minimum=0, maximum=1000, step=1, label="Upper Threshold:", default=120)
       canny_input_seed = gr.Slider(0, 99999999999999, label="seed", dtype=int, value=random_seed, interactive=True, step=1)
       canny_input_invert = gr.inputs.Checkbox(label="Invert Image")
+      canny_input_rotate = gr.Dropdown([0, 90, 180, 270], label="rotate image (for smartphones)")
     with gr.Column():
       canny_input_image = gr.inputs.Image(label="input image")
       go_button = gr.Button('generate image')
@@ -541,7 +547,7 @@ with gr.Blocks() as canny_blocks_interface:
         canny_output_1 = gr.outputs.Image(type="pil", label="traced edges")
   with gr.Row():
     canny_output_2 = gr.outputs.Image(type="pil", label="final image")
-  go_button.click(fn=controlnet_edges, inputs=[canny_input_prompt, canny_input_image, canny_input_low_threshold, canny_input_high_threshold, canny_input_invert, canny_input_seed], outputs=[canny_output_1, canny_output_2])
+  go_button.click(fn=controlnet_edges, inputs=[canny_input_prompt, canny_input_image, canny_input_low_threshold, canny_input_high_threshold, canny_input_invert, canny_input_seed, canny_input_rotate], outputs=[canny_output_1, canny_output_2])
 
 
 # canny_blocks_interface.launch(debug=False)
@@ -605,15 +611,15 @@ with gr.Blocks() as canny_blocks_interface:
 #     return output_image[0][0]
 
 
-canny_input_prompt = gr.inputs.Textbox(label="Enter a single word or phrase")
-canny_input_image = gr.inputs.Image()
-canny_input_low_threshold = gr.inputs.Slider(minimum=0, maximum=1000, step=1, label="Lower Threshold:", default=100)
-canny_input_high_threshold = gr.inputs.Slider(minimum=0, maximum=1000, step=1, label="Upper Threshold:", default=200)
-canny_input_invert = gr.inputs.Checkbox(label="Invert Image")
-canny_outputs = gr.outputs.Image(type="pil")
+# canny_input_prompt = gr.inputs.Textbox(label="Enter a single word or phrase")
+# canny_input_image = gr.inputs.Image()
+# canny_input_low_threshold = gr.inputs.Slider(minimum=0, maximum=1000, step=1, label="Lower Threshold:", default=100)
+# canny_input_high_threshold = gr.inputs.Slider(minimum=0, maximum=1000, step=1, label="Upper Threshold:", default=200)
+# canny_input_invert = gr.inputs.Checkbox(label="Invert Image")
+# canny_outputs = gr.outputs.Image(type="pil")
 
 # make and launch the gradio app...
-controlnet_canny_interface = gr.Interface(fn=controlnet_edges, inputs=[canny_input_prompt, canny_input_image, canny_input_low_threshold, canny_input_high_threshold, canny_input_invert], outputs=canny_outputs, title='Canny Edge Tracing', allow_flagging='never')
+# controlnet_canny_interface = gr.Interface(fn=controlnet_edges, inputs=[canny_input_prompt, canny_input_image, canny_input_low_threshold, canny_input_high_threshold, canny_input_invert], outputs=canny_outputs, title='Canny Edge Tracing', allow_flagging='never')
 # controlnet_canny_interface.launch()