Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import pandas as pd
 import firebase_admin
-from firebase_admin import credentials, firestore
+from firebase_admin import credentials, firestore, initialize_app
 import io
 from io import StringIO
 from datetime import datetime, timedelta
@@ -10,11 +10,19 @@ import uuid
 from PIL import Image
 import os
 import base64
+import json
+
 
 
 # Initialize Firebase
 if not firebase_admin._apps:
-    cred = credentials.Certificate("/app/src/firebase_key.json")
+
+    # HF codes
+    firebase_key = os.environ["FIREBASE_KEY"]
+    cred = credentials.Certificate(json.loads(firebase_key))
+    
+    # Streamlit, Google Cloud
+    #cred = credentials.Certificate("/app/src/firebase_key.json")
     firebase_admin.initialize_app(cred)
 
 db = firestore.client()