seawolf2357 commited on
Commit
1801079
·
verified ·
1 Parent(s): 436cf31

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -57,17 +57,14 @@ import sqlite3
57
  from datetime import datetime
58
 
59
  def init_db():
60
- print("Initializing database...")
61
- conn = None
62
  try:
63
  conn = sqlite3.connect('chat_history.db')
64
  c = conn.cursor()
65
 
66
- # 테이블 생성
67
  c.execute('''CREATE TABLE IF NOT EXISTS sessions
68
  (session_id TEXT PRIMARY KEY,
69
  created_at TIMESTAMP)''')
70
-
71
  c.execute('''CREATE TABLE IF NOT EXISTS chat_history
72
  (id INTEGER PRIMARY KEY AUTOINCREMENT,
73
  session_id TEXT,
@@ -75,12 +72,11 @@ def init_db():
75
  response TEXT,
76
  timestamp TIMESTAMP,
77
  FOREIGN KEY (session_id) REFERENCES sessions(session_id))''')
78
-
79
  conn.commit()
80
  print("Database initialized successfully")
81
 
82
  except sqlite3.Error as e:
83
- print(f"Database initialization error: {e}")
84
  raise
85
  finally:
86
  if conn:
 
57
  from datetime import datetime
58
 
59
  def init_db():
 
 
60
  try:
61
  conn = sqlite3.connect('chat_history.db')
62
  c = conn.cursor()
63
 
64
+ # 테이블이 없을 때만 생성
65
  c.execute('''CREATE TABLE IF NOT EXISTS sessions
66
  (session_id TEXT PRIMARY KEY,
67
  created_at TIMESTAMP)''')
 
68
  c.execute('''CREATE TABLE IF NOT EXISTS chat_history
69
  (id INTEGER PRIMARY KEY AUTOINCREMENT,
70
  session_id TEXT,
 
72
  response TEXT,
73
  timestamp TIMESTAMP,
74
  FOREIGN KEY (session_id) REFERENCES sessions(session_id))''')
 
75
  conn.commit()
76
  print("Database initialized successfully")
77
 
78
  except sqlite3.Error as e:
79
+ print(f"Database error: {e}")
80
  raise
81
  finally:
82
  if conn: