Update app.py

app.py

@@ -1,5 +1,4 @@
-
-# Part 1/4 - Imports and Initial Setup
+```python
 import gradio as gr
 import numpy as np
 import librosa
@@ -9,6 +8,7 @@ import os
 import requests
 import json
 import time
+import threading
 
 # 데이터 저장을 위한 DB 클래스
 class SimpleDB:
@@ -61,7 +61,6 @@ text_analyzer = pipeline(
     model="nlptown/bert-base-multilingual-uncased-sentiment"
 )
 
-# 상수 정의
 IMAGE_DISPLAY_TIME = 30
 WELCOME_MESSAGE = """
 # 디지털 굿판에 오신 것을 환영합니다
@@ -82,8 +81,6 @@ WORLDVIEW_MESSAGE = """
 온천천 온천장역에서 장전역까지 걸으며 더 깊은 체험이 가능합니다.
 """
 
-
-# Part 2/4 - Analysis Functions
 def calculate_baseline_features(audio_path):
     """기준점 음성 특성 분석"""
     try:
@@ -208,8 +205,6 @@ def analyze_voice(audio_path, state):
     except Exception as e:
         return state, f"오류 발생: {str(e)}", "", "", ""
 
-
-# Part 3/4 - Generation and Processing Functions
 def generate_detailed_prompt(text, emotions, text_sentiment):
     """감정 기반 상세 프롬프트 생성"""
     emotion_colors = {
@@ -260,7 +255,6 @@ def generate_image_from_prompt(prompt):
         
         if response.status_code == 200:
             print("Image generated successfully")
-            # 이미지 저장
             timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
             image_path = f"generated_images/{timestamp}.png"
             os.makedirs("generated_images", exist_ok=True)
@@ -275,24 +269,10 @@ def generate_image_from_prompt(prompt):
         print(f"Error generating image: {str(e)}")
         return None, None
 
-def update_final_prompt(state):
-    """청신의 감상들을 종합하여 최종 프롬프트 업데이트"""
-    combined_prompt = "한국 전통 민화 스타일의 추상화, 온천천에서의 감상과 소원을 담아내기:\n\n"
-    
-    if state.get("reflections"):
-        combined_prompt += "청신의 감상들:\n"
-        for time, text, sentiment in state["reflections"]:
-            combined_prompt += f"- {time}: {text} ({sentiment})\n"
-    
-    if state.get("wish"):
-        combined_prompt += f"\n소원:\n{state['wish']}"
-    
-    return combined_prompt
-
-def handle_image_timeout(result_image, delay=IMAGE_DISPLAY_TIME):
-    """이미지 자동 사라짐 처리"""
-    time.sleep(delay)
-    return gr.update(value=None)
+def handle_image_timeout(result_image):
+    """이미지 자동 사라짐을 처리하는 함수"""
+    time.sleep(IMAGE_DISPLAY_TIME)
+    result_image.update(value=None)
 
 def save_reflection(text, state):
     """감상 저장"""
@@ -313,8 +293,6 @@ def save_reflection(text, state):
         print(f"Error in save_reflection: {str(e)}")
         return state, []
 
-
-# Part 4/4 - Interface and Main
 def create_interface():
     db = SimpleDB()
     
@@ -328,12 +306,10 @@ def create_interface():
             "image_path": None
         })
 
-        # 헤더
         header = gr.Markdown("# 디지털 굿판")
         user_display = gr.Markdown("")
 
         with gr.Tabs() as tabs:
-            # 입장
             with gr.Tab("입장"):
                 gr.Markdown(WELCOME_MESSAGE)
                 name_input = gr.Textbox(
@@ -343,139 +319,133 @@ def create_interface():
                 worldview_display = gr.Markdown(visible=False)
                 start_btn = gr.Button("여정 시작하기")
 
-            # 기준 설정
             with gr.Tab("기준 설정"):
-                gr.Markdown("""### 축원의 문장을 평온한 마음으로 읽어주세요
-                
-                먼저, 평온한 마음으로 축원의 문장을 읽어주세요. 
-                이 축원은 당신에게 평화와 안정을 불러일으키며, 
-                감정을 정확히 이해하기 위한 **기준점**이 될 것입니다.""")
+                gr.Markdown("""### 축원의 문장을 평온한 마음으로 읽어주세요""")
                 gr.Markdown("'당신의 건강과 행복이 늘 가득하기를'")
-                baseline_audio = gr.Audio(
-                    label="축원 문장 녹음하기",
-                    sources=["microphone"]
-                )
-                set_baseline_btn = gr.Button("기준점 설정 완료")
-                baseline_status = gr.Markdown("")
-
-            # 청신
-            with gr.Tab("청신"):
-                gr.Markdown("""## 청신 - 소리로 정화하기
-                
-                온천천의 물소리에 귀 기울이며 **30분간 마음을 정화**해보세요.""")
-                play_music_btn = gr.Button("온천천의 소리 듣기")
-                with gr.Row():
-                    audio = gr.Audio(
-                        value=None,
-                        type="filepath",
-                        label="온천천의 소리",
-                        interactive=False,
-                        autoplay=False
-                    )
-                    with gr.Column():
-                        reflection_input = gr.Textbox(
-                            label="지금 이 순간의 감상을 자유롭게 적어보세요",
-                            lines=3
-                        )
-                        save_btn = gr.Button("감상 저장하기")
-                        reflections_display = gr.Dataframe(
-                            headers=["시간", "감상", "감정 분석"],
-                            label="기록된 감상들"
-                        )
-
-            # 기원
-            with gr.Tab("기원"):
-                gr.Markdown("## 기원 - 소원을 전해보세요")
-                with gr.Row():
-                    with gr.Column():
-                        voice_input = gr.Audio(
-                            label="소원을 나누고 싶은 마음을 말해주세요",
-                            sources=["microphone"]
-                        )
-                        clear_btn = gr.Button("녹음 지우기")
-                        analyze_btn = gr.Button("소원 분석하기")
-                    
-                    with gr.Column():
-                        transcribed_text = gr.Textbox(label="인식된 텍스트")
-                        voice_emotion = gr.Textbox(label="음성 감정 분석")
-                        text_emotion = gr.Textbox(label="텍스트 감정 분석")
-
-            # 송신
-            with gr.Tab("송신"):
-                gr.Markdown("## 송신 - 마음의 그림을 남기고, 보내기")
-                final_prompt = gr.Textbox(
-                    label="생성된 프롬프트",
-                    interactive=False,
-                    lines=3
-                )
-                with gr.Row():
-                    generate_btn = gr.Button("마음의 그림 그리기")
-                    save_image_btn = gr.Button("이미지 저장하기")
-                result_image = gr.Image(label="생성된 이미지")
-                
-                gr.Markdown("## 마지막 감상을 남겨주세요")
-                final_reflection = gr.Textbox(
-                    label="마지막 감상",
-                    placeholder="한 줄로 남겨주세요..."
-                )
-                save_final_btn = gr.Button("감상 남기기")
-
-        # 이벤트 연결
-        start_btn.click(
-            fn=lambda name: (
-                WORLDVIEW_MESSAGE if name.strip() else "이름을 입력해주세요",
-                gr.update(visible=True) if name.strip() else gr.update(),
-                {"user_name": name} if name.strip() else state
-            ),
-            inputs=[name_input],
-            outputs=[worldview_display, tabs, state]
-        )
-
-        set_baseline_btn.click(
-            fn=lambda x, s: ({"baseline_features": calculate_baseline_features(x)}, "기준점이 설정되었습니다."),
-            inputs=[baseline_audio, state],
-            outputs=[state, baseline_status]
-        )
-
-        save_btn.click(
-            fn=save_reflection,
-            inputs=[reflection_input, state],
-            outputs=[state, reflections_display]
-        )
-
-        clear_btn.click(
-            fn=lambda: None,
-            outputs=[voice_input]
-        )
-
-        analyze_btn.click(
-            fn=analyze_voice,
-            inputs=[voice_input, state],
-            outputs=[state, transcribed_text, voice_emotion, text_emotion, final_prompt]
-        )
-
-        generate_btn.click(
-            fn=lambda p: generate_image_from_prompt(p)[0],
-            inputs=[final_prompt],
-            outputs=[result_image]
-        )
-
-        save_final_btn.click(
-            fn=lambda t, s: (db.save_wish(s["user_name"], t), "감상이 저장되었습니다."),
-            inputs=[final_reflection, state],
-            outputs=[gr.Markdown("")]
-        )
-
-        # 이미지 자동 사라짐
-        result_image.change(
-            fn=lambda: gr.update(value=None),
-            inputs=[],
-            outputs=[result_image],
-            _js=f"() => setTimeout(() => {{document.querySelector('#result_image image').src = ''}}, {IMAGE_DISPLAY_TIME*1000})"
-        )
-
-        return app
+               baseline_audio = gr.Audio(
+                   label="축원 문장 녹음하기",
+                   sources=["microphone"]
+               )
+               set_baseline_btn = gr.Button("기준점 설정 완료")
+               baseline_status = gr.Markdown("")
+
+           with gr.Tab("청신"):
+               gr.Markdown("## 청신 - 소리로 정화하기")
+               play_music_btn = gr.Button("온천천의 소리 듣기")
+               with gr.Row():
+                   audio = gr.Audio(
+                       value=None,
+                       type="filepath",
+                       label="온천천의 소리",
+                       interactive=False,
+                       autoplay=False
+                   )
+                   with gr.Column():
+                       reflection_input = gr.Textbox(
+                           label="지금 이 순간의 감상을 자유롭게 적어보세요",
+                           lines=3
+                       )
+                       save_btn = gr.Button("감상 저장하기")
+                       reflections_display = gr.Dataframe(
+                           headers=["시간", "감상", "감정 분석"],
+                           label="기록된 감상들"
+                       )
+
+           with gr.Tab("기원"):
+               gr.Markdown("## 기원 - 소원을 전해보세요")
+               with gr.Row():
+                   with gr.Column():
+                       voice_input = gr.Audio(
+                           label="소원을 나누고 싶은 마음을 말해주세요",
+                           sources=["microphone"]
+                       )
+                       clear_btn = gr.Button("녹음 지우기")
+                       analyze_btn = gr.Button("소원 분석하기")
+                   
+                   with gr.Column():
+                       transcribed_text = gr.Textbox(label="인식된 텍스트")
+                       voice_emotion = gr.Textbox(label="음성 감정 분석")
+                       text_emotion = gr.Textbox(label="텍스트 감정 분석")
+
+           with gr.Tab("송신"):
+               gr.Markdown("## 송신 - 마음의 그림을 남기고, 보내기")
+               final_prompt = gr.Textbox(
+                   label="생성된 프롬프트",
+                   interactive=False,
+                   lines=3
+               )
+               generate_btn = gr.Button("마음의 그림 그리기")
+               result_image = gr.Image(label="생성된 이미지")
+               
+               gr.Markdown("## 마지막 감상을 남겨주세요")
+               final_reflection = gr.Textbox(
+                   label="마지막 감상",
+                   placeholder="한 줄로 남겨주세요..."
+               )
+               save_final_btn = gr.Button("감상 남기기")
+
+       def delayed_image_clear():
+           time.sleep(IMAGE_DISPLAY_TIME)
+           return gr.update(value=None)
+
+       def on_image_generated(prompt):
+           image_content = generate_image_from_prompt(prompt)
+           if image_content:
+               # Start a timer to clear the image
+               timer = threading.Timer(IMAGE_DISPLAY_TIME, lambda: result_image.update(value=None))
+               timer.start()
+               return image_content
+           return None
+
+       # 이벤트 연결
+       start_btn.click(
+           fn=lambda name: (
+               WORLDVIEW_MESSAGE if name.strip() else "이름을 입력해주세요",
+               gr.update(visible=True) if name.strip() else gr.update(),
+               {"user_name": name} if name.strip() else state
+           ),
+           inputs=[name_input],
+           outputs=[worldview_display, tabs, state]
+       )
+
+       set_baseline_btn.click(
+           fn=lambda x, s: ({"baseline_features": calculate_baseline_features(x)}, "기준점이 설정되었습니다."),
+           inputs=[baseline_audio, state],
+           outputs=[state, baseline_status]
+       )
+
+       save_btn.click(
+           fn=save_reflection,
+           inputs=[reflection_input, state],
+           outputs=[state, reflections_display]
+       )
+
+       clear_btn.click(
+           fn=lambda: None,
+           outputs=[voice_input]
+       )
+
+       analyze_btn.click(
+           fn=analyze_voice,
+           inputs=[voice_input, state],
+           outputs=[state, transcribed_text, voice_emotion, text_emotion, final_prompt]
+       )
+
+       generate_btn.click(
+           fn=on_image_generated,
+           inputs=[final_prompt],
+           outputs=[result_image]
+       )
+
+       save_final_btn.click(
+           fn=lambda t, s: (db.save_wish(s["user_name"], t), "감상이 저장되었습니다."),
+           inputs=[final_reflection, state],
+           outputs=[gr.Markdown("")]
+       )
+
+       return app
 
 if __name__ == "__main__":
-    demo = create_interface()
-    demo.launch(debug=True)
+   demo = create_interface()
+   demo.launch(debug=True)
+