commit

src/database.py

@@ -1,96 +1,96 @@
-# database.py
-import sqlite3
-import json
-from datetime import datetime
-from typing import Optional, Dict
+# # database.py
+# import sqlite3
+# import json
+# from datetime import datetime
+# from typing import Optional, Dict
 
-DB_PATH = "users.db"
+# DB_PATH = "users.db"
 
-# Ensure DB and table are created on import
-def init_db():
-    conn = sqlite3.connect(DB_PATH)
-    cursor = conn.cursor()
-    cursor.execute("""
-    CREATE TABLE IF NOT EXISTS users (
-        session_alias TEXT PRIMARY KEY,
-        name TEXT,
-        created_at TEXT,
-        profile TEXT,
-        chat_history TEXT,
-        workout_plan TEXT,
-        nutrition_plan TEXT,
-        last_updated TEXT
-    )
-    """)
-    conn.commit()
-    conn.close()
+# # Ensure DB and table are created on import
+# def init_db():
+#     conn = sqlite3.connect(DB_PATH)
+#     cursor = conn.cursor()
+#     cursor.execute("""
+#     CREATE TABLE IF NOT EXISTS users (
+#         session_alias TEXT PRIMARY KEY,
+#         name TEXT,
+#         created_at TEXT,
+#         profile TEXT,
+#         chat_history TEXT,
+#         workout_plan TEXT,
+#         nutrition_plan TEXT,
+#         last_updated TEXT
+#     )
+#     """)
+#     conn.commit()
+#     conn.close()
 
 
-def save_user_data(session_alias: str, user_data: Dict):
-    conn = sqlite3.connect(DB_PATH)
-    cursor = conn.cursor()
+# def save_user_data(session_alias: str, user_data: Dict):
+#     conn = sqlite3.connect(DB_PATH)
+#     cursor = conn.cursor()
 
-    cursor.execute("""
-    INSERT INTO users (
-        session_alias, name, created_at, profile, chat_history,
-        workout_plan, nutrition_plan, last_updated
-    ) VALUES (?, ?, ?, ?, ?, ?, ?, ?)
-    ON CONFLICT(session_alias) DO UPDATE SET
-        name=excluded.name,
-        profile=excluded.profile,
-        chat_history=excluded.chat_history,
-        workout_plan=excluded.workout_plan,
-        nutrition_plan=excluded.nutrition_plan,
-        last_updated=excluded.last_updated
-    """, (
-        session_alias,
-        user_data.get("name", ""),
-        user_data.get("created_at", datetime.now().isoformat()),
-        json.dumps(user_data.get("profile", {})),
-        json.dumps(user_data.get("chat_history", [])),
-        json.dumps(user_data.get("workout_plan", {})),
-        json.dumps(user_data.get("nutrition_plan", {})),
-        datetime.now().isoformat()
-    ))
+#     cursor.execute("""
+#     INSERT INTO users (
+#         session_alias, name, created_at, profile, chat_history,
+#         workout_plan, nutrition_plan, last_updated
+#     ) VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+#     ON CONFLICT(session_alias) DO UPDATE SET
+#         name=excluded.name,
+#         profile=excluded.profile,
+#         chat_history=excluded.chat_history,
+#         workout_plan=excluded.workout_plan,
+#         nutrition_plan=excluded.nutrition_plan,
+#         last_updated=excluded.last_updated
+#     """, (
+#         session_alias,
+#         user_data.get("name", ""),
+#         user_data.get("created_at", datetime.now().isoformat()),
+#         json.dumps(user_data.get("profile", {})),
+#         json.dumps(user_data.get("chat_history", [])),
+#         json.dumps(user_data.get("workout_plan", {})),
+#         json.dumps(user_data.get("nutrition_plan", {})),
+#         datetime.now().isoformat()
+#     ))
 
-    conn.commit()
-    conn.close()
+#     conn.commit()
+#     conn.close()
 
 
-def get_user_data(session_alias: str) -> Optional[Dict]:
-    conn = sqlite3.connect(DB_PATH)
-    cursor = conn.cursor()
-    cursor.execute("SELECT * FROM users WHERE session_alias = ?", (session_alias,))
-    row = cursor.fetchone()
-    conn.close()
+# def get_user_data(session_alias: str) -> Optional[Dict]:
+#     conn = sqlite3.connect(DB_PATH)
+#     cursor = conn.cursor()
+#     cursor.execute("SELECT * FROM users WHERE session_alias = ?", (session_alias,))
+#     row = cursor.fetchone()
+#     conn.close()
 
-    if not row:
-        return None
+#     if not row:
+#         return None
 
-    return {
-        "session_alias": row[0],
-        "name": row[1],
-        "created_at": row[2],
-        "profile": json.loads(row[3] or "{}"),
-        "chat_history": json.loads(row[4] or "[]"),
-        "workout_plan": json.loads(row[5] or "{}"),
-        "nutrition_plan": json.loads(row[6] or "{}"),
-        "last_updated": row[7]
-    }
+#     return {
+#         "session_alias": row[0],
+#         "name": row[1],
+#         "created_at": row[2],
+#         "profile": json.loads(row[3] or "{}"),
+#         "chat_history": json.loads(row[4] or "[]"),
+#         "workout_plan": json.loads(row[5] or "{}"),
+#         "nutrition_plan": json.loads(row[6] or "{}"),
+#         "last_updated": row[7]
+#     }
 
 
-def create_user(session_alias: str, name: str):
-    # Only called when new user logs in
-    user_data = {
-        "name": name,
-        "created_at": datetime.now().isoformat(),
-        "profile": {},
-        "chat_history": [],
-        "workout_plan": {},
-        "nutrition_plan": {}
-    }
-    save_user_data(session_alias, user_data)
+# def create_user(session_alias: str, name: str):
+#     # Only called when new user logs in
+#     user_data = {
+#         "name": name,
+#         "created_at": datetime.now().isoformat(),
+#         "profile": {},
+#         "chat_history": [],
+#         "workout_plan": {},
+#         "nutrition_plan": {}
+#     }
+#     save_user_data(session_alias, user_data)
 
 
-# Always call on app start
-init_db()
+# # Always call on app start
+# init_db()

src/streamlit_app.py

@@ -6,7 +6,7 @@ import pandas as pd
 import hashlib
 import os
 from service import generate_ai_response, create_user_profile_prompt, create_workout_type_prompt, create_nutrition_type_prompt, create_conversation_chat_prompt
-from database import create_user, get_user_data, save_user_data
+# from database import create_user, get_user_data, save_user_data
 
 # Configure page
 st.set_page_config(