File size: 5,367 Bytes
a1aef88
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
import { useState, useCallback } from 'react';
import { Upload, FileSpreadsheet, AlertTriangle } from 'lucide-react';
import { api } from '../../api/client';
import { useStore } from '../../store/useStore';
import { Panel } from '../common/Panel';
import { LoadingSpinner } from '../common/LoadingSpinner';
import { DataTable } from './DataTable';
import { FilterPanel } from './FilterPanel';
import type { ColumnStat, DataResponse } from '../../types';

export function DataExplorer() {
  const { data, columnStats, setData, dataLoaded } = useStore();
  const [loading, setLoading] = useState(false);
  const [error, setError] = useState<string | null>(null);
  const [activeData, setActiveData] = useState<DataResponse | null>(null);
  const [activeStats, setActiveStats] = useState<ColumnStat[]>([]);

  const displayData = activeData ?? data;
  const displayStats = activeStats.length > 0 ? activeStats : columnStats;


  const handleUpload = async (e: React.ChangeEvent<HTMLInputElement>) => {
    const file = e.target.files?.[0];
    if (!file) return;
    setLoading(true);
    setError(null);
    try {
      const res = await api.uploadFile(file);
      setData(res.data, res.column_stats);
      setActiveData(null);
      setActiveStats([]);
    } catch (err: unknown) {
      setError(err instanceof Error ? err.message : 'Upload failed');
    } finally {
      setLoading(false);
    }
  };

  const handleFilter = useCallback(
    async (
      filters: {
        column: string;
        type: string;
        values?: string[];
        min_val?: number;
        max_val?: number;
        pattern?: string;
      }[]
    ) => {
      if (filters.length === 0) {
        setActiveData(null);
        setActiveStats([]);
        setError(null);
        return;
      }
      setLoading(true);
      setError(null);
      try {
        const res = await api.filterData(filters);
        setActiveData(res.data);
        setActiveStats(res.column_stats);
      } catch (err: unknown) {
        setError(err instanceof Error ? err.message : 'Filtering failed');
      } finally {
        setLoading(false);
      }
    },
    []
  );

  return (
    <div className="space-y-4">
      <div className="flex flex-wrap items-center gap-3">


        <label className="flex cursor-pointer items-center gap-2 rounded-md border border-border bg-surface px-4 py-2 text-sm text-text-primary transition-colors hover:border-purple hover:bg-elevated">
          <Upload className="h-4 w-4 text-purple" />
          Upload Files
          <input type="file" accept=".csv,.xlsx,.xls,.json" onChange={handleUpload} className="hidden" />
        </label>

        {dataLoaded && displayData && (
          <div className="ml-auto flex items-center gap-3 text-xs text-text-muted">
            <FileSpreadsheet className="h-4 w-4" />
            <span>
              {displayData.returned_rows.toLocaleString()} / {displayData.total_rows.toLocaleString()} rows
            </span>
            <span>{displayData.columns.length} columns</span>
          </div>
        )}
      </div>

      {error && (
        <div className="flex items-center gap-2 rounded-md border border-danger/30 bg-danger/10 px-4 py-2.5 text-sm text-danger">
          <AlertTriangle className="h-4 w-4" /> {error}
        </div>
      )}

      {loading && <LoadingSpinner text="Processing data..." />}

      {/* Filters */}
      {dataLoaded && displayStats.length > 0 && (
        <FilterPanel columns={displayStats} onApply={handleFilter} />
      )}

      {/* Column Stats */}
      {dataLoaded && displayStats.length > 0 && (
        <Panel title="Column Statistics" subtitle={`${displayStats.length} columns`}>
          <div className="grid grid-cols-2 gap-2 sm:grid-cols-3 lg:grid-cols-4 xl:grid-cols-6">
            {displayStats.map((col) => (
              <div
                key={col.name}
                className="rounded border border-border/50 bg-raised p-2.5"
              >
                <p className="truncate text-xs font-medium text-text-primary" title={col.name}>
                  {col.name}
                </p>
                <p className="mt-0.5 text-[10px] text-text-muted">
                  {col.dtype} | {col.unique} unique
                </p>
                {col.top_values && col.top_values.length > 0 && (
                  <div className="mt-1.5 space-y-0.5">
                    {col.top_values.slice(0, 3).map((tv) => (
                      <div key={tv.value} className="flex items-center gap-1">
                        <div
                          className="h-1 rounded-full bg-accent/60"
                          style={{
                            width: `${Math.min(100, (tv.count / (col.non_null || 1)) * 100)}%`,
                          }}
                        />
                        <span className="whitespace-nowrap text-[9px] text-text-muted">
                          {tv.value.slice(0, 15)}
                        </span>
                      </div>
                    ))}
                  </div>
                )}
              </div>
            ))}
          </div>
        </Panel>
      )}

      {/* Data Table */}
      {dataLoaded && displayData && (
        <Panel title="Data View" noPad>
          <DataTable data={displayData} maxHeight="calc(100vh - 420px)" />
        </Panel>
      )}
    </div>
  );
}