Update app.py

app.py

@@ -456,9 +456,8 @@ def main():
         secondary_hue="purple",
         neutral_hue="slate",
         font=["Arial", "sans-serif"]
-    ), css=css) as demo:    # 여기 괄호 수정
+    ), css=css) as demo:
         with gr.Column(elem_id="main-container"):
- 
             # 헤더 섹션
             with gr.Row(elem_id="header"):
                 gr.Markdown(
@@ -524,13 +523,14 @@ def main():
 
                 # 히스토리 탭
                 with gr.TabItem("📚 History", id="history"):
-                    history_list = gr.Dataset(
-                        components=[gr.Audio, gr.Textbox, gr.Textbox],
-                        headers=["Generated Music", "Genre", "Lyrics"],
-                        samples=[],
-                        elem_id="history-list"
-                    )
-                    gr.Markdown("*Click on any entry to play the music*")
+                    with gr.Row():
+                        history_container = gr.HTML("""
+                            <div id="history-container" style="width: 100%; padding: 10px;">
+                                <h3>Generation History</h3>
+                                <div id="history-list"></div>
+                            </div>
+                        """)
+                    history_state = gr.State([])
 
             # 예제 섹션
             with gr.Accordion("📖 Examples", open=False):
@@ -550,8 +550,7 @@ Don't let this moment fade, hold me close tonight
                         ],
                         [
                             "K-pop bright energetic synth dance electronic",
-                            """
-[verse]
+                            """[verse]
 언젠가 마주한 눈빛 속에서
 어두운 밤을 지날 때마다
 
@@ -603,8 +602,30 @@ Don't let this moment fade, hold me close tonight
                 f"📊 Verses: {sections['verse']}, Chorus: {sections['chorus']}"
             )
 
-        def update_history(audio, genre, lyrics):
-            return history_list.update(samples=[[audio, genre, lyrics]] + history_list.samples)
+        def update_history(audio_path, genre, lyrics, history):
+            if audio_path:
+                new_entry = {
+                    "audio": audio_path,
+                    "genre": genre,
+                    "lyrics": lyrics,
+                    "timestamp": gr.utils.get_datetime()
+                }
+                history = [new_entry] + (history or [])
+                
+                history_html = "<div class='history-entries'>"
+                for entry in history:
+                    history_html += f"""
+                        <div class='history-entry' style='margin: 10px 0; padding: 10px; border: 1px solid #ddd; border-radius: 8px;'>
+                            <audio controls src='{entry["audio"]}'></audio>
+                            <div style='margin-top: 5px;'><strong>Genre:</strong> {entry["genre"]}</div>
+                            <div style='margin-top: 5px;'><strong>Lyrics:</strong><pre>{entry["lyrics"]}</pre></div>
+                            <div style='color: #666; font-size: 0.9em;'>{entry["timestamp"]}</div>
+                        </div>
+                    """
+                history_html += "</div>"
+                
+                return history, history_html
+            return history, ""
 
         # 이벤트 핸들러
         lyrics_txt.change(
@@ -613,25 +634,25 @@ Don't let this moment fade, hold me close tonight
             outputs=[duration_info, sections_info]
         )
         
-        def generate_with_progress(genre, lyrics, num_segments, max_tokens):
+        def generate_with_progress(genre, lyrics, num_segments, max_tokens, history):
             progress.update(value="🎵 Starting generation...")
             try:
                 result = infer(genre, lyrics, num_segments, max_tokens)
                 if result:
                     progress.update(value="✅ Generation complete!")
-                    update_history(result, genre, lyrics)
-                    return result
+                    new_history, history_html = update_history(result, genre, lyrics, history)
+                    return result, new_history, history_html
                 else:
                     progress.update(value="❌ Generation failed")
-                    return None
+                    return None, history, ""
             except Exception as e:
                 progress.update(value=f"❌ Error: {str(e)}")
-                return None
+                return None, history, ""
 
         submit_btn.click(
             fn=generate_with_progress,
-            inputs=[genre_txt, lyrics_txt, num_segments, max_new_tokens],
-            outputs=[music_out]
+            inputs=[genre_txt, lyrics_txt, num_segments, max_new_tokens, history_state],
+            outputs=[music_out, history_state, history_container]
         )
 
         return demo