Update app.py

app.py

@@ -639,28 +639,34 @@ Don't let this moment fade, hold me close tonight
             inputs=[lyrics_txt],
             outputs=[duration_info, sections_info]
         )
-        
+
+
         def generate_with_progress(genre, lyrics, segments, tokens, history):
-            progress.update(value="🎵 Starting generation...")
             try:
+                # 진행 상태를 반환값에 포함
+                status_text = "🎵 Starting generation..."
                 result = infer(genre, lyrics, segments, tokens)
+        
                 if result:
-                    progress.update(value="✅ Generation complete!")
+                    status_text = "✅ Generation complete!"
                     new_history, history_html = update_history(result, genre, lyrics, history)
-                    return result, new_history, history_html
+                    return result, new_history, history_html, status_text
                 else:
-                    progress.update(value="❌ Generation failed")
-                    return None, history, ""
+                    status_text = "❌ Generation failed"
+                    return None, history, "", status_text
             except Exception as e:
-                progress.update(value=f"❌ Error: {str(e)}")
-                return None, history, ""
+                status_text = f"❌ Error: {str(e)}"
+                return None, history, "", status_text
 
+        # submit_btn.click 이벤트 핸들러도 수정
         submit_btn.click(
             fn=generate_with_progress,
             inputs=[genre_txt, lyrics_txt, num_segments, max_new_tokens, history_state],
-            outputs=[music_out, history_state, history_container]
+            outputs=[music_out, history_state, history_container, progress]
         )
 
+
+
         return demo
 
 if __name__ == "__main__":