Spaces:
Sleeping
Sleeping
File size: 2,425 Bytes
48f6350 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
import firebase_admin
from firebase_admin import credentials, firestore
from google.cloud.firestore_v1.base_query import FieldFilter
import streamlit as st
import os
@st.cache_resource
def get_db():
try:
cred = credentials.Certificate("enigmaescape-6506f-firebase-adminsdk-nqhe6-a2b5dacb8d.json")
except FileNotFoundError:
cred = credentials.Certificate({
"type": "service_account",
"project_id": "enigmaescape-6506f",
"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": "https://accounts.google.com/o/oauth2/auth",
"token_uri": "https://oauth2.googleapis.com/token",
"auth_provider_x509_cert_url": "https://www.googleapis.com/oauth2/v1/certs",
"client_x509_cert_url": os.getenv("client_x509_cert_url"),
"universe_domain": "googleapis.com",
})
firebase_admin.initialize_app(cred)
return firestore.client()
db = get_db()
users_col = db.collection(u'users')
points_col = db.collection(u'points')
def auth(teamname, password):
user = users_col.where(filter=FieldFilter(u'teamname', u'==', teamname)).where(
filter=FieldFilter(u'password', u'==', password)).get()
if user and user[0].exists: return user[0]
def add_points(user, points: int, level: int, tokens: int, prompt: str):
scores = user.reference.get().to_dict().get('scores', {})
tokens_next = min(tokens, scores.get(str(level), {}).get('tokens', tokens))
print(prompt)
user.reference.update({u'scores.' + str(level): {
u'points': points,
u'tokens': tokens_next,
u'best_prompt': prompt if tokens <= tokens_next else scores.get(str(level), {}).get("best_prompt", ""),
u'done': True,
# u'response_list': scores.get(str(level), {}).get("response_list", []) + [prompt],
}})
def get_points(user):
scores = user.reference.get().to_dict().get('scores', {})
return sum(
score.get('points', score.get('tokens', 0)) - score.get('tokens', 0) for score in scores.values()
)
def levels_done(user, levels):
scores = user.reference.get().to_dict().get('scores', {})
return [scores.get(lev, {}).get("done", False) for lev in levels]
if __name__ == '__main__':
pass
|