Update app.py

app.py

@@ -38,9 +38,13 @@ class ZhipuAILLM(LLM):
         return response.choices[0].message.content
 
 # Database setup
-DB_PATH = "/data/chat_history.db"
+DB_DIR = "/data"
+DB_PATH = os.path.join(DB_DIR, "chat_history.db")
 
 def init_db():
+    # Create the data directory if it doesn't exist
+    os.makedirs(DB_DIR, exist_ok=True)
+    
     conn = sqlite3.connect(DB_PATH)
     cursor = conn.cursor()
     cursor.execute('''