Update app.py

app.py

@@ -41,7 +41,9 @@ from moviepy.editor import (
     TextClip,
     concatenate_videoclips
 )
-
+# MoviePy 설정
+from moviepy.config import change_settings
+change_settings({"IMAGEMAGICK_BINARY": "convert"})  # ImageMagick 바이너리 설정
 # .env 파일에서 환경 변수 로드
 load_dotenv()
 
@@ -901,7 +903,7 @@ def analyze_top_spaces(progress=gr.Progress()) -> Tuple[str, str]:
                 
                 response = hf_client.chat_completion(
                     messages,
-                    max_tokens=200,
+                    max_tokens=1000,
                     temperature=0.7
                 )
                 
@@ -1029,66 +1031,95 @@ def create_editable_space_analysis(progress=gr.Progress()) -> List[str]:
 def generate_video(texts: List[str], progress=gr.Progress()) -> str:
     """영상 생성"""
     try:
+        # 임시 디렉토리 생성
         temp_dir = tempfile.mkdtemp()
         clips = []
         
-        # 인트로 생성 (이미지 대신 텍스트로)
-        intro_clip = TextClip(
-            txt=texts[0],
-            fontsize=30,
-            color='white',
-            bg_color='black',
-            size=(800, 600)
-        ).set_duration(5)
-        
-        # 인트로 음성
-        intro_audio = gTTS(text=texts[0], lang='ko', slow=False)
-        intro_audio_path = os.path.join(temp_dir, "intro.mp3")
-        intro_audio.save(intro_audio_path)
-        intro_audio_clip = AudioFileClip(intro_audio_path)
-        intro_clip = intro_clip.set_audio(intro_audio_clip)
-        clips.append(intro_clip)
-        
-        # 각 스페이스별 클립 생성
-        for idx, text in enumerate(texts[1:], 1):
-            progress((idx / 24), desc=f"영상 생성 중... {idx}/24")
-            
-            # 텍스트 클립 생성
-            text_clip = TextClip(
-                txt=text,
-                fontsize=25,
+        # 작업 디렉토리를 임시 디렉토리로 변경
+        current_dir = os.getcwd()
+        os.chdir(temp_dir)
+        
+        try:
+            # 인트로 생성
+            intro_text = texts[0]
+            intro_clip = TextClip(
+                txt=intro_text,
+                fontsize=30,
                 color='white',
                 bg_color='black',
-                size=(800, 600)
-            ).set_duration(5)
+                size=(800, 600),
+                method='label'  # 'caption' 대신 'label' 사용
+            ).set_duration(10)  # 인트로는 좀 더 길게
             
-            # 음성 생성
-            tts = gTTS(text=text, lang='ko', slow=False)
-            audio_path = os.path.join(temp_dir, f"audio_{idx}.mp3")
-            tts.save(audio_path)
-            audio_clip = AudioFileClip(audio_path)
+            # 인트로 음성
+            intro_audio = gTTS(text=intro_text, lang='ko', slow=False)
+            intro_audio_path = os.path.join(temp_dir, "intro.mp3")
+            intro_audio.save(intro_audio_path)
+            intro_audio_clip = AudioFileClip(intro_audio_path)
+            
+            # 오디오 길이에 맞춰 비디오 길이 조정
+            intro_clip = intro_clip.set_duration(intro_audio_clip.duration)
+            intro_clip = intro_clip.set_audio(intro_audio_clip)
+            clips.append(intro_clip)
+            
+            # 각 스페이스별 클립 생성
+            for idx, text in enumerate(texts[1:], 1):
+                progress((idx / 24), desc=f"영상 생성 중... {idx}/24")
+                
+                # 텍스트 클립 생성
+                text_clip = TextClip(
+                    txt=text,
+                    fontsize=25,
+                    color='white',
+                    bg_color='black',
+                    size=(800, 600),
+                    method='label'  # 'caption' 대신 'label' 사용
+                ).set_duration(5)
+                
+                # 음성 생성
+                tts = gTTS(text=text, lang='ko', slow=False)
+                audio_path = os.path.join(temp_dir, f"audio_{idx}.mp3")
+                tts.save(audio_path)
+                audio_clip = AudioFileClip(audio_path)
+                
+                # 오디오 길이에 맞춰 비디오 길이 조정
+                text_clip = text_clip.set_duration(audio_clip.duration)
+                video_clip = text_clip.set_audio(audio_clip)
+                clips.append(video_clip)
+            
+            # 모든 클립 연결
+            final_clip = concatenate_videoclips(clips)
+            
+            # MP4로 저장
+            output_path = os.path.join(temp_dir, "output_video.mp4")
+            final_clip.write_videofile(
+                output_path,
+                fps=24,
+                codec='libx264',
+                audio_codec='aac'
+            )
+            
+            # 결과 파일을 현재 디렉토리로 복사
+            final_output = os.path.join(current_dir, "output_video.mp4")
+            shutil.copy2(output_path, final_output)
+            
+            return final_output
+            
+        finally:
+            # 작업 디렉토리 복원
+            os.chdir(current_dir)
             
-            # 결합
-            video_clip = text_clip.set_audio(audio_clip)
-            clips.append(video_clip)
-        
-        # 모든 클립 연결
-        final_clip = concatenate_videoclips(clips)
-        
-        # MP4로 저장
-        output_path = "output_video.mp4"
-        final_clip.write_videofile(output_path, fps=24, codec='libx264')
-        
-        # 임시 파일 정리
-        shutil.rmtree(temp_dir)
-        
-        return output_path
-        
     except Exception as e:
         print(f"Error generating video: {e}")
-        if 'temp_dir' in locals():
-            shutil.rmtree(temp_dir)
         return ""
+        
+    finally:
+        # 임시 파일 정리
+        if 'temp_dir' in locals():
+            try:
+                shutil.rmtree(temp_dir)
+            except Exception as e:
+                print(f"Error cleaning up temp files: {e}")
 
 def create_interface():
     with gr.Blocks(title="HuggingFace Trending Board", css="""