Update app.py

app.py

@@ -168,7 +168,6 @@ def calculate_baseline_features(audio_data):
         print(f"Error calculating baseline: {str(e)}")
         return None
 
-
 def map_acoustic_to_emotion(features, baseline_features=None):
     if features is None:
         return {
@@ -246,7 +245,6 @@ def map_acoustic_to_emotion(features, baseline_features=None):
 
     return emotions
 
-
 def analyze_voice(audio_data, state):
     if audio_data is None:
         return state, "음성을 먼저 녹음해주세요.", "", "", ""
@@ -344,7 +342,6 @@ def generate_detailed_prompt(text, emotions, text_sentiment):
 
     return prompt
 
-
 def generate_image_from_prompt(prompt):
     if not prompt:
         print("No prompt provided")
@@ -380,8 +377,7 @@ def generate_image_from_prompt(prompt):
     except Exception as e:
         print(f"Error generating image: {str(e)}")
         return None
-
-
+        
 def create_pwa_files():
     """PWA 필요 파일들 생성"""
     # manifest.json 생성
@@ -397,14 +393,54 @@ def create_pwa_files():
             "theme_color": "#000000",
             "orientation": "portrait",
             "icons": [
-                {"src": "/static/icons/icon-72x72.png", "sizes": "72x72", "type": "image/png", "purpose": "any maskable"},
-                {"src": "/static/icons/icon-96x96.png", "sizes": "96x96", "type": "image/png", "purpose": "any maskable"},
-                {"src": "/static/icons/icon-128x128.png", "sizes": "128x128", "type": "image/png", "purpose": "any maskable"},
-                {"src": "/static/icons/icon-144x144.png", "sizes": "144x144", "type": "image/png", "purpose": "any maskable"},
-                {"src": "/static/icons/icon-152x152.png", "sizes": "152x152", "type": "image/png", "purpose": "any maskable"},
-                {"src": "/static/icons/icon-192x192.png", "sizes": "192x192", "type": "image/png", "purpose": "any maskable"},
-                {"src": "/static/icons/icon-384x384.png", "sizes": "384x384", "type": "image/png", "purpose": "any maskable"},
-                {"src": "/static/icons/icon-512x512.png", "sizes": "512x512", "type": "image/png", "purpose": "any maskable"}
+                {
+                    "src": "/static/icons/icon-72x72.png",
+                    "sizes": "72x72",
+                    "type": "image/png",
+                    "purpose": "any maskable"
+                },
+                {
+                    "src": "/static/icons/icon-96x96.png",
+                    "sizes": "96x96",
+                    "type": "image/png",
+                    "purpose": "any maskable"
+                },
+                {
+                    "src": "/static/icons/icon-128x128.png",
+                    "sizes": "128x128",
+                    "type": "image/png",
+                    "purpose": "any maskable"
+                },
+                {
+                    "src": "/static/icons/icon-144x144.png",
+                    "sizes": "144x144",
+                    "type": "image/png",
+                    "purpose": "any maskable"
+                },
+                {
+                    "src": "/static/icons/icon-152x152.png",
+                    "sizes": "152x152",
+                    "type": "image/png",
+                    "purpose": "any maskable"
+                },
+                {
+                    "src": "/static/icons/icon-192x192.png",
+                    "sizes": "192x192",
+                    "type": "image/png",
+                    "purpose": "any maskable"
+                },
+                {
+                    "src": "/static/icons/icon-384x384.png",
+                    "sizes": "384x384",
+                    "type": "image/png",
+                    "purpose": "any maskable"
+                },
+                {
+                    "src": "/static/icons/icon-512x512.png",
+                    "sizes": "512x512",
+                    "type": "image/png",
+                    "purpose": "any maskable"
+                }
             ]
         }
         with open(manifest_path, 'w', encoding='utf-8') as f:
@@ -524,7 +560,6 @@ self.addEventListener('fetch', event => {
 </body>
 </html>''')
 
-
 def safe_state_update(state, updates):
     try:
         new_state = {**state, **updates}
@@ -539,7 +574,6 @@ def safe_state_update(state, updates):
         print(f"State update error: {e}")
         return state
 
-
 def create_interface():
     db = SimpleDB()
     
@@ -554,368 +588,333 @@ def create_interface():
     }
 
     css = """
-       @media (max-width: 600px) {
-                #logo-image {
-                    content: url('/static/DIGITAL_GUTPAN_LOGO_m.png') !important;
-                    width: 75px !important;
-                    height: auto !important;
-                }
-                .container { padding: 10px !important; }
-                .gradio-button { 
-                    width: 100% !important; 
-                    margin: 5px 0 !important;
-                    min-height: 44px !important;
-                }
-                .gradio-textbox, 
-                .gradio-audio, 
-                .gradio-image { 
-                    width: 100% !important;
-                }
-                .gradio-dataframe {
-                    overflow-x: auto !important;
-                    max-width: 100% !important;
-                }
-            }
-            
-            /* 데스크톱 뷰 */
-            @media (min-width: 601px) {
-                #logo-image {
-                    content: url('/static/DIGITAL_GUTPAN_LOGO_w.png');
-                    width: auto;
-                    height: auto;
-                }
-            }
-            
-            /* 공통 스타일 */
-            .gradio-button {
-                transition: all 0.3s ease;
+        @media (max-width: 600px) {
+            .container { padding: 10px !important; }
+            .gradio-row { 
+                flex-direction: column !important;
+                gap: 10px !important;
             }
-            .gradio-button:active {
-                transform: scale(0.98);
+            .gradio-button { 
+                width: 100% !important; 
+                margin: 5px 0 !important;
+                min-height: 44px !important;  /* 모바일 터치 영역 개선 */
             }
-            #logo-container {
-                text-align: center;
-                margin: 0 auto;
-                padding: 20px 0;
+            .gradio-textbox { width: 100% !important; }
+            .gradio-audio { width: 100% !important; }
+            .gradio-image { width: 100% !important; }
+            #audio-recorder { width: 100% !important; }
+            #result-image { width: 100% !important; }
+            .gradio-dataframe {
+                overflow-x: auto !important;
+                max-width: 100% !important;
             }
+        }
+    
+        /* 전반적인 UI 개선 */
+        .gradio-button {
+            transition: all 0.3s ease;
+        }
+        .gradio-button:active {
+            transform: scale(0.98);
+        }
     """
 
     with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
         state = gr.State(value=initial_state)
-          
-        with gr.Row(elem_id="logo-container"):
-            gr.Image(value="/static/DIGITAL_GUTPAN_LOGO_w.png", 
-                    show_label=False, 
-                    interactive=False,
-                    elem_id="logo-image")
+        
+        gr.Markdown("# 디지털 굿판")
+        gr.Markdown("""
+        1. 입장 → 2. 청신 → 3. 기원 → 4. 송신 
+        순서대로 진행해주세요.
+        """)
 
         with gr.Tabs(selected=0) as tabs:
             # 입장 탭 (축원 포함)
             with gr.TabItem("입장") as tab_entrance:
                 # 1단계: 첫 화면
-                with gr.Column(visible=True) as welcome_section:
-                    # Main image
-                    gr.Image(
-                        value="static/main-image.png",
-                        show_label=False,
-                        interactive=False,
-                        elem_id="main-image"
+                welcome_section = gr.Column(visible=True)
+                with welcome_section:
+                    gr.Markdown(WELCOME_MESSAGE)
+                    name_input = gr.Textbox(
+                        label="이름을 알려주세요",
+                        placeholder="이름을 입력해주세요",
+                        interactive=True
                     )
-    
-                # 기존 텍스트 및 다른 요소들
-                gr.Markdown(WELCOME_MESSAGE)
-                name_input = gr.Textbox(
-                    label="이름을 알려주세요",
-                    placeholder="이름을 입력해주세요",
-                    interactive=True
-                )
-                name_submit_btn = gr.Button("굿판 시작하기", variant="primary")
-
-                gr.Markdown(
-                    """
-                    <p style="color: #374151; text-align: center; font-size: 0.9em;">
-                    ‘디지털 굿판’은 부산문화재단의 지원으로 루츠리딤이 제작한 다원예술 프로젝트입니다.<br>
-                    본 프로젝트에서 기록된 정보 중<br>
-                    ‘송신’ 단계의 ‘소원전하기’를 제외한 모든 과정의 정보는 (목소리, 감상 등)<br>
-                    별도로 저장되지 않으며 AI 학습이나 외부 공유에 사용되지 않습니다.
-                    </p>
-                    """,
-                    visible=True
-                )
+                    name_submit_btn = gr.Button("굿판 시작하기", variant="primary")
             
-            # 2단계: 세계관 설명
-            story_section = gr.Column(visible=False)
-            with story_section:
-                gr.Markdown(ONCHEON_STORY)
-                continue_btn = gr.Button("준비하기", variant="primary")
-        
-            # 3단계: 축원 의식
-            blessing_section = gr.Column(visible=False)
-            with blessing_section:
-                gr.Markdown("### 축원의식을 시작하겠습니다")
-                gr.Markdown("'명짐 복짐 짊어지고 안가태평하시기를 비도발원 축원 드립니다'")
-                baseline_audio = gr.Audio(
-                    label="축원 문장 녹음하기",
-                    sources=["microphone"],
-                    type="numpy",
-                    streaming=False
-                )
-                set_baseline_btn = gr.Button("축원 마치기", variant="primary")
-                baseline_status = gr.Markdown("")
-
-            # 4단계: 굿판 입장 안내
-            entry_guide_section = gr.Column(visible=False)
-            with entry_guide_section:
-                gr.Markdown("## 굿판으로 입장하기")
-                gr.Markdown("""
-                * 청신 탭으로 이동해 주세요.
-                * 부산광역시 동래구 온천장역에서 시작하면 더욱 깊은 경험을 시작할 수 있습니다.
-                * (본격적인 경험을 시작하기에 앞서 이동을 권장드립니다)
-                """)
-                enter_btn = gr.Button("청신 의식 시작하기", variant="primary")
-
-        with gr.TabItem("청신") as tab_listen:
-            gr.Markdown("## 청신 - 소리로 정화하기")
-            gr.Markdown("""
-            온천천의 소리를 들으며 마음을 정화해보세요.
+                # 2단계: 세계관 설명
+                story_section = gr.Column(visible=False)
+                with story_section:
+                    gr.Markdown(ONCHEON_STORY)
+                    continue_btn = gr.Button("준비하기", variant="primary")
             
-            🇫 이 앱은 온천천의 사운드스케이프를 녹음하여 제작되었으며,
-            온천천 온천장역에서 장전역까지 걸으며 더 깊은 체험이 가능합니다.
-            """)
-            play_music_btn = gr.Button("온천천의 소리 듣기", variant="secondary")
-            with gr.Row():
-                audio = gr.Audio(
-                    value="assets/main_music.mp3",
-                    type="filepath",
-                    label="온천천의 소리",
-                    interactive=False,
-                    show_download_button=True,
-                    visible=True
-                )
-                with gr.Column():
-                    reflection_input = gr.Textbox(
-                        label="지금 이 순간의 감상을 자유롭게 적어보세요",
-                        lines=3,
-                        max_lines=5
-                    )
-                    save_btn = gr.Button("감상 저장하기", variant="secondary")
-                    reflections_display = gr.Dataframe(
-                        headers=["시간", "감상", "감정 분석"],
-                        label="기록된 감상들",
-                        value=[],  # 초기값은 빈 리스트
-                        interactive=False,
-                        wrap=True,
-                        row_count=(5, "dynamic")  # 동적으로 행 수 조정
-                    )
-
-        # 기원 탭
-        with gr.TabItem("기원") as tab_wish:
-            gr.Markdown("## 기원 - 소원을 전해보세요")
-            with gr.Row():
-                with gr.Column():
-                    voice_input = gr.Audio(
-                        label="소원을 나누고 싶은 마음을 말해주세요",
+                # 3단계: 축원 의식
+                blessing_section = gr.Column(visible=False)
+                with blessing_section:
+                    gr.Markdown("### 축원의식을 시작하겠습니다")
+                    gr.Markdown("'명짐 복짐 짊어지고 안가태평하시기를 비도발원 축원 드립니다'")
+                    baseline_audio = gr.Audio(
+                        label="축원 문장 녹음하기",
                         sources=["microphone"],
                         type="numpy",
                         streaming=False
                     )
-                    with gr.Row():
-                        clear_btn = gr.Button("녹음 지우기", variant="secondary")
-                        analyze_btn = gr.Button("소원 분석하기", variant="primary")
-
-                with gr.Column():
-                    transcribed_text = gr.Textbox(
-                        label="인식된 텍스트",
-                        interactive=False
-                    )
-                    voice_emotion = gr.Textbox(
-                        label="음성 감정 분석",
-                        interactive=False
-                    )
-                    text_emotion = gr.Textbox(
-                        label="텍스트 감정 분석",
-                        interactive=False
+                    set_baseline_btn = gr.Button("축원 마치기", variant="primary")
+                    baseline_status = gr.Markdown("")
+
+                # 4단계: 굿판 입장 안내
+                entry_guide_section = gr.Column(visible=False)
+                with entry_guide_section:
+                    gr.Markdown("## 굿판으로 입장하기")
+                    gr.Markdown("""
+                    * 청신 탭으로 이동해 주세요.
+                    * 부산광역시 동래구 온천장역에서 시작하면 더욱 깊은 경험을 시작할 수 있습니다.
+                    * (본격적인 경험을 시작하기에 앞서 이동을 권장드립니다)
+                    """)
+                    enter_btn = gr.Button("청신 의식 시작하기", variant="primary")
+
+            with gr.TabItem("청신") as tab_listen:
+                gr.Markdown("## 청신 - 소리로 정화하기")
+                gr.Markdown("""
+                온천천의 소리를 들으며 마음을 정화해보세요.
+                
+                💫 이 앱은 온천천의 사운드스케이프를 녹음하여 제작되었으며,
+                온천천 온천장역에서 장전역까지 걸으며 더 깊은 체험이 가능합니다.
+                """)
+                play_music_btn = gr.Button("온천천의 소리 듣기", variant="secondary")
+                with gr.Row():
+                    audio = gr.Audio(
+                        value="assets/main_music.mp3",
+                        type="filepath",
+                        label="온천천의 소리",
+                        interactive=False,
+                        show_download_button=True,
+                        visible=True
                     )
-
-        # 송신 탭
-        with gr.TabItem("송신") as tab_send:
-            gr.Markdown("## 송신 - 소지(소원지)를 그려 날려 태워봅시다")
-            final_prompt = gr.Textbox(
-                label="생성된 프롬프트",
-                interactive=False,
-                lines=3
-            )
-            generate_btn = gr.Button("마음의 그림 그리기", variant="primary")
-            result_image = gr.Image(
-                label="생성된 이미지",
-                show_download_button=True
-            )
-        
-            gr.Markdown("## 온천천에 전하고 싶은 소원을 남겨주세요")
-            final_reflection = gr.Textbox(
-                label="소원",
-                placeholder="당신의 소원을 한 줄로 남겨주세요...",
-                max_lines=3
-            )
-            save_final_btn = gr.Button("소원 전하기", variant="primary")
-            gr.Markdown("""
-🇫 여러분의 소원은 11월 25일 온천천 벽면에 설치될 소원나무에 전시될 예정입니다.
+                    with gr.Column():
+                        reflection_input = gr.Textbox(
+                            label="지금 이 순간의 감상을 자유롭게 적어보세요",
+                            lines=3,
+                            max_lines=5
+                        )
+                        save_btn = gr.Button("감상 저장하기", variant="secondary")
+                        reflections_display = gr.Dataframe(
+                            headers=["시간", "감상", "감정 분석"],
+                            label="기록된 감상들",
+                            value=[],  # 초기값은 빈 리스트
+                            interactive=False,
+                            wrap=True,
+                            row_count=(5, "dynamic")  # 동적으로 행 수 조정
+                        )
+
+            # 기원 탭
+            with gr.TabItem("기원") as tab_wish:
+                gr.Markdown("## 기원 - 소원을 전해보세요")
+                with gr.Row():
+                    with gr.Column():
+                        voice_input = gr.Audio(
+                            label="소원을 나누고 싶은 마음을 말해주세요",
+                            sources=["microphone"],
+                            type="numpy",
+                            streaming=False
+                        )
+                        with gr.Row():
+                            clear_btn = gr.Button("녹음 지우기", variant="secondary")
+                            analyze_btn = gr.Button("소원 분석하기", variant="primary")
+
+                    with gr.Column():
+                        transcribed_text = gr.Textbox(
+                            label="인식된 텍스트",
+                            interactive=False
+                        )
+                        voice_emotion = gr.Textbox(
+                            label="음성 감정 분석",
+                            interactive=False
+                        )
+                        text_emotion = gr.Textbox(
+                            label="텍스트 감정 분석",
+                            interactive=False
+                        )
+
+            # 송신 탭
+            with gr.TabItem("송신") as tab_send:
+                gr.Markdown("## 송신 - 소지(소원지)를 그려 날려 태워봅시다")
+                final_prompt = gr.Textbox(
+                    label="생성된 프롬프트",
+                    interactive=False,
+                    lines=3
+                )
+                generate_btn = gr.Button("마음의 그림 그리기", variant="primary")
+                result_image = gr.Image(
+                    label="생성된 이미지",
+                    show_download_button=True
+                )
+            
+                gr.Markdown("## 온천천에 전하고 싶은 소원을 남겨주세요")
+                final_reflection = gr.Textbox(
+                    label="소원",
+                    placeholder="당신의 소원을 한 줄로 남겨주세요...",
+                    max_lines=3
+                )
+                save_final_btn = gr.Button("소원 전하기", variant="primary")
+                gr.Markdown("""
+💫 여러분의 소원은 11월 25일 온천천 벽면에 설치될 소원나무에 전시될 예정입니다.
 따뜻한 마음을 담아 작성해주세요.
 """)
-            wishes_display = gr.Dataframe(
-                headers=["시간", "소원", "이름"],
-                label="기록된 소원들",
-                value=[],
-                interactive=False,
-                wrap=True
-            )
+                wishes_display = gr.Dataframe(
+                    headers=["시간", "소원", "이름"],
+                    label="기록된 소원들",
+                    value=[],
+                    interactive=False,
+                    wrap=True
+                )
 
-    # 이벤트 핸들러들
-    def handle_name_submit(name, state):
-        if not name.strip():
+        # 이벤트 핸들러들
+        def handle_name_submit(name, state):
+            if not name.strip():
+                return (
+                    gr.update(visible=True),
+                    gr.update(visible=False),
+                    gr.update(visible=False),
+                    gr.update(visible=False),
+                    state
+                )
+            state = safe_state_update(state, {"user_name": name})
             return (
-                gr.update(visible=True),
                 gr.update(visible=False),
+                gr.update(visible=True),
                 gr.update(visible=False),
                 gr.update(visible=False),
                 state
             )
-        state = safe_state_update(state, {"user_name": name})
-        return (
-            gr.update(visible=False),
-            gr.update(visible=True),
-            gr.update(visible=False),
-            gr.update(visible=False),
-            state
-        )
-    
-    def handle_continue():
-        return (
-            gr.update(visible=False),
-            gr.update(visible=False),
-            gr.update(visible=True),
-            gr.update(visible=False)
-        )
-    
-    def handle_blessing_complete(audio, state):
-        if audio is None:
-            return state, "음성을 먼저 녹음해주세요.", gr.update(visible=True), gr.update(visible=False)
         
-        try:
-            # ... (기존 음성 처리 코드)
+        def handle_continue():
             return (
-                state,
-                "축원이 완료되었습니다.",
                 gr.update(visible=False),
-                gr.update(visible=True)
+                gr.update(visible=False),
+                gr.update(visible=True),
+                gr.update(visible=False)
             )
-        except Exception as e:
-            return state, f"오류가 발생했습니다: {str(e)}", gr.update(visible=True), gr.update(visible=False)
-    
-    def handle_enter():
-        return gr.update(selected=1)  # 청신 탭으로 이동
-
-    def handle_start():
-        return gr.update(visible=False), gr.update(visible=True)
-
-    def handle_baseline(audio, current_state):
-        if audio is None:
-            return current_state, "음성을 먼저 녹음해주세요.", gr.update(selected=0)
-        
-        try:
-            sr, y = audio
-            y = y.astype(np.float32)
-            features = calculate_baseline_features((sr, y))
-            if features:
-                current_state = safe_state_update(current_state, {
-                    "baseline_features": features,
-                    "current_tab": 1
-                })
-                return current_state, "기준점이 설정되었습니다. 청신 탭으로 이동합니다.", gr.update(selected=1)
-            return current_state, "기준점 설정에 실패했습니다. 다시 시도해주세요.", gr.update(selected=0)
-        except Exception as e:
-            print(f"Baseline error: {str(e)}")
-            return current_state, "오류가 발생했습니다. 다시 시도해주세요.", gr.update(selected=0)
-
-    def handle_save_reflection(text, state):
-        if not text.strip():
-            return state, []
         
-        try:
-            current_time = datetime.now().strftime("%H:%M:%S")
-            if text_analyzer:
-                sentiment = text_analyzer(text)[0]
-                sentiment_text = f"{sentiment['label']} ({sentiment['score']:.2f})"
-                db.save_reflection(state.get("user_name", "익명"), text, sentiment)
-            else:
-                sentiment_text = "분석 불가"
-                db.save_reflection(state.get("user_name", "익명"), text, {"label": "unknown", "score": 0.0})
-
-            new_reflection = [current_time, text, sentiment_text]
-            reflections = state.get("reflections", [])
-            reflections.append(new_reflection)
-            state = safe_state_update(state, {"reflections": reflections})
+        def handle_blessing_complete(audio, state):
+            if audio is None:
+                return state, "음성을 먼저 녹음해주세요.", gr.update(visible=True), gr.update(visible=False)
             
-            return state, db.get_all_reflections()
-        except Exception as e:
-            print(f"Error saving reflection: {e}")
-            return state, state.get("reflections", [])
-
-    def save_reflection_fixed(text, state):
-        if not text.strip():
-            return state, []
+            try:
+                # ... (기존 음성 처리 코드)
+                return (
+                    state,
+                    "축원이 완료되었습니다.",
+                    gr.update(visible=False),
+                    gr.update(visible=True)
+                )
+            except Exception as e:
+                return state, f"오류가 발생했습니다: {str(e)}", gr.update(visible=True), gr.update(visible=False)
         
-        try:
-            current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-            name = state.get("user_name", "익명")
+        def handle_enter():
+            return gr.update(selected=1)  # 청신 탭으로 이동
+
+        def handle_start():
+            return gr.update(visible=False), gr.update(visible=True)
+
+        def handle_baseline(audio, current_state):
+            if audio is None:
+                return current_state, "음성을 먼저 녹음해주세요.", gr.update(selected=0)
             
-            if text_analyzer:
-                sentiment = text_analyzer(text)[0]
-                sentiment_text = f"{sentiment['label']} ({sentiment['score']:.2f})"
-            else:
-                sentiment_text = "분석 불가"
-    
-            # DB에 저장
-            db.save_reflection(name, text, sentiment_text)
+            try:
+                sr, y = audio
+                y = y.astype(np.float32)
+                features = calculate_baseline_features((sr, y))
+                if features:
+                    current_state = safe_state_update(current_state, {
+                        "baseline_features": features,
+                        "current_tab": 1
+                    })
+                    return current_state, "기준점이 설정되었습니다. 청신 탭으로 이동합니다.", gr.update(selected=1)
+                return current_state, "기준점 설정에 실패했습니다. 다시 시도해주세요.", gr.update(selected=0)
+            except Exception as e:
+                print(f"Baseline error: {str(e)}")
+                return current_state, "오류가 발생했습니다. 다시 시도해주세요.", gr.update(selected=0)
+
+        def handle_save_reflection(text, state):
+            if not text.strip():
+                return state, []
             
-            # 화면에 표시할 데이터 형식으로 변환
-            display_data = []
-            all_reflections = db.get_all_reflections()
-            for ref in all_reflections:
-                display_data.append([
-                    ref["timestamp"],
-                    ref["reflection"],
-                    ref["sentiment"]
-                ])
-    
-            # 상태 업데이트
-            state = safe_state_update(state, {"reflections": display_data})
+            try:
+                current_time = datetime.now().strftime("%H:%M:%S")
+                if text_analyzer:
+                    sentiment = text_analyzer(text)[0]
+                    sentiment_text = f"{sentiment['label']} ({sentiment['score']:.2f})"
+                    db.save_reflection(state.get("user_name", "익명"), text, sentiment)
+                else:
+                    sentiment_text = "분석 불가"
+                    db.save_reflection(state.get("user_name", "익명"), text, {"label": "unknown", "score": 0.0})
+
+                new_reflection = [current_time, text, sentiment_text]
+                reflections = state.get("reflections", [])
+                reflections.append(new_reflection)
+                state = safe_state_update(state, {"reflections": reflections})
+                
+                return state, db.get_all_reflections()
+            except Exception as e:
+                print(f"Error saving reflection: {e}")
+                return state, state.get("reflections", [])
+
+        def save_reflection_fixed(text, state):
+            if not text.strip():
+                return state, []
             
-            return state, display_data
-    
-        except Exception as e:
-            print(f"Error saving reflection: {e}")
-            return state, []
+            try:
+                current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+                name = state.get("user_name", "익명")
+                
+                if text_analyzer:
+                    sentiment = text_analyzer(text)[0]
+                    sentiment_text = f"{sentiment['label']} ({sentiment['score']:.2f})"
+                else:
+                    sentiment_text = "분석 불가"
+        
+                # DB에 저장
+                db.save_reflection(name, text, sentiment_text)
+                
+                # 화면에 표시할 데이터 형식으로 변환
+                display_data = []
+                all_reflections = db.get_all_reflections()
+                for ref in all_reflections:
+                    display_data.append([
+                        ref["timestamp"],
+                        ref["reflection"],
+                        ref["sentiment"]
+                    ])
+        
+                # 상태 업데이트
+                state = safe_state_update(state, {"reflections": display_data})
+                
+                return state, display_data
+        
+            except Exception as e:
+                print(f"Error saving reflection: {e}")
+                return state, []
 
 
-    def handle_save_wish(text, state):
-        if not text.strip():
-            return "소원을 입력해주세요.", []
-        
-        try:
-            name = state.get("user_name", "익명")
-            db.save_wish(name, text)
-            wishes = db.get_all_wishes()
-            wish_display_data = [
-                [wish["timestamp"], wish["wish"], wish["name"]]
-                for wish in wishes
-            ]
-            return "소원이 저장되었습니다.", wish_display_data
-        except Exception as e:
-            print(f"Error saving wish: {e}")
-            return "오류가 발생했습니다.", []
+        def handle_save_wish(text, state):
+            if not text.strip():
+                return "소원을 입력해주세요.", []
+            
+            try:
+                name = state.get("user_name", "익명")
+                db.save_wish(name, text)
+                wishes = db.get_all_wishes()
+                wish_display_data = [
+                    [wish["timestamp"], wish["wish"], wish["name"]]
+                    for wish in wishes
+                ]
+                return "소원이 저장되었습니다.", wish_display_data
+            except Exception as e:
+                print(f"Error saving wish: {e}")
+                return "오류가 발생했습니다.", []
 
-    # 이벤트 연결
+        # 이벤트 연결
         name_submit_btn.click(
             fn=handle_name_submit,
             inputs=[name_input, state],
@@ -971,7 +970,6 @@ def create_interface():
             inputs=[final_reflection, state],
             outputs=[baseline_status, wishes_display]
         )
-
     return app
 
 if __name__ == "__main__":
@@ -992,4 +990,4 @@ if __name__ == "__main__":
         show_error=True,
         height=None,
         width="100%"
-    )
+    )