Update app.py

app.py

@@ -11,6 +11,104 @@ import json
 
 DB_FILE = "life_tracker.db"
 
+def create_tables():
+    conn = create_connection()
+    cursor = conn.cursor()
+    
+    cursor.execute('''
+    CREATE TABLE IF NOT EXISTS users (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        name TEXT UNIQUE
+    )
+    ''')
+    
+    cursor.execute('''
+    CREATE TABLE IF NOT EXISTS daily_activities (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        user_id INTEGER,
+        date TEXT,
+        category TEXT,
+        subcategory TEXT,
+        start_time TEXT,
+        end_time TEXT,
+        FOREIGN KEY (user_id) REFERENCES users (id)
+    )
+    ''')
+    
+    cursor.execute('''
+    CREATE TABLE IF NOT EXISTS qualitative_metrics (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        user_id INTEGER,
+        date TEXT,
+        life_score INTEGER,
+        work_score INTEGER,
+        health_score INTEGER,
+        FOREIGN KEY (user_id) REFERENCES users (id)
+    )
+    ''')
+    
+    cursor.execute('''
+    CREATE TABLE IF NOT EXISTS quantitative_metrics (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        user_id INTEGER,
+        date TEXT,
+        wake_up_time TEXT,
+        workouts INTEGER,
+        meditation_minutes INTEGER,
+        brain_training_minutes INTEGER,
+        FOREIGN KEY (user_id) REFERENCES users (id)
+    )
+    ''')
+    
+    cursor.execute('''
+    CREATE TABLE IF NOT EXISTS goals (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        user_id INTEGER,
+        category TEXT,
+        description TEXT,
+        target_value REAL,
+        current_value REAL,
+        start_date TEXT,
+        end_date TEXT,
+        FOREIGN KEY (user_id) REFERENCES users (id)
+    )
+    ''')
+    
+    cursor.execute('''
+    CREATE TABLE IF NOT EXISTS user_settings (
+        user_id INTEGER PRIMARY KEY,
+        default_wake_time TEXT,
+        work_weight REAL,
+        life_weight REAL,
+        health_weight REAL,
+        FOREIGN KEY (user_id) REFERENCES users (id)
+    )
+    ''')
+    
+    cursor.execute('''
+    CREATE TABLE IF NOT EXISTS custom_categories (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        user_id INTEGER,
+        category_name TEXT,
+        subcategories TEXT,
+        FOREIGN KEY (user_id) REFERENCES users (id)
+    )
+    ''')
+    
+    cursor.execute('''
+    CREATE TABLE IF NOT EXISTS daily_checklist (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        user_id INTEGER,
+        date TEXT,
+        checklist_data TEXT,
+        notes TEXT,
+        FOREIGN KEY (user_id) REFERENCES users (id)
+    )
+    ''')
+    
+    conn.commit()
+    conn.close()
+
 def create_connection():
     return sqlite3.connect(DB_FILE)
 
@@ -503,6 +601,7 @@ function edit_day(date) {
 
 # Create the main Gradio interface with the custom theme and CSS
 with gr.Blocks(theme=custom_theme, css=custom_css) as demo:
+    create_tables() 
     with gr.Row(equal_height=True):
         # Left sidebar
         with gr.Column(scale=1, min_width=200):
@@ -1007,6 +1106,10 @@ with gr.Blocks(theme=custom_theme, css=custom_css) as demo:
         cursor = conn.cursor()
         cursor.execute("SELECT name FROM users")
         users = [row[0] for row in cursor.fetchall()]
+        if not users:
+            cursor.execute("INSERT INTO users (name) VALUES (?)", ("John Doe",))
+            conn.commit()
+            users = ["John Doe"]
         conn.close()
         return users