app.py

import sqlite3

def init_db():
    conn = sqlite3.connect("chat_history.db")
    c = conn.cursor()
    c.execute('''
        CREATE TABLE IF NOT EXISTS messages (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            user TEXT,
            assistant TEXT,
            timestamp DATETIME DEFAULT CURRENT_TIMESTAMP
        )
    ''')
    conn.commit()
    conn.close()

def save_to_db(user, assistant):
    conn = sqlite3.connect("chat_history.db")
    c = conn.cursor()
    c.execute('INSERT INTO messages (user, assistant) VALUES (?, ?)', (user, assistant))
    conn.commit()
    conn.close()

def generate_response(prompt):
    return f"I'm a placeholder AI. You said: {prompt}"