image-retrieval / src /firebase /firebase_config.py
ABAO77's picture
Upload 5 files
0295bf1 verified
import firebase_admin
from firebase_admin import credentials
from firebase_admin import storage
import os
from dotenv import load_dotenv
# Load environment variables
load_dotenv()
firebase_url_storageBucket = os.getenv("URL_STORAGEBUCKET")
# Get credentials from environment variables
credential_firebase = {
"type": os.getenv("TYPE"),
"project_id": os.getenv("PROJECT_ID"),
"private_key_id": os.getenv("PRIVATE_KEY_ID"),
"private_key": os.getenv("PRIVATE_KEY"),
"client_email": os.getenv("CLIENT_EMAIL"),
"client_id": os.getenv("CLIENT_ID"),
"auth_uri": os.getenv("AUTH_URI"),
"token_uri": os.getenv("TOKEN_URI"),
"auth_provider_x509_cert_url": os.getenv("AUTH_PROVIDER_X509_CERT_URL"),
"client_x509_cert_url": os.getenv("CLIENT_X509_CERT_URL"),
"universe_domain": os.getenv("UNIVERSE_DOMAIN"),
}
# Check if the app is not initialized yet
if not firebase_admin._apps:
# Initialize the app with the credentials
cred = credentials.Certificate(credential_firebase)
firebase_admin.initialize_app(cred, {"storageBucket": firebase_url_storageBucket})
# Initialize Firestore
firebase_bucket = storage.bucket(app=firebase_admin.get_app())
print("Storage connected")