update

backend/app/database.py

@@ -111,6 +111,70 @@ def get_direct_psycopg2_connection():
         print(f"Failed to create direct psycopg2 connection: {e}")
         return None
 
+def ensure_default_user():
+    """Ensure a default user (id=1) exists in the database"""
+    try:
+        conn = get_direct_psycopg2_connection()
+        if not conn:
+            return 1  # Return default ID if connection fails
+        
+        try:
+            cursor = conn.cursor()
+            # Check if user with id=1 exists
+            cursor.execute("SELECT id FROM users WHERE id = 1")
+            if cursor.fetchone():
+                cursor.close()
+                conn.close()
+                return 1
+            
+            # Create default user if it doesn't exist
+            # Try with ON CONFLICT first (PostgreSQL/CockroachDB)
+            try:
+                cursor.execute("""
+                    INSERT INTO users (id, email, name, created_at)
+                    VALUES (1, 'default@postgen.app', 'Default User', NOW())
+                    ON CONFLICT (id) DO NOTHING
+                """)
+            except Exception:
+                # If ON CONFLICT fails, try without it (might be unique constraint on email)
+                try:
+                    cursor.execute("""
+                        INSERT INTO users (id, email, name, created_at)
+                        VALUES (1, 'default@postgen.app', 'Default User', NOW())
+                    """)
+                except Exception as insert_error:
+                    # User might already exist (race condition), check again
+                    cursor.execute("SELECT id FROM users WHERE id = 1 OR email = 'default@postgen.app' LIMIT 1")
+                    row = cursor.fetchone()
+                    if row:
+                        cursor.close()
+                        conn.close()
+                        return row[0]
+                    # If still fails, re-raise
+                    raise insert_error
+            
+            conn.commit()
+            cursor.close()
+            conn.close()
+            return 1
+        except Exception as e:
+            # If everything fails, try to get any existing user or return default
+            try:
+                cursor.execute("SELECT id FROM users LIMIT 1")
+                row = cursor.fetchone()
+                cursor.close()
+                conn.close()
+                if row:
+                    return row[0]
+            except:
+                if conn:
+                    conn.close()
+            print(f"Warning: Could not ensure default user: {e}")
+            return 1  # Return default ID as fallback
+    except Exception as e:
+        print(f"Error ensuring default user: {e}")
+        return 1  # Return default ID as fallback
+
 def init_db():
     """Initialize database tables"""
     try:

backend/app/main.py

@@ -16,7 +16,7 @@ from app.schemas import (
 from app.services.canva_service import CanvaService
 from app.services.linkedin_service import LinkedInService
 from app.services.ai_service import AIService
-from app.database import init_db, get_db, get_direct_psycopg2_connection
+from app.database import init_db, get_db, get_direct_psycopg2_connection, ensure_default_user
 from sqlalchemy.orm import Session
 
 app = FastAPI(title="PostGen API", version="1.0.0")
@@ -41,6 +41,12 @@ async def startup_event():
     db_initialized = init_db()
     if db_initialized:
         print("✓ Database initialized successfully")
+        # Ensure default user exists
+        try:
+            user_id = ensure_default_user()
+            print(f"✓ Default user ready (id={user_id})")
+        except Exception as e:
+            print(f"⚠ Could not ensure default user: {e}")
     else:
         print("⚠ Database not available - using mock data")
         print("⚠ App will function normally with dummy content")
@@ -186,7 +192,9 @@ async def upload_asset(
         # Save to database (keep dummy content as requested)
         try:
             from app.models import Asset
-            # For now, use a default user_id of 1 (in production, get from auth)
+            # Ensure default user exists and get user_id
+            user_id = ensure_default_user()
+            
             db_asset = Asset(
                 name=file.filename,  # Keep original filename for display
                 file_path=str(file_path),  # Store unique filename on disk
@@ -194,7 +202,7 @@ async def upload_asset(
                 product_category=product_category or "ocr",
                 sub_category=sub_category if sub_category and sub_category != "none" else None,
                 size=file_size,
-                user_id=1  # Default user - in production, get from session/auth
+                user_id=user_id
             )
             db.add(db_asset)
             try:
@@ -234,6 +242,8 @@ async def upload_asset(
                 error_str = str(commit_error)
                 if "Could not determine version" in error_str:
                     # Use direct psycopg2 connection to bypass SQLAlchemy
+                    # Ensure default user exists first
+                    user_id = ensure_default_user()
                     conn = get_direct_psycopg2_connection()
                     if conn:
                         try:
@@ -249,7 +259,7 @@ async def upload_asset(
                                 product_category or "ocr",
                                 sub_category if sub_category and sub_category != "none" else None,
                                 file_size,
-                                1
+                                user_id
                             ))
                             row = cursor.fetchone()
                             conn.commit()