Update app.py

app.py

@@ -391,9 +391,6 @@ def safe_state_update(state, updates):
         print(f"State update error: {e}")
         return state
 
-# 상태 업데이트 시 사용
-state = safe_state_update(state, {"user_name": name})
-
 def create_interface():
     db = SimpleDB()
     
@@ -446,7 +443,7 @@ def create_interface():
                     label="축원 문장 녹음하기",
                     sources=["microphone"],
                     type="numpy",
-                    streaming=False
+                    streaming=False,
                     min_width=300,     # 최소 너비 설정
                     elem_id="audio-recorder"  # CSS 식별자 추가
                 )
@@ -524,7 +521,7 @@ def create_interface():
                 generate_btn = gr.Button("마음의 그림 그리기", variant="primary")
                 result_image = gr.Image(
                     label="생성된 이미지",
-                    show_download_button=True
+                    show_download_button=True,
                     min_width=300,
                     elem_id="result-image"
                 )
@@ -564,21 +561,26 @@ def create_interface():
                 current_state,
                 gr.update(selected=1)  # 축원 탭(두 번째 탭)으로 이동
             )
+            
         def handle_baseline(audio, current_state):
             if audio is None:
-                return current_state, "음성을 먼저 녹음해주세요."
+                return current_state, "음성을 먼저 녹음해주세요.", gr.update(selected=1)  # 현재 탭 유지
             
             try:
                 sr, y = audio
-                y = y.astype(np.float32)  # float32로 변환
+                y = y.astype(np.float32)
                 features = calculate_baseline_features((sr, y))
                 if features:
                     current_state = {**current_state, "baseline_features": features}
-                    return current_state, "기준점이 설정되었습니다. 다음 단계로 진행해주세요."
-                return current_state, "기준점 설정에 실패했습니다. 다시 시도해주세요."
+                    return (
+                        current_state, 
+                        "기준점이 설정되었습니다. 청신 탭으로 이동합니다.", 
+                        gr.update(selected=2)  # 청신 탭으로 이동
+                    )
+                return current_state, "기준점 설정에 실패했습니다. 다시 시도해주세요.", gr.update(selected=1)
             except Exception as e:
                 print(f"Baseline error: {str(e)}")
-                return current_state, "오류가 발생했습니다. 다시 시도해주세요."
+                return current_state, "오류가 발생했습니다. 다시 시도해주세요.", gr.update(selected=1)
 
         # 음악 재생 이벤트 핸들러 수정
         def play_music():
@@ -655,7 +657,7 @@ def create_interface():
         set_baseline_btn.click(
             fn=handle_baseline,
             inputs=[baseline_audio, state],
-            outputs=[state, baseline_status]
+            outputs=[state, baseline_status, tabs]  # tabs 추가
         )
 
         play_music_btn.click(