Update app.py

app.py

@@ -55,11 +55,16 @@ from config import DEMO_LIST
 import sqlite3
 from datetime import datetime
 
-
 def init_db():
     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)''')
@@ -73,19 +78,35 @@ def init_db():
         conn.commit()
     except sqlite3.Error as e:
         print(f"Database error: {e}")
+        raise
     finally:
         if conn:
             conn.close()
 
-# 새로운 세션 생성
+
 def create_session():
-    session_id = datetime.now().strftime("%Y%m%d_%H%M%S")
-    conn = sqlite3.connect('chat_history.db')
-    c = conn.cursor()
-    c.execute("INSERT INTO sessions VALUES (?, ?)", (session_id, datetime.now()))
-    conn.commit()
-    conn.close()
-    return session_id
+    max_attempts = 5
+    for attempt in range(max_attempts):
+        try:
+            # 밀리초까지 포함한 더 상세한 타임스탬프 사용
+            session_id = datetime.now().strftime("%Y%m%d_%H%M%S_%f")
+            conn = sqlite3.connect('chat_history.db')
+            c = conn.cursor()
+            c.execute("INSERT INTO sessions VALUES (?, ?)", (session_id, datetime.now()))
+            conn.commit()
+            conn.close()
+            return session_id
+        except sqlite3.IntegrityError:
+            if attempt == max_attempts - 1:
+                raise
+            time.sleep(0.1)  # 잠시 대기 후 재시도
+        finally:
+            if 'conn' in locals():
+                conn.close()
+    
+    raise Exception("Failed to create unique session ID after multiple attempts")
+    
+
 
 # 대화 내용 저장
 def save_chat(session_id, prompt, response):
@@ -581,7 +602,11 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
 
 
 if __name__ == "__main__":
-    init_db()
-    clear_expired_sessions()  # 만료된 세션 정리
-    demo_instance = Demo()
-    demo.queue(default_concurrency_limit=20).launch(ssr_mode=False)
+    try:
+        init_db()
+        clear_expired_sessions()
+        demo_instance = Demo()
+        demo.queue(default_concurrency_limit=20).launch(ssr_mode=False)
+    except Exception as e:
+        print(f"Initialization error: {e}")
+        raise