File size: 599 Bytes
a5b0b44
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
# firebase/firebase_config.py

import firebase_admin
from firebase_admin import credentials, firestore
import os

def init_firebase():
    if not firebase_admin._apps:
        cred_path = "firebase_credentials.json"  # upload this via HF interface
        cred = credentials.Certificate(cred_path)
        firebase_admin.initialize_app(cred)
    return firestore.client()

def log_message_to_firestore(message_dict):
    db = init_firebase()
    chat_ref = db.collection("conversations").document("bot_talk_log")
    chat_ref.update({
        "messages": firestore.ArrayUnion([message_dict])
    })