import { ApiResponse } from "@/types/api";
import { FrameDataResponse } from "@/features/visualization/types";
import { DifferenceMapData } from "@/features/comparison/types";

const BASE_URL = process.env.NEXT_PUBLIC_API_URL || "https://sid385-fill-the-frames.hf.space/api/v1";

export interface ValidationAlignmentRequest {
  generated_file_id: string;
  ground_truth_file_id: string;
  variable?: string;
}

export interface ValidationAlignmentResponse {
  generated_file_id: string;
  ground_truth_file_id: string;
  aligned_generated: FrameDataResponse;
  aligned_ground_truth: FrameDataResponse;
  difference_map: DifferenceMapData;
  dimensions: number[];
  metadata: Record<string, unknown>;
}

export interface MetricsResponse {
  ssim: number;
  psnr: number;
  mse: number;
  fsim: number | null;
  issm: number | null;
  quality_score: number;
  summary: string;
}

export const validationClient = {
  compareMetrics: async (request: ValidationAlignmentRequest): Promise<MetricsResponse> => {
    const searchParams = new URLSearchParams();
    searchParams.append("generated_file_id", request.generated_file_id);
    searchParams.append("truth_file_id", request.ground_truth_file_id);
    if (request.variable) searchParams.append("variable", request.variable);

    const response = await fetch(`${BASE_URL}/metrics/compare?${searchParams.toString()}`, {
      method: 'GET',
    });

    if (!response.ok) {
      throw new Error(`Failed to compute metrics with status ${response.status}`);
    }

    const json = await response.json();
    return json.data as MetricsResponse; // Because backend returns ApiResponse wrapper
  }
};