File size: 6,953 Bytes
2c0e275
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
import React, { useState, useEffect } from "react";
import { DocumentChunk } from "../contracts/tool_outputs.js";
import { FileText, Search, ExternalLink, RefreshCw } from "lucide-react";
import { useLanguage } from "../context/LanguageContext.js";
import { getBackendUrl } from "../utils/api.js";

const COUNTRIES = ["SAU", "ARE", "EGY", "USA", "DEU", "JPN", "BRA", "CHN"];

export const DocumentLibraryView: React.FC = () => {
  const { language, t } = useLanguage();
  const isAr = language === "ar";
  const [query, setQuery] = useState(isAr ? "الميزانية العامة والدين السيادي" : "sovereign debt budget");
  const [debouncedQuery, setDebouncedQuery] = useState(query);
  const [selectedCountry, setSelectedCountry] = useState<string>("SAU");
  const [alpha, setAlpha] = useState(0.6);
  const [documents, setDocuments] = useState<DocumentChunk[]>([]);
  const [loading, setLoading] = useState(false);
  const [error, setError] = useState("");

  useEffect(() => {
    const id = setTimeout(() => setDebouncedQuery(query), 500);
    return () => clearTimeout(id);
  }, [query]);

  useEffect(() => {
    if (!debouncedQuery.trim()) return;
    const controller = new AbortController();
    setLoading(true);
    setError("");
    const params = new URLSearchParams({ query: debouncedQuery, alpha: String(alpha), country: selectedCountry });
    fetch(`${getBackendUrl()}/api/rag/documents?${params}`, { signal: controller.signal })
      .then(async (res) => {
        const json = await res.json().catch(() => ({}));
        if (!res.ok) throw new Error(json.error || `HTTP ${res.status}`);
        setDocuments(Array.isArray(json) ? json : []);
      })
      .catch((err: Error) => {
        if (err.name !== "AbortError") setError(err.message);
      })
      .finally(() => setLoading(false));
    return () => controller.abort();
  }, [debouncedQuery, selectedCountry, alpha]);

  return (
    <div className="space-y-6 text-slate-100">
      <div className="bg-slate-900 border border-slate-800 rounded-xl p-6 shadow-xl">
        <div className="flex items-center gap-3 mb-4 border-b border-slate-800 pb-3">
          <div className="w-8 h-8 rounded-lg bg-purple-500/20 text-purple-400 flex items-center justify-center shrink-0">
            <FileText className="w-5 h-5" />
          </div>
          <div>
            <h1 className="text-lg font-bold text-white font-display">{t.documents.title}</h1>
            <p className="text-xs text-slate-400">{t.documents.subtitle}</p>
          </div>
        </div>

        <div className="grid grid-cols-1 md:grid-cols-12 gap-4 mb-6 p-4 bg-slate-950/80 rounded-xl border border-slate-800">
          <div className="md:col-span-6">
            <label className="text-[10px] font-mono text-slate-400 block mb-1">{isAr ? "عبارة البحث" : "Search query"}</label>
            <div className="relative">
              <Search className="w-4 h-4 text-slate-500 absolute start-3 top-2.5" />
              <input
                type="text"
                value={query}
                maxLength={1000}
                onChange={(e) => setQuery(e.target.value)}
                placeholder={t.documents.searchPlaceholder}
                className="w-full bg-slate-900 border border-slate-800 rounded-lg ps-9 pe-3 py-1.5 text-xs text-slate-200 focus:outline-none focus:border-purple-500"
              />
            </div>
          </div>

          <div className="md:col-span-3">
            <label className="text-[10px] font-mono text-slate-400 block mb-1">{isAr ? "الدولة" : "Country"}</label>
            <select
              value={selectedCountry}
              onChange={(e) => setSelectedCountry(e.target.value)}
              className="w-full bg-slate-900 border border-slate-800 rounded-lg px-3 py-1.5 text-xs font-mono text-slate-200"
            >
              {COUNTRIES.map((c) => (
                <option key={c} value={c}>{c}</option>
              ))}
            </select>
          </div>

          <div className="md:col-span-3 flex flex-col justify-center">
            <div className="flex items-center justify-between text-[10px] font-mono text-slate-400 mb-1">
              <span>{isAr ? "وزن العنوان مقابل النص" : "Title vs body weight"}</span>
              <span className="text-purple-300 font-bold">{alpha.toFixed(1)}</span>
            </div>
            <input
              type="range"
              min={0}
              max={1}
              step={0.1}
              value={alpha}
              onChange={(e) => setAlpha(parseFloat(e.target.value))}
              className="w-full accent-purple-400 cursor-pointer"
            />
          </div>
        </div>

        {loading && (
          <div className="flex items-center gap-2 text-xs font-mono text-slate-400 mb-3">
            <RefreshCw className="w-3.5 h-3.5 animate-spin" /> {isAr ? "جارٍ البحث..." : "Searching..."}
          </div>
        )}
        {error && <div className="text-xs text-rose-400 mb-3">{error}</div>}
        {!loading && !error && documents.length === 0 && (
          <div className="p-8 text-center text-xs text-slate-500 bg-slate-950/60 rounded-xl border border-slate-800">{t.documents.noDocsFound}</div>
        )}

        <div className="space-y-3">
          {documents.map((doc) => (
            <a
              key={doc.chunk_id}
              href={doc.url || undefined}
              target="_blank"
              rel="noopener noreferrer"
              className="block p-4 bg-slate-950/70 hover:bg-slate-800/60 border border-slate-800 hover:border-purple-500/50 rounded-xl transition-all group"
            >
              <div className="flex flex-wrap items-center justify-between gap-2 mb-2">
                <div className="flex flex-wrap items-center gap-2">
                  <span className="px-2 py-0.5 bg-purple-950 text-purple-300 border border-purple-800/80 text-[10px] font-mono font-bold rounded">
                    [{doc.chunk_id}]
                  </span>
                  <span className="text-xs text-slate-400">{doc.source_authority}</span>
                  <span className="text-[11px] font-mono text-slate-500">
                    {new Date(doc.publication_timestamp).toLocaleDateString(isAr ? "ar-SA" : "en-US")}
                  </span>
                </div>
                <div className="text-xs font-mono text-purple-400 flex items-center gap-1">
                  <span>{isAr ? "الصلة" : "Relevance"}: {doc.score ?? 0}</span>
                  {doc.url && <ExternalLink className="w-3 h-3" />}
                </div>
              </div>
              <h3 className="text-sm font-bold text-slate-200 group-hover:text-purple-300 transition-colors mb-1">{doc.title}</h3>
              <p className="text-xs text-slate-400 line-clamp-2 leading-relaxed">{doc.content}</p>
            </a>
          ))}
        </div>
      </div>
    </div>
  );
};