Update main.py

main.py

@@ -106,8 +106,11 @@ async def verify_api_key(api_key: str = Security(api_key_header)):
     return api_key
 
 # SQLite setup
+DB_PATH = '/app/data/conversations.db'
+
 def init_db():
-    conn = sqlite3.connect('conversations.db')
+    os.makedirs(os.path.dirname(DB_PATH), exist_ok=True)
+    conn = sqlite3.connect(DB_PATH)
     c = conn.cursor()
     c.execute('''CREATE TABLE IF NOT EXISTS conversations
                  (id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -122,7 +125,7 @@ def init_db():
 init_db()
 
 def update_db(user_id, conversation_id, message, response):
-    conn = sqlite3.connect('conversations.db')
+    conn = sqlite3.connect(DB_PATH)
     c = conn.cursor()
     c.execute('''INSERT INTO conversations (user_id, conversation_id, message, response)
                  VALUES (?, ?, ?, ?)''', (user_id, conversation_id, message, response))