Update app.py

app.py

@@ -56,18 +56,17 @@ import sqlite3
 from datetime import datetime
 
 def init_db():
+    print("Initializing database...")
+    conn = None
     try:
         conn = sqlite3.connect('chat_history.db')
         c = conn.cursor()
         
-        # 기존 테이블 삭제 (선택적)
-        c.execute("DROP TABLE IF EXISTS chat_history")
-        c.execute("DROP TABLE IF EXISTS sessions")
-        
-        # 테이블 새로 생성
+        # 테이블 생성
         c.execute('''CREATE TABLE IF NOT EXISTS sessions
                      (session_id TEXT PRIMARY KEY, 
                       created_at TIMESTAMP)''')
+                      
         c.execute('''CREATE TABLE IF NOT EXISTS chat_history
                      (id INTEGER PRIMARY KEY AUTOINCREMENT,
                       session_id TEXT,
@@ -75,15 +74,36 @@ def init_db():
                       response TEXT,
                       timestamp TIMESTAMP,
                       FOREIGN KEY (session_id) REFERENCES sessions(session_id))''')
+                      
         conn.commit()
+        print("Database initialized successfully")
+        
     except sqlite3.Error as e:
-        print(f"Database error: {e}")
+        print(f"Database initialization error: {e}")
         raise
     finally:
         if conn:
             conn.close()
 
-
+def check_db_status():
+    try:
+        conn = sqlite3.connect('chat_history.db')
+        c = conn.cursor()
+        
+        # 테이블 존재 여부 확인
+        c.execute("SELECT name FROM sqlite_master WHERE type='table'")
+        tables = c.fetchall()
+        print(f"Existing tables: {tables}")
+        
+        # 데이터 개수 확인
+        c.execute("SELECT COUNT(*) FROM chat_history")
+        chat_count = c.fetchone()[0]
+        print(f"Number of chat records: {chat_count}")
+        
+        conn.close()
+    except Exception as e:
+        print(f"Error checking database status: {e}")
+        
 def create_session():
     max_attempts = 5
     for attempt in range(max_attempts):
@@ -108,14 +128,7 @@ def create_session():
     
 
 
-# 대화 내용 저장
-def save_chat(session_id, prompt, response):
-    conn = sqlite3.connect('chat_history.db')
-    c = conn.cursor()
-    c.execute("INSERT INTO chat_history (session_id, prompt, response, timestamp) VALUES (?, ?, ?, ?)",
-              (session_id, prompt, response, datetime.now()))
-    conn.commit()
-    conn.close()
+
 
 # 세션별 히스토리 조회
 def get_session_history(session_id):
@@ -282,8 +295,19 @@ class Demo:
                 collected_content = content
 
             if collected_content:
-                # 채팅 내용 저장
-                save_chat(self.current_session, query, collected_content)
+                try:
+                    print(f"Attempting to save chat with session_id: {self.current_session}")
+                    print(f"Query: {query}")
+                    print(f"Content length: {len(collected_content)}")
+            
+            # 채팅 내용 저장
+                    save_chat(self.current_session, query, collected_content)
+                    print("Chat saved successfully")
+            
+                except Exception as save_error:
+                    print(f"Error saving chat: {save_error}")
+
+
                 
                 _history = messages_to_history([
                     {'role': Role.SYSTEM, 'content': system_message}
@@ -399,22 +423,41 @@ def load_session_history(selected_session):
         print(f"Error loading session history: {e}")
         return []
 
-        
 def save_chat(session_id, prompt, response):
+    print(f"Starting save_chat with session_id: {session_id}")
+    conn = None
     try:
         conn = sqlite3.connect('chat_history.db')
         c = conn.cursor()
-        current_time = datetime.now()
+        
+        # 세션이 존재하는지 확인
+        c.execute("SELECT 1 FROM sessions WHERE session_id = ?", (session_id,))
+        if not c.fetchone():
+            print(f"Session {session_id} not found, creating new session")
+            c.execute("INSERT INTO sessions (session_id, created_at) VALUES (?, ?)",
+                     (session_id, datetime.now()))
+        
+        # 채팅 저장
         c.execute("""
             INSERT INTO chat_history (session_id, prompt, response, timestamp) 
             VALUES (?, ?, ?, ?)
-        """, (session_id, prompt, response, current_time))
+        """, (session_id, prompt, response, datetime.now()))
+        
         conn.commit()
-        print(f"Chat saved for session {session_id}")
+        print(f"Successfully saved chat for session {session_id}")
+        
+    except sqlite3.Error as e:
+        print(f"Database error: {e}")
+        if conn:
+            conn.rollback()
+        raise
     except Exception as e:
-        print(f"Error saving chat: {e}")
+        print(f"Unexpected error: {e}")
+        if conn:
+            conn.rollback()
+        raise
     finally:
-        if 'conn' in locals():
+        if conn:
             conn.close()
 
 # Demo 인스턴스 먼저 생성