Update app.py

app.py

@@ -20,7 +20,7 @@ WELCOME_MESSAGE = """
 이제, 평온함과 치유의 여정을 시작해보세요.
 """
 
-WORLDVIEW_MESSAGE = """
+ONCHEON_STORY = """
 ## 온천천 이야기 🌌
 
 온천천의 물줄기는 신성한 금샘에서 시작됩니다. 금샘은 생명과 창조의 원천이며, 
@@ -32,48 +32,67 @@ WORLDVIEW_MESSAGE = """
 """
 
 class SimpleDB:
-    def __init__(self, file_path="wishes.json"):
-        self.file_path = file_path
-        self.wishes = self._load_wishes()
-        
-        # wishes.json 파일이 없으면 생성
-        if not os.path.exists(self.file_path):
-            with open(self.file_path, 'w', encoding='utf-8') as f:
+    def __init__(self, reflections_path="data/reflections.json", wishes_path="data/wishes.json"):
+        self.reflections_path = reflections_path
+        self.wishes_path = wishes_path
+        os.makedirs('data', exist_ok=True)
+        self.reflections = self._load_json(reflections_path)
+        self.wishes = self._load_json(wishes_path)
+
+    def _load_json(self, file_path):
+        if not os.path.exists(file_path):
+            with open(file_path, 'w', encoding='utf-8') as f:
                 json.dump([], f, ensure_ascii=False, indent=2)
-
-    def _load_wishes(self):
         try:
-            if os.path.exists(self.file_path):
-                with open(self.file_path, 'r', encoding='utf-8') as f:
-                    return json.load(f)
-            return []
+            with open(file_path, 'r', encoding='utf-8') as f:
+                return json.load(f)
         except Exception as e:
-            print(f"Error loading wishes: {e}")
+            print(f"Error loading {file_path}: {e}")
             return []
 
-    def save_wish(self, name, wish, timestamp=None):
+    def save_reflection(self, name, reflection, sentiment, timestamp=None):
+        if timestamp is None:
+            timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        reflection_data = {
+            "name": name,
+            "reflection": reflection,
+            "sentiment": sentiment,
+            "timestamp": timestamp
+        }
+        self.reflections.append(reflection_data)
+        self._save_json(self.reflections_path, self.reflections)
+        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")
         wish_data = {
             "name": name,
             "wish": wish,
+            "emotion": emotion_data,
             "timestamp": timestamp
         }
         self.wishes.append(wish_data)
+        self._save_json(self.wishes_path, self.wishes)
+        return True
+
+    def _save_json(self, file_path, data):
         try:
-            with open(self.file_path, 'w', encoding='utf-8') as f:
-                json.dump(self.wishes, f, ensure_ascii=False, indent=2)
+            with open(file_path, 'w', encoding='utf-8') as f:
+                json.dump(data, f, ensure_ascii=False, indent=2)
             return True
         except Exception as e:
-            print(f"Error saving wish: {e}")
+            print(f"Error saving to {file_path}: {e}")
             return False
 
+    def get_all_reflections(self):
+        return self.reflections
+
     def get_all_wishes(self):
-        """모든 소원 반환"""
         return self.wishes
 
 # API 설정
-HF_API_TOKEN = os.getenv("roots", "")  # 기본값을 빈 문자열로 설정
+HF_API_TOKEN = os.getenv("roots", "")
 if not HF_API_TOKEN:
     print("Warning: HuggingFace API token not found. Some features may be limited.")
 
@@ -92,7 +111,6 @@ try:
     )
 except Exception as e:
     print(f"Error initializing AI models: {e}")
-    # 기본 파이프라인 설정
     speech_recognizer = None
     text_analyzer = None
 
@@ -100,11 +118,11 @@ except Exception as e:
 os.makedirs("generated_images", exist_ok=True)
 
 # 음성 분석 관련 함수들
+```python
 def calculate_baseline_features(audio_data):
     try:
         if isinstance(audio_data, tuple):
             sr, y = audio_data
-            # 데이터 타입을 float32로 변환
             y = y.astype(np.float32)
         elif isinstance(audio_data, str):
             y, sr = librosa.load(audio_data, sr=16000)
@@ -118,7 +136,6 @@ def calculate_baseline_features(audio_data):
 
         features = {
             "energy": float(np.mean(librosa.feature.rms(y=y))),
-            # tempo 함수 업데이트
             "tempo": float(librosa.feature.tempo(y=y, sr=sr)[0]),
             "pitch": float(np.mean(librosa.feature.zero_crossing_rate(y=y))),
             "volume": float(np.mean(np.abs(y))),
@@ -130,7 +147,6 @@ def calculate_baseline_features(audio_data):
         return None
 
 def map_acoustic_to_emotion(features, baseline_features=None):
-    """음향학적 특성을 감정으로 매핑"""
     if features is None:
         return {
             "primary": "알 수 없음",
@@ -151,9 +167,7 @@ def map_acoustic_to_emotion(features, baseline_features=None):
     pitch_norm = min(features["pitch"] * 2, 1)
 
     if baseline_features:
-        if baseline_features["energy"] == 0 or baseline_features["tempo"] == 0 or baseline_features["pitch"] == 0:
-            print("Invalid baseline features")
-        else:
+        if baseline_features["energy"] > 0 and baseline_features["tempo"] > 0 and baseline_features["pitch"] > 0:
             energy_norm = (features["energy"] / baseline_features["energy"]) * 50
             tempo_norm = (features["tempo"] / baseline_features["tempo"])
             pitch_norm = (features["pitch"] / baseline_features["pitch"])
@@ -215,21 +229,15 @@ def analyze_voice(audio_data, state):
 
     try:
         sr, y = audio_data
-        # 데이터 타입을 float32로 변환
         y = y.astype(np.float32)
         
         if len(y) == 0:
             return state, "음성이 감지되지 않았습니다.", "", "", ""
 
-        # 음향학적 특성 분석
-        acoustic_features = {
-            "energy": float(np.mean(librosa.feature.rms(y=y))),
-            "tempo": float(librosa.feature.tempo(y=y, sr=sr)[0]),
-            "pitch": float(np.mean(librosa.feature.zero_crossing_rate(y=y))),
-            "volume": float(np.mean(np.abs(y)))
-        }
+        acoustic_features = calculate_baseline_features((sr, y))
+        if acoustic_features is None:
+            return state, "음성 분석에 실패했습니다.", "", "", ""
 
-     
         # 음성 인식
         if speech_recognizer:
             try:
@@ -257,7 +265,6 @@ def analyze_voice(audio_data, state):
             text_sentiment = {"label": "unknown", "score": 0.0}
             text_result = "텍스트 감정 분석을 수행할 수 없습니다."
 
-        # 결과 포맷팅 
         voice_result = (
             f"음성 감정: {voice_emotion['primary']} "
             f"(강도: {voice_emotion['intensity']:.1f}%, 신뢰도: {voice_emotion['confidence']:.2f})\n"
@@ -279,8 +286,8 @@ def analyze_voice(audio_data, state):
         print(f"Error in analyze_voice: {str(e)}")
         return state, f"오류 발생: {str(e)}", "", "", ""
 
+
 def generate_detailed_prompt(text, emotions, text_sentiment):
-    """감정 기반 상세 프롬프트 생성"""
     emotion_colors = {
         "기쁨/열정": "밝은 노랑과 따뜻한 주황색",
         "분노/강조": "강렬한 빨강과 짙은 검정",
@@ -307,7 +314,6 @@ def generate_detailed_prompt(text, emotions, text_sentiment):
     return prompt
 
 def generate_image_from_prompt(prompt):
-    """이미지 생성 함수"""
     if not prompt:
         print("No prompt provided")
         return None
@@ -327,7 +333,6 @@ def generate_image_from_prompt(prompt):
         )
 
         if response.status_code == 200:
-            # 이미지를 임시 파일로 저장
             timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
             image_path = f"generated_images/image_{timestamp}.png"
             os.makedirs("generated_images", exist_ok=True)
@@ -335,7 +340,7 @@ def generate_image_from_prompt(prompt):
             with open(image_path, "wb") as f:
                 f.write(response.content)
             
-            return image_path  # 파일 경로 반환
+            return image_path
         else:
             print(f"Error: {response.status_code}")
             print(f"Response: {response.text}")
@@ -344,46 +349,7 @@ def generate_image_from_prompt(prompt):
         print(f"Error generating image: {str(e)}")
         return None
 
-def save_reflection_fixed(text, state):
-    if not text.strip():
-        return state, []
-    
-    try:
-        current_time = datetime.now().strftime("%H:%M:%S")
-        if text_analyzer:
-            sentiment = text_analyzer(text)[0]
-            sentiment_text = f"{sentiment['label']} ({sentiment['score']:.2f})"
-            
-            # 감정 분석 결과를 프롬프트에 반영
-            emotion_prompt = generate_detailed_prompt(
-                text, 
-                {"primary": sentiment['label'], "intensity": sentiment['score'] * 100, 
-                 "characteristics": ["텍스트 기반 감정"], "confidence": sentiment['score']},
-                sentiment
-            )
-            state = {**state, "final_prompt": emotion_prompt}
-        else:
-            sentiment_text = "분석 불가"
-
-        new_reflection = [current_time, text, sentiment_text]
-        reflections = state.get("reflections", [])
-        reflections.append(new_reflection)
-        state = {**state, "reflections": reflections}
-        return state, reflections
-    except Exception as e:
-        print(f"Error saving reflection: {e}")
-        return state, state.get("reflections", [])
-
-def is_mobile(user_agent):
-    return any(device in user_agent.lower() for device in ['mobile', 'android', 'iphone'])
-
-def handle_tab_change(tab_index, state):
-    """탭 변경 시 상태 유지"""
-    state = {**state, "current_tab": tab_index}
-    return state
-
 def safe_state_update(state, updates):
-    """안전한 상태 업데이트"""
     try:
         new_state = {**state, **updates}
         return new_state
@@ -401,55 +367,63 @@ def create_interface():
         "wish": None,
         "final_prompt": "",
         "image_path": None,
-        "current_tab": 0,
-        "is_mobile": False,  # 모바일 여부
-        "session_active": True  # 세션 상태
+        "current_tab": 0
     }
 
-    with gr.Blocks(
-        theme=gr.themes.Soft(),
-        css="""
-            @media (max-width: 600px) {
-                .container { padding: 10px; }
-                .gradio-row { flex-direction: column; }
-                .gradio-button { width: 100%; margin: 5px 0; }
-                .gradio-textbox { width: 100%; }
-            }
-        """
-    ) as app:
+    css = """
+        @media (max-width: 600px) {
+            .container { padding: 10px; }
+            .gradio-row { flex-direction: column; }
+            .gradio-button { width: 100%; margin: 5px 0; }
+            .gradio-textbox { width: 100%; }
+            #audio-recorder { width: 100%; }
+            #result-image { width: 100%; }
+        }
+    """
+
+    with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
         state = gr.State(value=initial_state)
         
         gr.Markdown("# 디지털 굿판")
         gr.Markdown("""
-1. 입장 → 2. 축원(기준 설정) → 3. 청신 → 4. 기원 → 5. 송신 
-순서대로 진행해주세요.
-""")
+        1. 입장 → 2. 청신 → 3. 기원 → 4. 송신 
+        순서대로 진행해주세요.
+        """)
 
-        with gr.Tabs() as tabs:
+        with gr.Tabs(selected=0) as tabs:
+            # 입장 탭 (축원 포함)
             with gr.TabItem("입장") as tab_entrance:
-                gr.Markdown(WELCOME_MESSAGE)
-                name_input = gr.Textbox(
-                    label="이름을 알려주세요",
-                    placeholder="이름을 입력해주세요",
-                    interactive=True
-                )
-                worldview_display = gr.Markdown(visible=False)
-                start_btn = gr.Button("여정 시작하기", variant="primary")
-
-            with gr.TabItem("축원") as tab_baseline:
-                gr.Markdown("### 축원의 문장을 평온한 마음으로 읽어주세요")
-                gr.Markdown("'당신의 건강과 행복이 늘 가득하기를'")
-                baseline_audio = gr.Audio(
-                    label="축원 문장 녹음하기",
-                    sources=["microphone"],
-                    type="numpy",
-                    streaming=False,
-                    min_width=300,     # 최소 너비 설정
-                    elem_id="audio-recorder"  # CSS 식별자 추가
-                )
-                set_baseline_btn = gr.Button("기준점 설정 완료", variant="primary")
-                baseline_status = gr.Markdown("")
+                # 첫 화면
+                welcome_section = gr.Column(visible=True)
+                with welcome_section:
+                    gr.Markdown(WELCOME_MESSAGE)
+                    name_input = gr.Textbox(
+                        label="이름을 알려주세요",
+                        placeholder="이름을 입력해주세요",
+                        interactive=True
+                    )
+                
+                # 온천천 이야기
+                story_section = gr.Column(visible=False)
+                with story_section:
+                    gr.Markdown(ONCHEON_STORY)
+                    start_btn = gr.Button("여정 시작하기", variant="primary")
+                
+                # 축원 섹션
+                blessing_section = gr.Column(visible=False)
+                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("")
 
+            # 청신 탭
             with gr.TabItem("청신") as tab_listen:
                 gr.Markdown("## 청신 - 소리로 정화하기")
                 gr.Markdown("""
@@ -483,6 +457,7 @@ def create_interface():
                             wrap=True
                         )
 
+            # 기원 탭
             with gr.TabItem("기원") as tab_wish:
                 gr.Markdown("## 기원 - 소원을 전해보세요")
                 with gr.Row():
@@ -511,8 +486,9 @@ def create_interface():
                             interactive=False
                         )
 
+            # 송신 탭
             with gr.TabItem("송신") as tab_send:
-                gr.Markdown("## 송신 -소지(소원지)를 그려 날려 태워봅시다")
+                gr.Markdown("## 송신 - 소지(소원지)를 그려 날려 태워봅시다")
                 final_prompt = gr.Textbox(
                     label="생성된 프롬프트",
                     interactive=False,
@@ -521,9 +497,7 @@ def create_interface():
                 generate_btn = gr.Button("마음의 그림 그리기", variant="primary")
                 result_image = gr.Image(
                     label="생성된 이미지",
-                    show_download_button=True,
-                    min_width=300,
-                    elem_id="result-image"
+                    show_download_button=True
                 )
             
                 gr.Markdown("## 온천천에 전하고 싶은 소원을 남겨주세요")
@@ -538,69 +512,42 @@ def create_interface():
 따뜻한 마음을 담아 작성해주세요.
 """)
                 wishes_display = gr.Dataframe(
-                    headers=["시간", "소원", "이름"],  # 헤더 수정
+                    headers=["시간", "소원", "이름"],
                     label="기록된 소원들",
                     value=[],
                     interactive=False,
                     wrap=True
                 )
 
-        # 이벤트 연결
-        def handle_start(name, current_state):
+        # 이벤트 핸들러들
+        def handle_name_input(name):
             if not name.strip():
-                return (
-                    "이름을 입력해주세요", 
-                    gr.update(visible=False), 
-                    current_state,
-                    gr.update(selected=0)  # 현재 탭 유지
-                )
-            current_state = {**current_state, "user_name": name}
-            return (
-                WORLDVIEW_MESSAGE,
-                gr.update(visible=True),
-                current_state,
-                gr.update(selected=1)  # 축원 탭(두 번째 탭)으로 이동
-            )
-            
+                return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)
+            return gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)
+
+        def handle_start():
+            return gr.update(visible=False), gr.update(visible=True)
+
         def handle_baseline(audio, current_state):
             if audio is None:
-                return current_state, "음성을 먼저 녹음해주세요.", gr.update(selected=1)  # 현재 탭 유지
+                return current_state, "음성을 먼저 녹음해주세요.", gr.update(selected=0)
             
             try:
                 sr, y = audio
                 y = y.astype(np.float32)
                 features = calculate_baseline_features((sr, y))
                 if features:
-                    current_state = {**current_state, "baseline_features": features}
-                    return (
-                        current_state, 
-                        "기준점이 설정되었습니다. 청신 탭으로 이동합니다.", 
-                        gr.update(selected=2)  # 청신 탭으로 이동
-                    )
-                return current_state, "기준점 설정에 실패했습니다. 다시 시도해주세요.", gr.update(selected=1)
+                    current_state = safe_state_update(current_state, {
+                        "baseline_features": features,
+                        "current_tab": 1
+                    })
+                    return current_state, "기준점이 설정되었습니다. 청신 탭으로 이동합니다.", gr.update(selected=1)
+                return current_state, "기준점 설정에 실패했습니다. 다시 시도해주세요.", gr.update(selected=0)
             except Exception as e:
                 print(f"Baseline error: {str(e)}")
-                return current_state, "오류가 발생했습니다. 다시 시도해주세요.", gr.update(selected=1)
+                return current_state, "오류가 발생했습니다. 다시 시도해주세요.", gr.update(selected=0)
 
-        # 음악 재생 이벤트 핸들러 수정
-        def play_music():
-            try:
-                return "assets/main_music.mp3"
-            except Exception as e:
-                print(f"Error playing music: {e}")
-                return None
-
-        def handle_analysis(audio, current_state):
-            state, text, voice_result, text_result, prompt = analyze_voice(audio, current_state)
-            return state, text, voice_result, text_result, prompt
-
-        def handle_image_generation(prompt):  # 들여쓰기 수정
-            image_path = generate_image_from_prompt(prompt)
-            if image_path and os.path.exists(image_path):
-                return image_path
-            return None
-
-        def save_reflection_fixed(text, state):  # 들여쓰기 수정
+        def handle_save_reflection(text, state):
             if not text.strip():
                 return state, []
             
@@ -609,82 +556,85 @@ def create_interface():
                 if text_analyzer:
                     sentiment = text_analyzer(text)[0]
                     sentiment_text = f"{sentiment['label']} ({sentiment['score']:.2f})"
+                    db.save_reflection(state.get("user_name", "익명"), text, sentiment)
                 else:
                     sentiment_text = "분석 불가"
+                    db.save_reflection(state.get("user_name", "익명"), text, {"label": "unknown", "score": 0.0})
 
                 new_reflection = [current_time, text, sentiment_text]
                 reflections = state.get("reflections", [])
                 reflections.append(new_reflection)
-                state = {**state, "reflections": reflections}
-                return state, reflections
+                state = safe_state_update(state, {"reflections": reflections})
+                
+                return state, db.get_all_reflections()
             except Exception as e:
                 print(f"Error saving reflection: {e}")
                 return state, state.get("reflections", [])
 
-        def save_wish(text, state):  # 들여쓰기 수정
+        def handle_save_wish(text, state):
             if not text.strip():
                 return "소원을 입력해주세요.", []
             
             try:
-                current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")  # 날짜 포함
                 name = state.get("user_name", "익명")
-                
-                # DB에 저장
                 db.save_wish(name, text)
-                
-                # 모든 소원 불러오기
-                all_wishes = db.wishes  # SimpleDB에서 모든 소원 가져오기
-                
-                # 표시할 데이터 형식으로 변환
+                wishes = db.get_all_wishes()
                 wish_display_data = [
                     [wish["timestamp"], wish["wish"], wish["name"]]
-                    for wish in all_wishes
+                    for wish in wishes
                 ]
-                
                 return "소원이 저장되었습니다.", wish_display_data
-        
             except Exception as e:
                 print(f"Error saving wish: {e}")
                 return "오류가 발생했습니다.", []
 
-        # 버튼 이벤트 연결
+        # 이벤트 연결
+        name_input.submit(
+            fn=handle_name_input,
+            outputs=[welcome_section, story_section, blessing_section]
+        )
+
         start_btn.click(
             fn=handle_start,
-            inputs=[name_input, state],
-            outputs=[worldview_display, worldview_display, state, tabs]
+            outputs=[story_section, blessing_section]
         )
 
         set_baseline_btn.click(
             fn=handle_baseline,
             inputs=[baseline_audio, state],
-            outputs=[state, baseline_status, tabs]  # tabs 추가
+            outputs=[state, baseline_status, tabs]
         )
 
         play_music_btn.click(
-            fn=play_music,
+            fn=lambda: "assets/main_music.mp3",
             outputs=[audio]
         )
 
+        save_btn.click(
+            fn=handle_save_reflection,
+            inputs=[reflection_input, state],
+            outputs=[state, reflections_display]
+        )
+
+        clear_btn.click(
+            fn=lambda: None,
+            outputs=[voice_input]
+        )
+
         analyze_btn.click(
-            fn=handle_analysis,
+            fn=analyze_voice,
             inputs=[voice_input, state],
             outputs=[state, transcribed_text, voice_emotion, text_emotion, final_prompt]
         )
 
         generate_btn.click(
-            fn=handle_image_generation,
+            fn=generate_image_from_prompt,
             inputs=[final_prompt],
             outputs=[result_image]
         )
 
-        save_btn.click(
-            fn=save_reflection_fixed,
-            inputs=[reflection_input, state],
-            outputs=[state, reflections_display]
-        )
-
         save_final_btn.click(
-            fn=save_wish,
+            fn=handle_save_wish,
             inputs=[final_reflection, state],
             outputs=[baseline_status, wishes_display]
         )
@@ -699,6 +649,6 @@ if __name__ == "__main__":
         server_name="0.0.0.0",
         server_port=7860,
         show_error=True,
-        height=None,          # 자동 높이 조정
-        width="100%"          # 전체 너비 사용
+        height=None,
+        width="100%"
     )