Spaces:
Sleeping
Sleeping
File size: 2,062 Bytes
3c91d73 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 | import sqlite3
# Step 1: Connect to the database (or create it if it doesn't exist)
conn = sqlite3.connect('huggingface_app.db')
cursor = conn.cursor()
# Step 2: Create tables
def create_tables():
cursor.execute('''
CREATE TABLE IF NOT EXISTS users (
id INTEGER PRIMARY KEY AUTOINCREMENT,
username TEXT NOT NULL,
email TEXT UNIQUE
)
''')
cursor.execute('''
CREATE TABLE IF NOT EXISTS model_interactions (
id INTEGER PRIMARY KEY AUTOINCREMENT,
user_id INTEGER,
query TEXT NOT NULL,
response TEXT NOT NULL,
model_name TEXT NOT NULL,
timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
FOREIGN KEY (user_id) REFERENCES users(id)
)
''')
print("Tables created successfully!")
# Step 3: Insert Data
def insert_user(username, email):
cursor.execute('''
INSERT INTO users (username, email)
VALUES (?, ?)
''', (username, email))
conn.commit()
print(f"User {username} added.")
def log_interaction(user_id, query, response, model_name):
cursor.execute('''
INSERT INTO model_interactions (user_id, query, response, model_name)
VALUES (?, ?, ?, ?)
''', (user_id, query, response, model_name))
conn.commit()
print("Interaction logged.")
# Step 4: Query Data
def get_user_interactions(user_id):
cursor.execute('''
SELECT * FROM model_interactions WHERE user_id = ?
''', (user_id,))
return cursor.fetchall()
# Step 5: Close connection
def close_connection():
conn.close()
print("Database connection closed.")
# Initialize the database
if __name__ == "__main__":
create_tables()
# Example usage
insert_user("test_user", "test@example.com")
log_interaction(1, "What is AI?", "AI stands for Artificial Intelligence.", "gpt-4")
# Fetch user interactions
interactions = get_user_interactions(1)
for interaction in interactions:
print(interaction)
close_connection()
|