import { useCallback, useState } from 'react';

export function useExtractionResults(extractionApi) {
  const [rows, setRows] = useState([]);
  const [loading, setLoading] = useState(false);
  const [error, setError] = useState(null);

  const load = useCallback(async ({ sessionId, modelId }) => {
    if (!extractionApi) return;
    setLoading(true);
    setError(null);
    try {
      const data = await extractionApi.listResultsBySessionModel({ sessionId, modelId });
      setRows(Array.isArray(data) ? data : []);
    } catch (e) {
      setError(e?.message ?? 'Erro ao carregar resultados');
    } finally {
      setLoading(false);
    }
  }, [extractionApi]);

  return { rows, loading, error, load };
}