Update app.py

app.py

@@ -82,12 +82,17 @@ def generate_image_from_text(prompt, seed=0):
     return image
 
 @spaces.GPU(duration=300)
-def infer(image, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, seed=123, video_length=2):
+def infer_t2v(prompt, video_prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, seed=123, video_length=2):
     # 한글 입력 감지 및 번역
     if any('\u3131' <= char <= '\u318E' or '\uAC00' <= char <= '\uD7A3' for char in prompt):
         translated = translator(prompt, max_length=512)[0]['translation_text']
         prompt = translated
-        print(f"Translated prompt: {prompt}")
+        print(f"Translated image prompt: {prompt}")
+    if any('\u3131' <= char <= '\u318E' or '\uAC00' <= char <= '\uD7A3' for char in video_prompt):
+        translated_video_prompt = translator(video_prompt, max_length=512)[0]['translation_text']
+        video_prompt = translated_video_prompt
+        print(f"Translated video prompt: {video_prompt}")
+
 
     resolution = (576, 1024)
     save_fps = 8
@@ -157,7 +162,35 @@ i2v_examples = [
 css = """#input_img {max-width: 1024px !important} #output_vid {max-width: 1024px; max-height: 576px}"""
 
 with gr.Blocks(analytics_enabled=False, css=css) as dynamicrafter_iface:
-    gr.Markdown("이미지로 영상 생성 테스트 (한글 프롬프트 지원)")
+    gr.Markdown("kAI 무비 스튜디오")
+
+    with gr.Tab(label='T2V'):
+        with gr.Column():
+            with gr.Row():
+                with gr.Column():
+                    with gr.Row():
+                        t2v_input_text = gr.Text(label='Image Generation Prompt')
+                    with gr.Row():
+                        t2v_video_prompt = gr.Text(label='Video Generation Prompt')
+                    with gr.Row():
+                        t2v_seed = gr.Slider(label='Random Seed', minimum=0, maximum=10000, step=1, value=123)
+                        t2v_eta = gr.Slider(minimum=0.0, maximum=1.0, step=0.1, label='ETA', value=1.0)
+                        t2v_cfg_scale = gr.Slider(minimum=1.0, maximum=15.0, step=0.5, label='CFG Scale', value=7.5)
+                    with gr.Row():
+                        t2v_steps = gr.Slider(minimum=1, maximum=50, step=1, label="Sampling steps", value=30)
+                        t2v_motion = gr.Slider(minimum=5, maximum=20, step=1, label="FPS", value=8)
+                    with gr.Row():
+                        t2v_video_length = gr.Slider(minimum=2, maximum=8, step=1, label="Video Length (seconds)", value=2)
+                    t2v_end_btn = gr.Button("Generate")
+                with gr.Row():
+                    t2v_output_video = gr.Video(label="Generated Video", autoplay=True, show_share_button=True)
+
+        t2v_end_btn.click(
+            inputs=[t2v_input_text, t2v_video_prompt, t2v_steps, t2v_cfg_scale, t2v_eta, t2v_motion, t2v_seed, t2v_video_length],
+            outputs=[t2v_output_video],
+            fn=infer_t2v
+        )
+    
     with gr.Tab(label='ImageAnimation_576x1024'):
         with gr.Column():
             with gr.Row():
@@ -190,31 +223,6 @@ with gr.Blocks(analytics_enabled=False, css=css) as dynamicrafter_iface:
                         fn = infer
         )
 
-    with gr.Tab(label='T2V'):
-        with gr.Column():
-            with gr.Row():
-                with gr.Column():
-                    with gr.Row():
-                        t2v_input_text = gr.Text(label='Image Generation Prompt')
-                    with gr.Row():
-                        t2v_video_prompt = gr.Text(label='Video Generation Prompt')
-                    with gr.Row():
-                        t2v_seed = gr.Slider(label='Random Seed', minimum=0, maximum=10000, step=1, value=123)
-                        t2v_eta = gr.Slider(minimum=0.0, maximum=1.0, step=0.1, label='ETA', value=1.0)
-                        t2v_cfg_scale = gr.Slider(minimum=1.0, maximum=15.0, step=0.5, label='CFG Scale', value=7.5)
-                    with gr.Row():
-                        t2v_steps = gr.Slider(minimum=1, maximum=50, step=1, label="Sampling steps", value=30)
-                        t2v_motion = gr.Slider(minimum=5, maximum=20, step=1, label="FPS", value=8)
-                    with gr.Row():
-                        t2v_video_length = gr.Slider(minimum=2, maximum=8, step=1, label="Video Length (seconds)", value=2)
-                    t2v_end_btn = gr.Button("Generate")
-                with gr.Row():
-                    t2v_output_video = gr.Video(label="Generated Video", autoplay=True, show_share_button=True)
 
-        t2v_end_btn.click(
-            inputs=[t2v_input_text, t2v_video_prompt, t2v_steps, t2v_cfg_scale, t2v_eta, t2v_motion, t2v_seed, t2v_video_length],
-            outputs=[t2v_output_video],
-            fn=infer_t2v
-        )
 
 dynamicrafter_iface.queue(max_size=12).launch(show_api=True)