update

README.md

@@ -59,8 +59,9 @@ CANVA_ACCESS_TOKEN=your_canva_access_token
 LINKEDIN_ACCESS_TOKEN=your_linkedin_access_token
 LINKEDIN_PERSON_URN=your_linkedin_person_urn
 
-# Database (for production)
-DATABASE_URL=postgresql://user:password@localhost/postgen
+# Database (CockroachDB connection string)
+# Format: postgresql://username:password@host:port/database?sslmode=verify-full
+DATABASE_URL=postgresql://seth:YOUR_PASSWORD@ezofis-11210.jxf.gcp-us-east1.cockroachlabs.cloud:26257/defaultdb?sslmode=verify-full
 ```
 
 ### Local Development
@@ -130,9 +131,12 @@ docker run -p 7860:7860 --env-file .env postgen
 2. Add the following secrets:
    - `OPENAI_API_KEY`: Your OpenAI API key
    - `OPENAI_MODEL`: gpt-4o (or your preferred model)
+   - `DATABASE_URL`: Your CockroachDB connection string (format: `postgresql://username:password@host:port/database?sslmode=verify-full`)
    - `CANVA_ACCESS_TOKEN`: (Optional, can be set per user)
    - `LINKEDIN_ACCESS_TOKEN`: (Optional, can be set per user)
 
+**Note**: The app will work with dummy/mock data if `DATABASE_URL` is not set. As you implement features, they will gradually use the database instead of mock data.
+
 ### Step 4: Connect Your Repository
 
 1. In Space settings, connect your Git repository
@@ -213,6 +217,28 @@ PostGen/
 └── README.md
 ```
 
+## Database Setup (CockroachDB)
+
+### Setting Up CockroachDB Connection
+
+1. **Get Your Connection String**:
+   - From your CockroachDB dashboard, copy the connection string
+   - Format: `postgresql://username:password@host:port/database?sslmode=verify-full`
+   - Example: `postgresql://seth:YOUR_PASSWORD@ezofis-11210.jxf.gcp-us-east1.cockroachlabs.cloud:26257/defaultdb?sslmode=verify-full`
+
+2. **Add to HuggingFace Spaces**:
+   - Go to your Space settings → "Variables and secrets"
+   - Add `DATABASE_URL` with your connection string
+
+3. **Database Tables**:
+   - Tables are automatically created on first startup
+   - The app will initialize: `users`, `integrations`, `assets`, `posts`, `campaigns`
+
+4. **Dummy Data**:
+   - The app currently uses dummy/mock data for all endpoints
+   - As features are implemented, they will gradually use the database
+   - Dummy data will be removed feature by feature as database integration is completed
+
 ## Next Steps After Deployment
 
 1. **Connect Integrations**:

backend/app/__init__.py

@@ -1,2 +1 @@
 # PostGen Backend Application
-

backend/app/database.py

@@ -0,0 +1,59 @@
+import os
+from sqlalchemy import create_engine, text
+from sqlalchemy.orm import sessionmaker, declarative_base
+from sqlalchemy.pool import NullPool
+
+# Get database URL from environment variable
+# Default to SQLite for local development if not set
+DATABASE_URL = os.getenv(
+    "DATABASE_URL",
+    "sqlite:///./postgen.db"
+)
+
+# For CockroachDB, we need to handle SSL and connection pooling
+if DATABASE_URL.startswith("postgresql://") or DATABASE_URL.startswith("postgres://"):
+    # CockroachDB connection - use NullPool to avoid connection issues
+    # CockroachDB requires SSL, so we ensure sslmode is set
+    if "sslmode" not in DATABASE_URL:
+        separator = "&" if "?" in DATABASE_URL else "?"
+        DATABASE_URL = f"{DATABASE_URL}{separator}sslmode=verify-full"
+    
+    engine = create_engine(
+        DATABASE_URL,
+        poolclass=NullPool,  # CockroachDB works better with NullPool
+        echo=False  # Set to True for SQL query debugging
+    )
+else:
+    # SQLite for local development
+    engine = create_engine(
+        DATABASE_URL,
+        connect_args={"check_same_thread": False} if "sqlite" in DATABASE_URL else {}
+    )
+
+SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+
+Base = declarative_base()
+
+def get_db():
+    """Dependency to get database session"""
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
+
+def init_db():
+    """Initialize database tables"""
+    try:
+        from app.models import User, Integration, Asset, Post, Campaign
+        # Test connection first
+        with engine.connect() as conn:
+            conn.execute(text("SELECT 1"))
+            conn.commit()
+        # Create tables if connection successful
+        Base.metadata.create_all(bind=engine)
+        return True
+    except Exception as e:
+        print(f"Database connection failed: {e}")
+        return False
+

backend/app/main.py

@@ -15,6 +15,8 @@ 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
+from sqlalchemy.orm import Session
 
 app = FastAPI(title="PostGen API", version="1.0.0")
 
@@ -26,6 +28,18 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+# Initialize database on startup
+@app.on_event("startup")
+async def startup_event():
+    """Initialize database tables on startup"""
+    db_initialized = init_db()
+    if db_initialized:
+        print("✓ Database initialized successfully")
+    else:
+        print("⚠ Database not available - using mock data")
+        print("⚠ App will function normally with dummy content")
+        print("⚠ To connect to database, set DATABASE_URL environment variable")
+
 # Services
 ai_service = AIService()
 

backend/app/models.py

@@ -1,9 +1,7 @@
 from sqlalchemy import Column, Integer, String, DateTime, Boolean, JSON, Text, ForeignKey
-from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import relationship
 from datetime import datetime
-
-Base = declarative_base()
+from app.database import Base
 
 class User(Base):
     __tablename__ = "users"