// Hugging Face Dataset Server API Types

export interface DatasetRow {
  [key: string]: any;
}

export interface DatasetFeature {
  _type: string;
  [key: string]: any;
}

export interface DatasetFeatures {
  [key: string]: DatasetFeature;
}

export interface DatasetRowsResponse {
  features: DatasetFeature[];
  rows: {
    row_idx: number;
    row: DatasetRow;
    truncated_cells: string[];
  }[];
  num_rows_total: number;
  num_rows_per_page: number;
  partial: boolean;
}

export interface DatasetFirstRowsResponse {
  features: DatasetFeature[];
  rows: {
    row_idx: number;
    row: DatasetRow;
    truncated_cells: string[];
  }[];
  truncated: boolean;
}

export interface DatasetSplit {
  dataset: string;
  config: string;
  split: string;
}

export interface DatasetSplitsResponse {
  splits: DatasetSplit[];
  pending: any[];
  failed: any[];
  partial: boolean;
}

export interface DatasetError {
  error: string;
  message?: string;
}

export interface DatasetConfig {
  dataset: string;
  token?: string;
  config?: string;
  split?: string;
}

export interface TermDefinition {
  term: string;
  definition: string;
}

export interface JurisdictionData {
  jurisdiction: string;
  terms: TermDefinition[];
}

export interface ComparisonState {
  jurisdiction1: string;
  jurisdiction2: string;
  data1: TermDefinition[];
  data2: TermDefinition[];
  selectedTerm1: string | null;
  selectedTerm2: string | null;
}

export interface Comparison {
  term_j1: string;
  term_j2: string;
  category: string;
  subcategory: string;
  similarity_score: number;
  reasoning: string;
  weight: number;
  weighted_similarity_score: number;
}

export interface EquivalencyScore {
  term_j1: string;
  term_j2: string;
  aggregated_similarity_score: number;
  comparative_law_note: string;
  comparisons?: Comparison[];
}

export interface GradingTemplateEntry {
  term: string;
  category: string;
  subcategory: string;
  question: string;
  answer: string;
}