Create app.py

app.py

@@ -0,0 +1,71 @@
+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()