firestore_init.py

from google.cloud import firestore

# Initialize Firestore client
db = firestore.Client()

# Create sample documents to initialize collections
collections = [
    {
        'name': 'generations',
        'sample': {
            'id': 'sample',
            'prompt': 'Sample generation',
            'status': 'completed',
            'created_at': firestore.SERVER_TIMESTAMP
        }
    },
    {
        'name': 'quality_reports',
        'sample': {
            'id': 'sample',
            'generation_id': 'sample',
            'overall_score': 0,
            'created_at': firestore.SERVER_TIMESTAMP
        }
    },
    {
        'name': 'users',
        'sample': {
            'id': 'sample',
            'created_at': firestore.SERVER_TIMESTAMP
        }
    }
]

for collection in collections:
    doc_ref = db.collection(collection['name']).document('_init')
    doc_ref.set(collection['sample'])
    print(f"✓ Initialized collection: {collection['name']}")

print("\n✅ Firestore collections initialized!")