Update app.py

app.py

@@ -248,7 +248,7 @@ def generate_video_from_image(image, prompt="", length=4.0, sound_generation="
             temp_path = fp.name
             Image.fromarray(image).save(temp_path)
         
-        # 비디오 생성 API 호출
+        # API 호출 (원래 코드와 동일)
         client = Client(VIDEO_API_URL)
         result = client.predict(
             input_image=handle_file(temp_path),
@@ -266,27 +266,33 @@ def generate_video_from_image(image, prompt="", length=4.0, sound_generation="
             video_dict = result[0]
             video_path = video_dict.get("video") if isinstance(video_dict, dict) else None
             
-            # 사운드 생성 옵션이 선택된 경우
-            if video_path and sound_generation == "사운드 생성" and MMAUDIO_LOADED:
-                # 사운드 프롬프트가 비어있으면 기본값 사용
-                if not sound_prompt:
-                    sound_prompt = prompt if prompt else "ambient sound"
+            # 비디오 생성이 성공한 경우에만 사운드 추가
+            if video_path:
+                # 사운드 생성 옵션이 선택된 경우
+                if sound_generation == "사운드 생성" and MMAUDIO_LOADED:
+                    try:
+                        # 사운드 프롬프트가 비어있으면 기본값 사용
+                        if not sound_prompt:
+                            sound_prompt = prompt if prompt else "ambient sound"
+                        
+                        # 비디오에 사운드 추가
+                        video_with_sound = video_to_audio(
+                            video_path=video_path, 
+                            prompt=sound_prompt, 
+                            negative_prompt=sound_negative_prompt,
+                            seed=random.randint(0, 9999999),
+                            num_steps=25,
+                            cfg_strength=4.5,
+                            target_duration=length
+                        )
+                        return video_with_sound
+                    except Exception as e:
+                        logging.error(f"Sound generation error: {str(e)}")
+                        # 사운드 생성 실패 시 원본 비디오 반환
+                        return video_path
                 
-                # 비디오에 사운드 추가 - 모든 매개변수를 명시적으로 전달
-# 비디오에 사운드 추가
-                video_with_sound = video_to_audio(
-                    video_path=video_path, 
-                    prompt=sound_prompt, 
-                    negative_prompt=sound_negative_prompt,
-                    seed=random.randint(0, 9999999),
-                    num_steps=25,
-                    cfg_strength=4.5,
-                    target_duration=length  # duration → target_duration
-                )
-
-                return video_with_sound
-            
-            return video_path
+                # 사운드 생성을 선택하지 않은 경우 원본 비디오 반환
+                return video_path
             
     except Exception as e:
         logging.error(f"Video generation error: {str(e)}")