Update app.py

app.py

@@ -21,18 +21,6 @@ logging.basicConfig(
 )
 
 
-# 프로젝트 루트 디렉토리에서 samples 폴더 경로 설정
-samples_dir = os.path.join(os.path.dirname(__file__), 'samples')
-sample_audio_path = os.path.join(samples_dir, 'metal.mp3')
-
-# samples 디렉토리가 없다면 생성
-os.makedirs(samples_dir, exist_ok=True)
-
-# 예제 음악이 존재하는지 확인
-if os.path.exists(sample_audio_path):
-    gr.Audio(sample_audio_path, label="Sample Music")
-else:
-    gr.Markdown("### Sample music file not found")
     
 ################################
 # 기존에 정의된 함수 및 로직들  #
@@ -466,9 +454,16 @@ def update_info(lyrics):
     )
 
 def main():
-    # 시스템 초기화 (필요한 모델 다운로드/설치 등)
+    # 시스템 초기화
     initialize_system()
 
+    # samples 디렉토리 및 예제 파일 처리
+    current_dir = os.path.dirname(os.path.abspath(__file__))
+    samples_dir = os.path.join(current_dir, 'samples')
+    sample_audio_path = os.path.join(samples_dir, 'metal.mp3')
+    
+    os.makedirs(samples_dir, exist_ok=True)
+
     with gr.Blocks(css="""
         /* 전체 배경 및 컨테이너 스타일 */
         body {
@@ -507,6 +502,22 @@ def main():
         </div>
         """)
 
+        # 예제 음악 섹션
+        with gr.Group():
+            gr.HTML("""
+            <div style="padding: 1rem; margin-bottom: 1.5rem; background-color: #f8f9fa; border-radius: 8px; text-align: center;">
+                <h3 style="margin: 0;">Sample Generated Music</h3>
+                <p style="color: #666; margin: 5px 0;">Listen to this example</p>
+            </div>
+            """)
+            if os.path.exists(sample_audio_path):
+                gr.Audio(
+                    value=sample_audio_path,
+                    label="Sample Music",
+                    type="filepath"
+                )
+            else:
+                gr.Markdown("### Sample music file not available")
 
         with gr.Row():
             # 왼쪽 입력 컬럼
@@ -524,7 +535,6 @@ def main():
 
             # 오른쪽 설정/정보 컬럼
             with gr.Column():
-                # 여기서 gr.Box -> gr.Group로 변경
                 with gr.Group():
                     gr.Markdown("### Generation Settings")
                     num_segments = gr.Number(
@@ -544,7 +554,6 @@ def main():
                         interactive=False
                     )
                 
-                # 여기서도 gr.Box -> gr.Group로 변경
                 with gr.Group():
                     gr.Markdown("### Song Info")
                     duration_info = gr.Label(label="Estimated Duration")
@@ -552,7 +561,6 @@ def main():
                 
                 submit_btn = gr.Button("Generate Music", variant="primary")
 
-        # 아래도 gr.Box -> gr.Group로 변경
         with gr.Group():
             music_out = gr.Audio(label="Generated Audio")
         
@@ -565,7 +573,6 @@ def main():
 [verse]
 Under the city lights, your hand in mine
 Every step we take, feels like a sign
-
 [chorus]
 Baby, you're my everything, my heart is yours
 """
@@ -589,7 +596,6 @@ Baby, you're my everything, my heart is yours
 [verse]
 夜の街に光る君の笑顔
 どんな時もそばにいるよ
-
 [chorus]
 この気持ち止められない
 """
@@ -601,7 +607,6 @@ Baby, you're my everything, my heart is yours
 [verse]
 夜色温柔像你的拥抱
 心跳随着你慢慢变高
-
 [chorus]
 永远不要放开我的手
 """