Update app.py

app.py

@@ -249,7 +249,8 @@ def encode_img_fn(img):
 def gen_video_fn(w1, w2, frame):
     if w1 is None or w2 is None:
         return None
-    model.gen_video(np.array(json.loads(w1), dtype=np.float32), np.array(json.loads(w2), dtype=np.float32), "video.mp4", int(frame))
+    model.gen_video(np.array(json.loads(w1), dtype=np.float32), np.array(json.loads(w2), dtype=np.float32), "video.mp4",
+                    int(frame))
     return "video.mp4"
 
 
@@ -262,50 +263,62 @@ if __name__ == '__main__':
                     "the model is not good, just for fun.")
         with gr.Tabs():
             with gr.TabItem("generate image"):
-                with gr.Column():
-                    with gr.Row():
+                with gr.Row():
+                    with gr.Column():
+                        gr.Markdown("generate image randomly or by seed")
                         gen_input1 = gr.Checkbox(value=False, label="use seed")
                         gen_input2 = gr.Number(value=1, label="seed")
-                    gen_input3 = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.6, label="psi")
-                    gen_submit = gr.Button("Run")
-                    gen_output1 = gr.Image()
-                    select_img_input_w1 = gr.Textbox(visible=False)
-                    select_img_input_img1 = gr.Image(visible=False)
+                        gen_input3 = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.6, label="truncation psi")
+                        with gr.Group():
+                            gen_submit = gr.Button("Run")
+                    with gr.Column():
+                        gen_output1 = gr.Image()
+                        select_img_input_w1 = gr.Textbox(visible=False)
+                        select_img_input_img1 = gr.Image(visible=False)
 
             with gr.TabItem("encode image"):
-                with gr.Column():
-                    encode_img_input = gr.Image()
-                    encode_img_submit = gr.Button("Run")
-                    encode_img_output1 = gr.Textbox(label="message")
-                    with gr.Row():
-                        encode_img_output2 = gr.Image(label="detected")
-                        encode_img_output3 = gr.Image(label="encoded")
-                    select_img_input_w2 = gr.Textbox(visible=False)
-                    select_img_input_img2 = gr.Image(visible=False)
+                with gr.Row():
+                    with gr.Column():
+                        gr.Markdown("you'd better upload a standing full-body image")
+                        encode_img_input = gr.Image()
+                        with gr.Group():
+                            encode_img_submit = gr.Button("Run")
+                    with gr.Column():
+                        encode_img_output1 = gr.Textbox(label="message")
+                        with gr.Row():
+                            encode_img_output2 = gr.Image(label="detected")
+                            encode_img_output3 = gr.Image(label="encoded")
+                        select_img_input_w2 = gr.Textbox(visible=False)
+                        select_img_input_img2 = gr.Image(visible=False)
 
             with gr.TabItem("generate video"):
-                with gr.Column():
-                    gr.Markdown("## generate video between 2 images")
-                    with gr.Row():
-                        with gr.Column():
-                            gr.Markdown("please select image 1")
-                            select_img1_dropdown = gr.Dropdown(label="source",
-                                                               choices=["current generated image",
-                                                                        "current encoded image"], type="index")
-                            select_img1_button = gr.Button("select")
-                            select_img1_output_img = gr.Image(label="image 1")
-                            select_img1_output_w = gr.Textbox(visible=False)
-                        with gr.Column():
-                            gr.Markdown("please select image 2")
-                            select_img2_dropdown = gr.Dropdown(label="source",
-                                                               choices=["current generated image",
-                                                                        "current encoded image"], type="index")
-                            select_img2_button = gr.Button("select")
-                            select_img2_output_img = gr.Image(label="image 2")
-                            select_img2_output_w = gr.Textbox(visible=False)
-                    generate_video_frame = gr.Slider(minimum=10, maximum=30, step=1, label="frame", value=10)
-                    generate_video_button = gr.Button("generate")
-                    generate_video_output = gr.Video()
+                with gr.Row():
+                    with gr.Column():
+                        gr.Markdown("## generate video between 2 images")
+                        with gr.Row():
+                            with gr.Column():
+                                gr.Markdown("please select image 1")
+                                select_img1_dropdown = gr.Radio(label="source",
+                                                                choices=["current generated image",
+                                                                         "current encoded image"], type="index")
+                                with gr.Group():
+                                    select_img1_button = gr.Button("select")
+                                select_img1_output_img = gr.Image(label="image 1")
+                                select_img1_output_w = gr.Textbox(visible=False)
+                            with gr.Column():
+                                gr.Markdown("please select image 2")
+                                select_img2_dropdown = gr.Radio(label="source",
+                                                                choices=["current generated image",
+                                                                         "current encoded image"], type="index")
+                                with gr.Group():
+                                    select_img2_button = gr.Button("select")
+                                select_img2_output_img = gr.Image(label="image 2")
+                                select_img2_output_w = gr.Textbox(visible=False)
+                        generate_video_frame = gr.Slider(minimum=10, maximum=30, step=1, label="frame", value=10)
+                        with gr.Group():
+                            generate_video_button = gr.Button("generate")
+                    with gr.Column():
+                        generate_video_output = gr.Video()
 
         gen_submit.click(gen_fn, [gen_input1, gen_input2, gen_input3],
                          [gen_output1, select_img_input_w1, select_img_input_img1])