Update app.py

app.py

@@ -238,7 +238,7 @@ def video_to_audio(video_path, prompt, negative_prompt="music", seed=0, num_step
         traceback.print_exc()
         return video_path
 
-def generate_video_from_image(image, prompt="", length=4.0, sound_generation="사운드 없음", sound_prompt="", sound_negative_prompt="music"):
+def generate_video_from_image(image, prompt="", length=4.0):
     if image is None:
         return None
     
@@ -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),
@@ -264,40 +264,31 @@ def generate_video_from_image(image, prompt="", length=4.0, sound_generation="
         
         if result and len(result) > 0:
             video_dict = result[0]
-            video_path = video_dict.get("video") if isinstance(video_dict, dict) else None
-            
-            # 비디오 생성이 성공한 경우에만 사운드 추가
-            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
-                
-                # 사운드 생성을 선택하지 않은 경우 원본 비디오 반환
-                return video_path
+            return video_dict.get("video") if isinstance(video_dict, dict) else None
             
     except Exception as e:
         logging.error(f"Video generation error: {str(e)}")
         return None
 
+def add_sound_to_video(video_path, sound_prompt, sound_negative_prompt="music"):
+    if not video_path or not MMAUDIO_LOADED:
+        return video_path
+    
+    try:
+        return 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=8.0  # 기본값 사용
+        )
+    except Exception as e:
+        logging.error(f"Sound addition error: {str(e)}")
+        return video_path
+
+
 def prepare_image_and_mask(image, width, height, overlap_percentage, alignment):
     """이미지와 마스크를 준비하는 함수"""
     if image is None:
@@ -626,12 +617,22 @@ with demo:
         [sound_generation],
         [sound_options]
     )
-    
+
+
     video_btn.click(
         generate_video_from_image,
-        [output_image, video_prompt, video_length, sound_generation, sound_prompt, sound_negative_prompt],
+        [output_image, video_prompt, video_length],  # 원래대로 3개 매개변수만
         [output_video]
     )
+
+# 사운드 추가는 별도 버튼으로
+    sound_btn = gr.Button("🔊 비디오에 사운드 추가", visible=False)
+    sound_btn.click(
+        add_sound_to_video,
+        [output_video, sound_prompt, sound_negative_prompt],
+        [output_video]
+    )
+
     
     # 이벤트 연결 - 두 번째 탭
     outpaint_size_preset.change(update_dimensions, [outpaint_size_preset], [outpaint_width, outpaint_height])