Update app.py

app.py

@@ -152,7 +152,107 @@ def generate_image(
     return image, debug_images, seed
 
 
-# Custom CSS for pastel theme
+# -----------------------------
+# (1) 여기에 영상 API 호출을 위한 추가 코드
+# -----------------------------
+import requests
+import random
+import tempfile
+import subprocess
+from gradio_client import Client, handle_file
+
+# 예시: 원격 서버 Endpoint (필요하다면 수정)
+REMOTE_ENDPOINT = "http://211.233.58.201:7875/"
+
+client = Client(REMOTE_ENDPOINT)
+
+def run_process_video_api(image_path: str, prompt: str, video_length: float = 2.0):
+    """
+    원격 /process 엔드포인트 호출하여 영상을 생성.
+    (예시: prompt, negative_prompt, seed 등은 하드코딩하거나 원하는대로 조정 가능)
+    """
+    # 랜덤 시드
+    seed_val = random.randint(0, 9999999)
+    # negative_prompt = "" 등 고정
+    negative_prompt = ""
+    use_teacache = True
+
+    # /process 호출 (gradio_client)
+    result = client.predict(
+        input_image=handle_file(image_path),
+        prompt=prompt,
+        n_prompt=negative_prompt,
+        seed=seed_val,
+        use_teacache=use_teacache,
+        video_length=video_length,
+        api_name="/process",
+    )
+    # result는 (video_dict, preview_dict, md_text, html_text) 구조
+    video_dict, preview_dict, md_text, html_text = result
+    video_path = video_dict.get("video") if isinstance(video_dict, dict) else None
+    return video_path
+
+def add_watermark_to_video(input_video_path: str, watermark_text="Ginigen.com") -> str:
+    """
+    FFmpeg로 영상에 오른쪽 하단 워터마크를 추가한 새 영상을 리턴
+    """
+    if not os.path.exists(input_video_path):
+        raise FileNotFoundError(f"Input video not found: {input_video_path}")
+
+    # 출력 경로
+    base, ext = os.path.splitext(input_video_path)
+    watermarked_path = base + "_wm" + ext
+    # ffmpeg 명령어 구성
+    # - y: 덮어쓰기
+    # drawtext 필터로 오른쪽 하단(x=w-tw-10, y=h-th-10)에 boxcolor=black 반투명 박스
+    cmd = [
+        "ffmpeg", "-y",
+        "-i", input_video_path,
+        "-vf", f"drawtext=fontsize=20:fontcolor=white:text='{watermark_text}':x=w-tw-10:y=h-th-10:box=1:[email protected]:boxborderw=5",
+        "-codec:a", "copy",
+        watermarked_path
+    ]
+    try:
+        subprocess.run(cmd, check=True)
+    except Exception as e:
+        print(f"[WARN] FFmpeg watermark failed: {e}")
+        return input_video_path  # 실패 시 원본 반환
+
+    return watermarked_path
+
+def generate_video_from_image(image_array: np.ndarray):
+    """
+    1) Numpy 이미지를 임시 파일로 저장
+    2) 원격 API로 2초 영상 생성 (기본 prompt 고정)
+    3) FFmpeg로 'Ginigen.com' 워터마크 추가
+    4) 최종 mp4 경로 반환
+    """
+    if image_array is None:
+        raise gr.Error("이미지가 없습니다.")
+
+    # (1) 임시 파일로 저장
+    with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as fp:
+        temp_img_path = fp.name
+        Image.fromarray(image_array).save(temp_img_path, format="PNG")
+
+    # (2) 원격 API 호출
+    default_video_prompt = "Generate a video with smooth and natural movement. Objects should have visible motion while maintaining fluid transitions."
+    result_video_path = run_process_video_api(
+        image_path=temp_img_path,
+        prompt=default_video_prompt,
+        video_length=2.0,
+    )
+    if result_video_path is None:
+        raise gr.Error("영상 API 호출 실패 또는 결과 없음")
+
+    # (3) FFmpeg 워터마크 추가
+    final_video = add_watermark_to_video(result_video_path, watermark_text="Ginigen.com")
+    return final_video
+
+
+# -----------------------------
+# Custom CSS, Headers, etc.
+# -----------------------------
 _CUSTOM_CSS_ = """
 :root {
     --primary-color: #f8c3cd;            /* Sakura pink - primary accent */
@@ -419,7 +519,6 @@ def create_demo():
 
         with gr.Row():
             with gr.Column(scale=6):
-                # Input panel - using a Group div with custom class instead of Box
                 with gr.Group(elem_id="input-panel", elem_classes="panel-box"):
                     gr.Markdown("### 📸 Reference Images")
                     with gr.Row():
@@ -461,19 +560,21 @@ def create_demo():
                     gr.HTML(_CITE_)
 
             with gr.Column(scale=6):
-                # Output panel - using a Group div with custom class instead of Box
                 with gr.Group(elem_id="output-panel", elem_classes="panel-box"):
                     gr.Markdown("### 🖼️ Generated Result")
                     output_image = gr.Image(label="Generated Image", elem_id="output-image", format='png')
                     seed_output = gr.Textbox(label="Used Seed", elem_id="seed-output")
                     
+                    # (2) 영상 생성 버튼 & 출력 영역 추가
+                    generate_video_btn = gr.Button("🎬 Generate Video from Image")
+                    output_video = gr.Video(label="Generated Video", elem_id="video-output")
+
                     gr.Markdown("### 🔍 Preprocessing")
                     debug_image = gr.Gallery(
                         label="Preprocessing Results (including face crop and background removal)",
                         elem_id="debug-gallery",
                     )
 
-        # Examples panel - using a Group div with custom class instead of Box
         with gr.Group(elem_id="examples-panel", elem_classes="panel-box"):
             gr.Markdown("## 📚 Examples")
             example_inps = [
@@ -559,6 +660,7 @@ def create_demo():
                 fn=generate_image,
             )
 
+        # 기존 이미지 생성 함수와 연결
         generate_btn.click(
             fn=generate_image,
             inputs=[
@@ -583,9 +685,22 @@ def create_demo():
             outputs=[output_image, debug_image, seed_output],
         )
 
+        # (3) 영상 생성 버튼 클릭 -> generate_video_from_image() 호출
+        def on_click_generate_video(img):
+            if img is None:
+                raise gr.Error("먼저 이미지를 생성해주세요.")
+            video_path = generate_video_from_image(img)
+            return video_path
+
+        generate_video_btn.click(
+            fn=on_click_generate_video,
+            inputs=[output_image],
+            outputs=[output_video],
+        )
+
     return demo
 
 
 if __name__ == '__main__':
     demo = create_demo()
-    demo.launch()
+    demo.launch(server_port=args.port)