Update app.py

app.py

@@ -2,9 +2,16 @@ import numpy as np
 import gradio as gr
 import cv2
 
-def to_black(image):
-    output = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
-    return output
+def to_black(image, transfer_style):
+    if transfer_style == "Hayao":
+         output = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)# 转换为灰度图像
+         return output
+    elif transfer_style == "Shinkai":
+        return image
+    elif transfer_style == "Kon Satoshi":
+        return image
+    else:
+        return image
 
 def clear_output(input_widget):
     input_widget = np.array([])
@@ -17,18 +24,20 @@ with gr.Blocks() as demo:
                 image_input = gr.Image()
                 image_output = gr.Image()
             with gr.Row():
+                image_dropdown = gr.Dropdown(label="Transfer Style",choices=["Hayao", "Shinkai", "Kon Satoshi"])
                 image_button = gr.Button("Transfer")
                 clear_image_button = gr.Button("Clear")
         with gr.TabItem("Transfer Video"):
             with gr.Row():
+                video_dropdown = gr.Dropdown(label="Transfer Style",choices=["Hayao", "Shinkai", "Kon Satoshi"])
                 video_input = gr.Video()
                 video_output = gr.Video()
             with gr.Row():
                 video_button = gr.Button("Transfer")
                 clear_video_button = gr.Button("Clear")
 
-    image_button.click(to_black, inputs=image_input, outputs=image_output)
-    video_button.click(to_black, inputs=video_input, outputs=video_output)
+    image_button.click(to_black, inputs=[image_input,image_dropdown], outputs=image_output)
+    video_button.click(to_black, inputs=[video_input,video_dropdown], outputs=video_output)
     clear_image_button.click(clear_output, inputs=image_input,outputs=image_output)
     clear_video_button.click(clear_output, inputs=video_input,outputs=video_output)
 demo.launch()