Create db table

app.py

@@ -4,6 +4,7 @@ from langchain.schema import AIMessage, HumanMessage
 import openai
 import gradio as gr
 import os
+import sqlite3
 
 os.environ["OPENAI_API_KEY"] = os.getenv('OPENAI_API_KEY')
 llm = ChatOpenAI(temperature=1.0, model='gpt-4o-mini')
@@ -11,6 +12,25 @@ llm = ChatOpenAI(temperature=1.0, model='gpt-4o-mini')
 question_json = os.getenv('QUESTION_JSON')
 cv_json = os.getenv('CV_JSON')
 
+DB_FILE = "./messages.db"
+db = sqlite3.connect(DB_FILE)
+
+# Create table if it doesn't already exist
+try:
+    db.execute("SELECT * FROM messages").fetchall()
+    db.close()
+except sqlite3.OperationalError:
+    db.execute(
+        '''
+        CREATE TABLE messages (
+            id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
+            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP NOT NULL,
+            message TEXT
+        )
+        ''')
+    db.commit()
+    db.close()
+
 def predict(message, history):
     if find(message):
         prompt = os.getenv('PREDICT_PROMPT').format(question_json, message)