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}")