Update app.py

app.py

@@ -53,12 +53,18 @@ def serve_wishes(path):
 
 
 class SimpleDB:
-    def __init__(self, reflections_path="data/reflections.json", wishes_path="data/wishes.json"):
-        self.reflections_path = reflections_path
+    """데이터 저장 및 관리 클래스"""
+    
+    def __init__(self, wishes_path="data/wishes.json"):
+        """
+        영구 저장이 필요한 소원 데이터만 파일로 관리
+        Args:
+            wishes_path: 소원 데이터 저장 경로
+        """
         self.wishes_path = wishes_path
-        os.makedirs('data', exist_ok=True)
-        self.reflections = self._load_json(reflections_path)
+        os.makedirs(os.path.dirname(wishes_path), exist_ok=True)
         self.wishes = self._load_json(wishes_path)
+        self.session_reflections = []  # 임시 메모리에만 저장되는 감상
 
     def _load_json(self, file_path):
         if not os.path.exists(file_path):
@@ -71,29 +77,28 @@ class SimpleDB:
             print(f"Error loading {file_path}: {e}")
             return []
 
-    def save_reflection(self, name, reflection, sentiment, timestamp=None):
-        if timestamp is None:
-            timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-        
+    def save_reflection(self, name, reflection, sentiment):
+        """
+        감상은 세션 메모리에만 저장
+        """
         reflection_data = {
-            "timestamp": timestamp,
+            "timestamp": datetime.now().strftime("%H:%M:%S"),
             "name": name,
             "reflection": reflection,
             "sentiment": sentiment
         }
-        
-        self.reflections.append(reflection_data)
-        self._save_json(self.reflections_path, self.reflections)
+        self.session_reflections.append(reflection_data)
         return True
 
-    def save_wish(self, name, wish, emotion_data=None, timestamp=None):
-        if timestamp is None:
-            timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    def save_wish(self, name, wish, emotion_data=None):
+        """
+        소원은 영구 저장
+        """
         wish_data = {
+            "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
             "name": name,
             "wish": wish,
-            "emotion": emotion_data,
-            "timestamp": timestamp
+            "emotion": emotion_data
         }
         self.wishes.append(wish_data)
         self._save_json(self.wishes_path, self.wishes)
@@ -108,11 +113,17 @@ class SimpleDB:
             print(f"Error saving to {file_path}: {e}")
             return False
 
-    def get_all_reflections(self):
-        return sorted(self.reflections, key=lambda x: x["timestamp"], reverse=True)
+    def get_session_reflections(self):
+        """현재 세션의 감상만 반환"""
+        return self.session_reflections
 
     def get_all_wishes(self):
-        return self.wishes
+        """저장된 모든 소원 반환"""
+        return sorted(self.wishes, key=lambda x: x["timestamp"], reverse=True)
+
+    def clear_session_data(self):
+        """세션 데이터 초기화"""
+        self.session_reflections = []
 
 # API 설정
 HF_API_TOKEN = os.getenv("roots", "")
@@ -646,59 +657,142 @@ def create_interface():
                 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("""
-                온천천의 소리를 들으며 마음을 정화해보세요.
-                
-                💫 이 앱은 온천천의 사운드스케이프를 녹음하여 제작되었으며,
-                온천천 온천장역에서 장전역까지 걸으며 더 깊은 체험이 가능합니다.
-                """)
-                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
+                    with gr.Column() as recording_section:
+                        baseline_audio = gr.Audio(
+                            label="축원 문장 녹음하기",
+                            sources=["microphone"],
+                            type="numpy",
+                            streaming=False
                         )
-                        save_btn = gr.Button("감상 저장하기", variant="secondary")
-                        reflections_display = gr.Dataframe(
-                            headers=["시간", "감상", "감정 분석"],
-                            label="기록된 감상들",
-                            value=[],  # 초기값은 빈 리스트
-                            interactive=False,
-                            wrap=True,
-                            row_count=(5, "dynamic")  # 동적으로 행 수 조정
+                        analysis_status = gr.Markdown(
+                            "축원 문장을 녹음해주세요.",
+                            visible=True
                         )
+                    
+                    with gr.Column(visible=False) as analysis_results:
+                        voice_analysis = gr.Markdown("")
+                        set_baseline_btn = gr.Button(
+                            "축원 마치기", 
+                            variant="primary",
+                            visible=False
+                        )
+                
+                def handle_baseline_recording(audio, state):
+                    """기준 음성 분석"""
+                    if audio is None:
+                        return {
+                            "state": state,
+                            "status": "축원 문장을 녹음해주세요.",
+                            "analysis_visible": False,
+                            "button_visible": False,
+                            "analysis_text": ""
+                        }
+                    
+                    try:
+                        # 분석 중임을 표시
+                        yield {
+                            "state": state,
+                            "status": "음성을 분석하고 있습니다...",
+                            "analysis_visible": False,
+                            "button_visible": False,
+                            "analysis_text": ""
+                        }
+                        
+                        # 음성 분석
+                        features = calculate_baseline_features(audio)
+                        if features is None:
+                            return {
+                                "state": state,
+                                "status": "음성 분석에 실패했습니다. 다시 시도해주세요.",
+                                "analysis_visible": False,
+                                "button_visible": False,
+                                "analysis_text": ""
+                            }
+                        
+                        # 감정 분석
+                        baseline_emotion = map_acoustic_to_emotion(features)
+                        
+                        # 분석 결과 생성
+                        analysis_text = (
+                            f"기준 음성 분석 결과:\n"
+                            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"이 음성을 기준으로 이후 소원을 비교 분석합니다."
+                        )
+                        
+                        # 상태 업데이트
+                        new_state = {**state, "baseline_features": features, "analysis_complete": True}
+                        
+                        return {
+                            "state": new_state,
+                            "status": "",
+                            "analysis_visible": True,
+                            "button_visible": True,
+                            "analysis_text": analysis_text
+                        }
+                        
+                    except Exception as e:
+                        return {
+                            "state": state,
+                            "status": f"오류가 발생했습니다: {str(e)}",
+                            "analysis_visible": False,
+                            "button_visible": False,
+                            "analysis_text": ""
+                        }
+                
+                def handle_blessing_complete(state):
+                    """축원 완료 처리"""
+                    if not state.get("analysis_complete"):
+                        return state, "음성 분석이 완료되지 않았습니다.", gr.update(visible=True), gr.update(visible=False)
+                    
+                    return state, "", gr.update(visible=False), gr.update(visible=True)
+                
+                                # 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.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: