Update app.py

app.py

@@ -667,36 +667,39 @@ def create_interface():
                     
                     with gr.Column() as action_section:
                         set_baseline_btn = gr.Button("축원 마치기", variant="primary")
-                        analysis_status = gr.Markdown(visible=False)
-                        analysis_result = gr.Markdown(visible=False)
+                        analysis_status = gr.Markdown(visible=True)
+                        analysis_result = gr.Markdown(visible=True)
                 
                 def handle_blessing_complete(audio, state):
                     """축원 완료 및 분석 처리"""
                     if audio is None:
                         return (
-                            state,                          # state
-                            "음성을 먼저 녹음해주세요.",         # 상태 메시지
-                            gr.update(visible=False),       # 분석 결과 
-                            gr.update(visible=True),        # blessing_section visibility
-                            gr.update(visible=False)        # entry_guide_section visibility
+                            state,                          
+                            gr.update(value="음성을 먼저 녹음해주세요.", visible=True),   # 상태 메시지
+                            gr.update(visible=False),                                  # 분석 결과 
+                            gr.update(visible=True),                                  # blessing_section
+                            gr.update(visible=False)                                  # entry_guide_section
                         )
                     
                     try:
-                        # 분석 중 상태 먼저 표시
-                        gr.update(value="축원 음성을 분석중입니다...", visible=True)
-                
+                        # 분석 시작 상태 표시
+                        status = gr.update(value="🔄 축원 음성을 분석중입니다...", visible=True)
+                        yield state, status, gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)
+                        
                         features = calculate_baseline_features(audio)
                         if features is None:
                             return (
                                 state,
-                                "음성 분석에 실패했습니다.",
+                                gr.update(value="❌ 음성 분석에 실패했습니다.", visible=True),
                                 gr.update(visible=False),
                                 gr.update(visible=True),
                                 gr.update(visible=False)
                             )
                         
-                        # 감정 분석 시작을 표시
-                        gr.update(value="감정을 분석중입니다...", visible=True)
+                        # 감정 분석 상태 표시
+                        status = gr.update(value="🔄 감정을 분석중입니다...", visible=True)
+                        yield state, status, gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)
+                        
                         baseline_emotion = map_acoustic_to_emotion(features)
                         
                         # 분석 결과 생성
@@ -713,17 +716,17 @@ def create_interface():
                         new_state = safe_state_update(state, {"baseline_features": features, "analysis_complete": True})
                         
                         return (
-                            new_state,                          # state
-                            "축원이 완료되었습니다. 이제 굿판으로 입장하실 수 있습니다.", # 상태 메시지
-                            gr.update(value=analysis_text, visible=True),  # 분석 결과
-                            gr.update(visible=False),           # blessing_section visibility
-                            gr.update(visible=True)             # entry_guide_section visibility
+                            new_state,
+                            gr.update(value="✅ 축원이 완료되었습니다. 이제 굿판으로 입장하실 수 있습니다.", visible=True),
+                            gr.update(value=analysis_text, visible=True),
+                            gr.update(visible=False),
+                            gr.update(visible=True)
                         )
                         
                     except Exception as e:
                         return (
                             state,
-                            f"오류가 발생했습니다: {str(e)}",
+                            gr.update(value=f"❌ 오류가 발생했습니다: {str(e)}", visible=True),
                             gr.update(visible=False),
                             gr.update(visible=True),
                             gr.update(visible=False)
@@ -932,26 +935,25 @@ def create_interface():
                     sentiment_text = "분석 불가"
         
                 # DB에 저장
-                db.save_reflection(name, text, sentiment_text)
+                reflection_data = 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"]
-                    ])
-        
+                session_reflections = state.get("reflections", [])
+                session_reflections.append([
+                    current_time,
+                    text,
+                    sentiment_text
+                ])
+                
                 # 상태 업데이트
-                state = safe_state_update(state, {"reflections": display_data})
+                state = safe_state_update(state, {"reflections": session_reflections})
                 
-                return state, display_data
-        
+                return state, session_reflections  # 세션 데이터만 반환
+            
             except Exception as e:
                 print(f"Error saving reflection: {e}")
-                return state, []
+                return state, state.get("reflections", [])
 
         def handle_save_wish(text, state):
             if not text.strip():