Update db/helpers.py

db/helpers.py

@@ -6,31 +6,31 @@ from sqlalchemy import func
 from db.models import User, ContentHistory, Feedback
 from db.connection import get_db, SessionLocal
 
-# ------------------------------------
-# ✅ User Management
-# ------------------------------------
-def create_user(user_id):
-    """Create a new user in the database"""
+def ensure_user_exists(user_id):
+    """Ensure user exists in database - compatibility function for existing code"""
     with next(get_db()) as db:
         try:
-            user = User(id=user_id)
-            db.add(user)
-            db.commit()
-            print(f"✅ Created new user: {user_id}")
+            user = db.query(User).filter_by(id=user_id).first()
+            if not user:
+                # Create a minimal user record if it doesn't exist
+                # This maintains compatibility with existing code that expects this function
+                user = User()
+                user.id = uuid.UUID(user_id)
+                user.fullname = "Legacy User"
+                user.username = f"user_{user_id[:8]}"
+                user.password_hash = ""
+                user.salt = ""
+                
+                db.add(user)
+                db.commit()
+                print(f"✅ Created legacy user record: {user_id}")
+                return True
             return True
         except SQLAlchemyError as e:
             db.rollback()
-            # User likely already exists - this is fine
+            print(f"❌ Error ensuring user exists: {e}")
             return False
 
-def ensure_user_exists(user_id):
-    """Ensure user exists in database, create if needed"""
-    with next(get_db()) as db:
-        user = db.query(User).filter_by(id=user_id).first()
-        if not user:
-            return create_user(user_id)
-        return True
-
 # ------------------------------------
 # ✅ Content History
 # ------------------------------------
@@ -38,9 +38,6 @@ def get_user_history(user_id):
     """Get user's content history from database"""
     with next(get_db()) as db:
         try:
-            # Ensure user exists first
-            ensure_user_exists(user_id)
-            
             history = db.query(ContentHistory).filter_by(user_id=user_id).order_by(ContentHistory.created_at.desc()).all()
             print(f"✅ Loaded {len(history)} history entries for user {user_id}")
             return history
@@ -49,15 +46,12 @@ def get_user_history(user_id):
             return []
 
 def save_content_to_history(data):
-    """Save content to history"""
+    """Save content to history - user_id comes from authenticated session"""
     with next(get_db()) as db:
         try:
-            # Ensure user exists
-            ensure_user_exists(data["user_id"])
-            
             content = ContentHistory(
                 id=uuid.uuid4(),
-                user_id=data["user_id"],
+                user_id=data["user_id"],  # From authenticated session
                 prompt=data["prompt"],
                 output=data["output"],
                 user_type=data["user_type"],