Update app.py

app.py

@@ -359,16 +359,18 @@ def create_interface():
         
         gr.Markdown("# 디지털 굿판")
 
-        with gr.Tabs(selected=0) as tabs:  # selected 속성 추가
-            with gr.TabItem("입장") as tab_entrance:  # Tab -> TabItem
-                gr.Markdown(WELCOME_MESSAGE)
+        with gr.Tabs(selected=0) as tabs:
+            with gr.TabItem("입장") as tab_entrance:
+                name_display = gr.Markdown(WELCOME_MESSAGE)  # worldview_display 대신 하나의 Markdown 사용
                 name_input = gr.Textbox(
                     label="이름을 알려주세요",
                     placeholder="이름을 입력해주세요",
-                    interactive=True  # 명시적으로 지정
+                    interactive=True
                 )
-                worldview_display = gr.Markdown(visible=False)
-                start_btn = gr.Button("여정 시작하기", variant="primary")  # variant 추가
+                start_btn = gr.Button("여정 시작하기", variant="primary")  
+                
+                
+# variant 추가
 
             with gr.TabItem("기준 설정") as tab_baseline:
                 gr.Markdown("""### 축원의 문장을 평온한 마음으로 읽어주세요""")
@@ -460,53 +462,68 @@ def create_interface():
                 save_final_status = gr.Markdown("")
 
         # 이벤트 연결 - gr.on 사용
+        def handle_start(name, current_state):
+            if not name.strip():
+                return "이름을 입력해주세요", gr.update(selected=0), current_state
+            current_state = {**current_state, "user_name": name}
+            return WORLDVIEW_MESSAGE, gr.update(selected=1), current_state
+
+        def handle_baseline(audio_data, current_state):
+            try:
+                features = calculate_baseline_features(audio_data)
+                if features:
+                    current_state = {**current_state, "baseline_features": features}
+                    return current_state, "기준점이 설정되었습니다."
+                return current_state, "기준점 설정에 실패했습니다."
+            except Exception as e:
+                return current_state, f"오류 발생: {str(e)}"
+
+        def handle_analysis(audio_data, current_state):
+            try:
+                result = analyze_voice(audio_data, current_state)
+                if isinstance(result, tuple) and len(result) == 5:
+                    return result
+                return current_state, "분석 실패", "", "", ""
+            except Exception as e:
+                return current_state, f"오류 발생: {str(e)}", "", "", ""
+
+        def handle_image_generation(prompt):
+            if not prompt:
+                return None
+            try:
+                image_data = generate_image_from_prompt(prompt)
+                return image_data[0] if image_data else None
+            except Exception as e:
+                print(f"Image generation error: {str(e)}")
+                return None
+
+        # 이벤트 연결
         start_btn.click(
-            fn=lambda name, state: (
-                WORLDVIEW_MESSAGE if name.strip() else "이름을 입력해주세요",
-                gr.update(visible=True) if name.strip() else gr.update(visible=False),
-                {**state, "user_name": name} if name.strip() else state
-            ),
+            fn=handle_start,
             inputs=[name_input, state],
-            outputs=[worldview_display, tabs, state]
+            outputs=[name_display, tabs, state]
         )
 
         set_baseline_btn.click(
-            fn=lambda x, s: (
-                {**s, "baseline_features": calculate_baseline_features(x)},
-                "기준점이 설정되었습니다."
-            ),
+            fn=handle_baseline,
             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: gr.update(value=None),
-            outputs=[voice_input]
-        )
-
         analyze_btn.click(
-            fn=analyze_voice,
+            fn=handle_analysis,
             inputs=[voice_input, state],
             outputs=[state, transcribed_text, voice_emotion, text_emotion, final_prompt]
         )
 
         generate_btn.click(
-            fn=lambda prompt: generate_image_from_prompt(prompt)[0],
+            fn=handle_image_generation,
             inputs=[final_prompt],
             outputs=[result_image]
         )
 
         save_final_btn.click(
-            fn=lambda t, s: (
-                db.save_wish(s.get("user_name", "익명"), t),
-                "감상이 저장되었습니다."
-            ),
+            fn=lambda t, s: "감상이 저장되었습니다." if db.save_wish(s.get("user_name", "익명"), t) else "저장 실패",
             inputs=[final_reflection, state],
             outputs=[save_final_status]
         )
@@ -522,8 +539,6 @@ if __name__ == "__main__":
     demo = create_interface()
     demo.launch(
         debug=True,
-        share=True,
-        show_error=True,
         server_name="0.0.0.0",
         server_port=7860
-    )
+    )