Latest2
Browse files
app.py
CHANGED
@@ -20,17 +20,20 @@ app.add_middleware(SessionMiddleware, secret_key=os.urandom(24))
|
|
20 |
client = Groq(api_key='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')
|
21 |
|
22 |
# Database initialization
|
|
|
|
|
23 |
def init_db():
|
24 |
-
|
|
|
|
|
|
|
25 |
c = conn.cursor()
|
26 |
-
|
27 |
-
# Users table
|
28 |
c.execute('''CREATE TABLE IF NOT EXISTS users
|
29 |
(id INTEGER PRIMARY KEY AUTOINCREMENT,
|
30 |
username TEXT UNIQUE NOT NULL,
|
31 |
password TEXT NOT NULL,
|
32 |
created_at DATETIME DEFAULT CURRENT_TIMESTAMP)''')
|
33 |
-
|
34 |
c.execute('''CREATE TABLE IF NOT EXISTS chats
|
35 |
(id INTEGER PRIMARY KEY AUTOINCREMENT,
|
36 |
user_id INTEGER NOT NULL,
|
@@ -44,7 +47,6 @@ def init_db():
|
|
44 |
conn.close()
|
45 |
|
46 |
init_db()
|
47 |
-
|
48 |
# Helper functions
|
49 |
def hash_password(password):
|
50 |
return hashlib.sha256(password.encode()).hexdigest()
|
|
|
20 |
client = Groq(api_key='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')
|
21 |
|
22 |
# Database initialization
|
23 |
+
DATABASE_PATH = '/app/chat_history.db'
|
24 |
+
|
25 |
def init_db():
|
26 |
+
# Ensure the directory exists
|
27 |
+
os.makedirs(os.path.dirname(DATABASE_PATH), exist_ok=True)
|
28 |
+
|
29 |
+
conn = sqlite3.connect(DATABASE_PATH)
|
30 |
c = conn.cursor()
|
|
|
|
|
31 |
c.execute('''CREATE TABLE IF NOT EXISTS users
|
32 |
(id INTEGER PRIMARY KEY AUTOINCREMENT,
|
33 |
username TEXT UNIQUE NOT NULL,
|
34 |
password TEXT NOT NULL,
|
35 |
created_at DATETIME DEFAULT CURRENT_TIMESTAMP)''')
|
36 |
+
|
37 |
c.execute('''CREATE TABLE IF NOT EXISTS chats
|
38 |
(id INTEGER PRIMARY KEY AUTOINCREMENT,
|
39 |
user_id INTEGER NOT NULL,
|
|
|
47 |
conn.close()
|
48 |
|
49 |
init_db()
|
|
|
50 |
# Helper functions
|
51 |
def hash_password(password):
|
52 |
return hashlib.sha256(password.encode()).hexdigest()
|