ReasMe / api /apis.py
MSNP1381
api up
316425b
raw
history blame
1.44 kB
import json
import firebase_admin
from firebase_admin import credentials
from firebase_admin import firestore
import os
# Initialize Firebase
cert_dict=json.loads(os.getenv('google_json'))
cred = credentials.Certificate(cert_dict)
firebase_admin.initialize_app(cred)
# Initialize Firestore DB
db = firestore.client()
def get_question_by_id(question_id):
doc_ref = db.collection('questions').document(question_id)
doc = doc_ref.get()
if doc.exists:
return doc.to_dict()
else:
return None
# Example usage
question_id = "Mercury_7090615"
question_data = get_question_by_id(question_id)
if question_data:
print(f"Question data for {question_id}: {question_data}")
else:
print(f"No data found for question ID: {question_id}")
def get_question_ids_with_correctness():
questions_ref = db.collection('questions')
docs = questions_ref.stream()
results = []
for doc in docs:
data = doc.to_dict()
question_id = data['original_question']['question_id']
correct_answer = data['original_question']['answer']
generated_answer = data['generated_result']['answer_key_vale']
correctness = "βœ…" if correct_answer == generated_answer else 'πŸ“›'
results.append(f"{question_id} {correctness}")
return results
# Example usage
correctness_list = get_question_ids_with_correctness()
for result in correctness_list:
print(result)