EvoTransformer-v2.1 / logger.py
HemanM's picture
Update logger.py
b39744e verified
raw
history blame
782 Bytes
import json
import firebase_admin
from firebase_admin import credentials, firestore
from datetime import datetime
# Initialize Firebase app once
if not firebase_admin._apps:
cred = credentials.Certificate("firebase_key.json")
firebase_admin.initialize_app(cred)
db = firestore.client()
def log_user_feedback(goal, sol1, sol2, winner):
formatted_winner = "1" if "1" in winner else "2" # Critical fix
doc = {
"timestamp": datetime.utcnow().isoformat(),
"goal": goal,
"solution_1": sol1,
"solution_2": sol2,
"winner": formatted_winner, # βœ… Needed by training and plots
"user_feedback": "from app.py"
}
db.collection("evo_feedback").add(doc)
print(f"πŸ“₯ Logged feedback: Solution {formatted_winner}")