import json, random
import numpy as np, pandas as pd
from sklearn.metrics import roc_auc_score
from app.database import SessionLocal
from app.repository import list_objects
from app.feature_engineering import FEATURE_COLUMNS, combine_features
from app.graph_features import build_graph, pair_graph_features
from app.ml import train_models

def run_training():
    db = SessionLocal()
    try:
        objs = list_objects(db, 5000)
        objects = [{"object_id": o.object_id, "object_type": o.object_type, "mean_motion": o.mean_motion, "inclination": o.inclination, "eccentricity": o.eccentricity, "raan": o.raan, "bstar": o.bstar, "launch_year": o.launch_year} for o in objs]
    finally:
        db.close()
    pairs = [tuple(random.sample(objects, 2)) for _ in range(4000)]
    g = build_graph([(a["object_id"], b["object_id"]) for a, b in pairs[:1000]])
    rows = []
    raw_scores = []
    for a, b in pairs:
        trend = {"recurrence_count": float(random.choice([0,1,2,3,4])), "trend_delta_score": float(random.uniform(-0.1, 0.3)), "score_volatility_proxy": float(random.uniform(0, 0.2))}
        f = combine_features(a, b, trend, pair_graph_features(g, a["object_id"], b["object_id"]))
        score = 0.30*f["close_approach_proxy"] + 0.16*f["same_shell"] + 0.10*min(1.0,f["shell_density_proxy"]/12.0) + 0.10*min(1.0,f["graph_local_density"]*2.0) + 0.09*min(1.0,f["graph_jaccard"]) + 0.10*min(1.0,f["recurrence_count"]/5.0) + 0.08*max(0.0,f["trend_delta_score"]) + np.random.normal(0,0.05)
        y = 1 if score > 0.48 else 0
        rows.append({**f, "label": y})
        raw_scores.append(score)
    df = pd.DataFrame(rows)
    path = train_models(df[FEATURE_COLUMNS].values, df["label"].values)
    auc = float(roc_auc_score(df["label"].values, np.array(raw_scores)))
    return path, len(df), {"pseudo_auc": round(auc, 4), "rows": int(len(df))}

if __name__ == "__main__":
    p, r, m = run_training()
    print(json.dumps({"model_path": p, "rows": r, "metrics": m}, indent=2))