feat: firestore setup

.env.example

@@ -0,0 +1 @@
+GOOGLE_APPLICATION_CREDENTIALS=

.gitignore

@@ -0,0 +1,2 @@
+.env
+limkopi-9d239-*.json

README.md

@@ -36,10 +36,20 @@ GIT_LFS_SKIP_SMUDGE=1
 
 ### 2. Environment Setup
 
+### Create virtual environment
+
 ```sh
 python3 -m venv venv
 ```
 
+### Set environment variables
+
+Fill in the corresponding credentials accordingly in `.env`
+
+```sh
+cp .env.example .env
+```
+
 #### Activate the virtual environment
 
 - **On Windows**

app.py

@@ -1,7 +1,23 @@
+import os
+import json
+from google.cloud import firestore
 import gradio as gr
 import whisper
 from transformers import pipeline
+from dotenv import load_dotenv
+import base64
 
+# Load google cloud credentials
+load_dotenv()
+base64_credentials = os.environ.get('GOOGLE_APPLICATION_CREDENTIALS')
+decoded_credentials = base64.b64decode(base64_credentials).decode()
+credentials_json = json.loads(decoded_credentials)
+db = firestore.Client.from_service_account_info(credentials_json)
+
+collection_ref = db.collection('Users')
+docs = collection_ref.stream()
+
+# Load model
 model = whisper.load_model("base")
 sentiment_analysis = pipeline("sentiment-analysis", framework="pt", model="SamLowe/roberta-base-go_emotions")
 

requirements.txt

@@ -4,3 +4,5 @@ gradio
 torch
 torchaudio
 torchvision
+google-cloud-firestore
+python-dotenv