Update app.py

app.py

@@ -701,12 +701,12 @@ def create_interface():
                             f"- 감정: {baseline_emotion['primary']}\n"
                             f"- 에너지 레벨: {baseline_emotion['details']['energy_level']}\n"
                             f"- 말하기 속도: {baseline_emotion['details']['speech_rate']}\n"
-                            f"- 음높이: {baseline_emotion['details']['pitch_variation']}\n"
+                            f"- 음높이 변화: {baseline_emotion['details']['pitch_variation']}\n"
                             f"이 음성을 기준으로 이후 소원을 비교 분석합니다."
                         )
                         
                         # 상태 업데이트
-                        new_state = {**state, "baseline_features": features, "analysis_complete": True}
+                        new_state = safe_state_update(state, {"baseline_features": features, "analysis_complete": True})
                         
                         return (
                             new_state,                          # state
@@ -724,7 +724,8 @@ def create_interface():
                             gr.update(visible=True),
                             gr.update(visible=False)
                         )
-                    
+                
+                        
                 # 4단계: 굿판 입장 안내
                 entry_guide_section = gr.Column(visible=False)
                 with entry_guide_section:
@@ -774,7 +775,6 @@ def create_interface():
                             row_count=(5, "dynamic")  # 동적으로 행 수 조정
                         )
 
-
             # 기원 탭
             with gr.TabItem("기원") as tab_wish:
                 gr.Markdown("## 기원 - 소원을 전해보세요")
@@ -964,10 +964,9 @@ def create_interface():
                 print(f"Error saving reflection: {e}")
                 return state, []
 
-
         def handle_save_wish(text, state):
             if not text.strip():
-                return "소원을 입력해주세요.", []
+                return state, "소원을 입력해주세요.", []
             
             try:
                 name = state.get("user_name", "익명")
@@ -977,10 +976,11 @@ def create_interface():
                     [wish["timestamp"], wish["wish"], wish["name"]]
                     for wish in wishes
                 ]
-                return "소원이 저장되었습니다.", wish_display_data
+                return state, "소원이 저장되었습니다.", wish_display_data
             except Exception as e:
                 print(f"Error saving wish: {e}")
-                return "오류가 발생했습니다.", []
+                return state, "오류가 발생했습니다.", []
+
 
         # 이벤트 연결
         name_submit_btn.click(
@@ -993,7 +993,8 @@ def create_interface():
             fn=handle_continue,
             outputs=[story_section, welcome_section, blessing_section, entry_guide_section]
         )
-
+        
+        # 이벤트 연결부터 수정
         set_baseline_btn.click(
             fn=handle_blessing_complete,
             inputs=[baseline_audio, state],
@@ -1042,8 +1043,9 @@ def create_interface():
         save_final_btn.click(
             fn=handle_save_wish,
             inputs=[final_reflection, state],
-            outputs=[gr.Markdown(""), wishes_display]  # baseline_status 대신 빈 Markdown 사용
+            outputs=[state, baseline_status, wishes_display]
         )
+
     return app
 
 if __name__ == "__main__":