Update app.py

app.py

@@ -5,9 +5,7 @@ from PIL import Image
 from diffusers import DiffusionPipeline
 import random
 from transformers import pipeline
-import pygame
-import os
-import threading
+
 
 torch.backends.cudnn.deterministic = True
 torch.backends.cudnn.benchmark = False
@@ -25,22 +23,7 @@ pipe.load_lora_weights(lora_repo)
 
 pipe.to("cuda")
 
-# pygame 초기화 및 음악 설정
-pygame.mixer.init()
-def play_music():
-    pygame.mixer.music.load("1.mp3")
-    pygame.mixer.music.play()
-    pygame.mixer.music.queue("2.mp3")
-    pygame.mixer.music.set_endevent(pygame.USEREVENT)
-    while True:
-        for event in pygame.event.get():
-            if event.type == pygame.USEREVENT:
-                pygame.mixer.music.queue("1.mp3")
-                pygame.mixer.music.queue("2.mp3")
 
-# 배경음악 재생 시작 (별도 스레드에서 실행)
-music_thread = threading.Thread(target=play_music, daemon=True)
-music_thread.start()
 
 MAX_SEED = 2**32-1
 
@@ -189,9 +172,40 @@ setInterval(createSnowflake, 100);
 """
 
 with gr.Blocks(css=css) as app:
+    # JavaScript 눈 효과
     gr.HTML(f"<script>{js_code}</script>")
     
+    # HTML5 오디오 요소 추가
+    gr.HTML("""
+        <audio id="bgMusic1" loop style="display: none;">
+            <source src="file/1.mp3" type="audio/mp3">
+        </audio>
+        <audio id="bgMusic2" loop style="display: none;">
+            <source src="file/2.mp3" type="audio/mp3">
+        </audio>
+        <script>
+            window.addEventListener('load', function() {
+                const audio1 = document.getElementById('bgMusic1');
+                const audio2 = document.getElementById('bgMusic2');
+                
+                audio1.addEventListener('ended', function() {
+                    audio2.play();
+                });
+                
+                audio2.addEventListener('ended', function() {
+                    audio1.play();
+                });
+                
+                // 자동 재생 시작
+                audio1.play().catch(function(error) {
+                    console.log("Audio playback failed:", error);
+                });
+            });
+        </script>
+    """)
+    
     with gr.Column(elem_classes="container"):
+
         gr.Markdown("# 🎄 X-MAS LoRA", elem_classes="header")
         
         # 프롬프트 입력 박스를 별도로 배치