package ml

import (
	"fmt"
	"os"
	"path/filepath"

	"forge.lthn.ai/core/go/pkg/cli"
	"forge.lthn.ai/core/go-ai/ml"
)

var (
	approveOutput    string
	approveThreshold float64
)

var approveCmd = &cli.Command{
	Use:   "approve",
	Short: "Filter scored expansions into training JSONL",
	Long:  "Filters scored expansion responses by quality threshold and exports approved ones as chat-format training JSONL.",
	RunE:  runApprove,
}

func init() {
	approveCmd.Flags().StringVar(&approveOutput, "output", "", "Output JSONL file (defaults to expansion-approved.jsonl in db dir)")
	approveCmd.Flags().Float64Var(&approveThreshold, "threshold", 6.0, "Min judge average to approve")
}

func runApprove(cmd *cli.Command, args []string) error {
	path := dbPath
	if path == "" {
		path = os.Getenv("LEM_DB")
	}
	if path == "" {
		return fmt.Errorf("--db or LEM_DB required")
	}

	output := approveOutput
	if output == "" {
		output = filepath.Join(filepath.Dir(path), "expansion-approved.jsonl")
	}

	db, err := ml.OpenDB(path)
	if err != nil {
		return fmt.Errorf("open db: %w", err)
	}
	defer db.Close()

	return ml.ApproveExpansions(db, ml.ApproveConfig{
		Output:    output,
		Threshold: approveThreshold,
	}, cmd.OutOrStdout())
}