動画生成タスクの進捗をポーリングするメソッドをKlingAPIクラスに追加し、動画ダウンロード時のエラーハンドリングを強化しました。また、generate_loop_video関数内で使用するモデルを指定するように変更しました。

app.py

@@ -198,6 +198,7 @@ async def generate_loop_video(
             task_id = await kling_api.create_video_task(
                 image_path=current_image_path,
                 prompt=clip_prompt,
+                model="kling-v1-6",
                 duration=clip_duration,
                 tail_image_path=tail_image_path
             )
@@ -206,7 +207,11 @@ async def generate_loop_video(
             video_url = await kling_api.poll_video_result(task_id, progress, progress_val, 0.7 / num_clips)
             
             # 動画をダウンロード
-            clip_path = await kling_api.download_video(video_url, f"clip_{i+1}")
+            clip_output_path = temp_manager.get_temp_path(f"_clip_{i+1}.mp4")
+            success = await kling_api.download_video(video_url, clip_output_path)
+            if not success:
+                raise Exception("動画ダウンロードに失敗しました")
+            clip_path = clip_output_path
             video_paths.append(clip_path)
             temp_manager.add_temp_file(clip_path)
             

utils/kling_api.py

@@ -702,4 +702,35 @@ class KlingAPI:
     
     async def close(self):
         """クライアントを閉じる"""
-        await self.client.aclose() 
+        await self.client.aclose()
+
+    # === 追加: 動画タスク結果をポーリング（進捗バー連携） ===
+    async def poll_video_result(
+        self,
+        task_id: str,
+        progress_callback=None,
+        base_progress: float = 0.0,
+        progress_delta: float = 0.2,
+        max_wait_minutes: int = 60
+    ) -> Optional[str]:
+        """動画生成タスクの完了を待機してURLを返す
+        progress_callback には gr.Progress インスタンスを渡すことを想定
+        base_progress はポーリング開始時点の進捗値、
+        progress_delta は完了時に上乗せする進捗幅"""
+        try:
+            # ポーリング開始を通知
+            if progress_callback:
+                progress_callback(base_progress, "動画生成を待機中...")
+
+            video_url = await self.poll_task_result(task_id, max_wait_minutes=max_wait_minutes)
+
+            # 完了を通知（進捗を一気に上げる）
+            if progress_callback:
+                progress_callback(min(base_progress + progress_delta, 1.0), "動画生成完了")
+
+            return video_url
+        except Exception as e:
+            # エラー時にも進捗を更新
+            if progress_callback:
+                progress_callback(base_progress, "動画生成エラー")
+            raise