Update app.py

app.py

@@ -137,7 +137,71 @@ examples = [
 def use_prompt_and_image(prompt, image):
     return prompt, image
 
-
+with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
+    with gr.Tab("Generate"):
+        with gr.Row():
+            input_image = gr.Image(label="Upload an image", type="filepath")
+            input_text = gr.Textbox(label="Enter your prompt for video generation")
+            output_video = gr.Video(label="Generated Video")
+        
+        with gr.Row():
+            steps = gr.Slider(minimum=1, maximum=100, step=1, label="Steps", value=30)
+            cfg_scale = gr.Slider(minimum=1, maximum=15, step=0.1, label="CFG Scale", value=3.5)
+            eta = gr.Slider(minimum=0, maximum=1, step=0.1, label="ETA", value=1)
+            fs = gr.Slider(minimum=1, maximum=30, step=1, label="FPS", value=8)
+            seed = gr.Slider(minimum=0, maximum=1000000, step=1, label="Seed", value=123)
+            video_length = gr.Slider(minimum=1, maximum=10, step=1, label="Video Length (seconds)", value=2)
+        
+        with gr.Row():
+            for prompt, image_file in examples:
+                with gr.Column():
+                    gr.Image(image_file, label=prompt[:50] + "...")
+                    gr.Button("Use this example").click(
+                        fn=use_prompt_and_image,
+                        inputs=[],
+                        outputs=[input_text, input_image],
+                        api_name=False
+                    ).then(
+                        lambda p=prompt, i=image_file: (p, i),
+                        inputs=[],
+                        outputs=[input_text, input_image]
+                    )
+    
+    with gr.Tab("Gallery"):
+        gallery = gr.Gallery(
+            label="Generated Videos",
+            show_label=False,
+            elem_id="gallery",
+            columns=[5],
+            rows=[3],
+            object_fit="contain",
+            height="auto"
+        )
+        selected_video = gr.Video(label="Selected Video")
+        refresh_btn = gr.Button("Refresh Gallery")
+
+    def update_gallery():
+        return load_gallery()
+
+    def show_video(evt: gr.SelectData):
+        return evt.value[2]  # 선택된 썸네일에 해당하는 비디오 경로 반환
+
+    refresh_btn.click(fn=update_gallery, inputs=None, outputs=gallery)
+    demo.load(fn=update_gallery, inputs=None, outputs=gallery)
+    gallery.select(show_video, None, selected_video)
+
+    input_text.submit(
+        fn=respond, 
+        inputs=[input_image, input_text, steps, cfg_scale, eta, fs, seed, video_length], 
+        outputs=output_video
+    ).then(
+        fn=update_gallery,
+        inputs=None,
+        outputs=gallery
+    )
+
+if __name__ == "__main__":
+    demo.launch()