Update app.py

app.py

@@ -68,6 +68,10 @@ class SimpleDB:
             print(f"Error saving wish: {e}")
             return False
 
+    def get_all_wishes(self):
+        """모든 소원 반환"""
+        return self.wishes
+
 # API 설정
 HF_API_TOKEN = os.getenv("roots", "")  # 기본값을 빈 문자열로 설정
 if not HF_API_TOKEN:
@@ -495,7 +499,7 @@ def create_interface():
 따뜻한 마음을 담아 작성해주세요.
 """)
                 wishes_display = gr.Dataframe(
-                    headers=["시간", "소원", "감정 분석"],
+                    headers=["시간", "소원", "이름"],  # 헤더 수정
                     label="기록된 소원들",
                     value=[],
                     interactive=False,
@@ -578,25 +582,26 @@ def handle_image_generation(prompt):
                 return "소원을 입력해주세요.", []
             
             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})"
-                else:
-                    sentiment_text = "분석 불가"
-
-                new_wish = [current_time, text, sentiment_text]
-                wishes = state.get("wishes", [])
-                wishes.append(new_wish)
-                state = {**state, "wishes": wishes}
-
+                current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")  # 날짜 포함
+                name = state.get("user_name", "익명")
+                
                 # DB에 저장
-                db.save_wish(state.get("user_name", "익명"), text)
-
-                return "소원이 저장되었습니다.", wishes
+                db.save_wish(name, text)
+                
+                # 모든 소원 불러오기
+                all_wishes = db.wishes  # SimpleDB에서 모든 소원 가져오기
+                
+                # 표시할 데이터 형식으로 변환
+                wish_display_data = [
+                    [wish["timestamp"], wish["wish"], wish["name"]]
+                    for wish in all_wishes
+                ]
+                
+                return "소원이 저장되었습니다.", wish_display_data
+        
             except Exception as e:
                 print(f"Error saving wish: {e}")
-                return "오류가 발생했습니다.", state.get("wishes", [])
+                return "오류가 발생했습니다.", []
 
         # 버튼 이벤트 연결
         start_btn.click(