From e4975fb37ee464877a09d23b9dd3d49b9332dd60 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E2=80=9CNaeel=E2=80=9D?= <tazet@narod.ru>
Date: Mon, 13 Jul 2026 10:40:00 +0400
Subject: [PATCH] =?UTF-8?q?v0.0.5:=20CSV=20=D0=BE=D1=82=D0=B4=D0=B5=D0=BB?=
 =?UTF-8?q?=D1=8C=D0=BD=D0=BE,=20timestamp=20=D0=B8=D0=BC=D0=B5=D0=BD?=
 =?UTF-8?q?=D0=B0,=20=D0=BF=D1=80=D0=BE=D0=B3=D1=80=D0=B5=D1=81=D1=81=20?=
 =?UTF-8?q?=D0=B2=20=D1=82=D0=B0=D0=B1=D0=BB=D0=B8=D1=86=D0=B5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 site/app.py               |   2 +-
 site/routes/api_bp.py     |  32 +++++++--
 site/session.py           |  32 +++++++++
 site/templates/index.html | 141 +++++++++++++++++++++++++++++++++++++-
 4 files changed, 197 insertions(+), 10 deletions(-)

diff --git a/site/app.py b/site/app.py
index 6d2cf36..8ed38c9 100644
--- a/site/app.py
+++ b/site/app.py
@@ -20,7 +20,7 @@ if _sys_path_root not in sys.path:
     sys.path.insert(0, _sys_path_root)
 
 # Версия приложения (меняется при изменениях)
-VERSION = "0.0.4"
+VERSION = "0.0.5"
 
 
 def create_app():
diff --git a/site/routes/api_bp.py b/site/routes/api_bp.py
index a32b301..a00f648 100644
--- a/site/routes/api_bp.py
+++ b/site/routes/api_bp.py
@@ -1,20 +1,23 @@
 """
 Blueprint: API DrHider.
 
-Three endpoints:
+Four endpoints:
 - POST /api/upload      — upload one file -> {session_id}
 - POST /api/process/{sid} — process all session files -> {status:"done"}
-- GET  /api/download/{sid} — download ZIP
+- GET  /api/download/{sid} — download ZIP (with timestamp name)
+- GET  /api/csv/{sid}      — download CSV separately
 """
 
 import io
 import zipfile
 import traceback
+from datetime import datetime
 from flask import Blueprint, request, send_file, jsonify
 
 from drhider import obfuscate_files, LLMClient
 from drhider.builder import build_zip, build_mapping_csv
-from session import create_session, add_file, get_files, store_result, get_result, cleanup, file_count
+from session import (create_session, add_file, get_files, store_result,
+                     get_result, store_csv, get_csv, cleanup, file_count)
 
 api_bp = Blueprint("api", __name__, url_prefix="/api")
 
@@ -64,8 +67,10 @@ def process(sid):
                     elif not name.endswith("/"):
                         all_results.append((name, zf.read(name)))
         csv_str = build_mapping_csv(all_mapping) if all_mapping else ""
-        final_zip = build_zip(all_results, mapping_csv=csv_str)
+        final_zip = build_zip(all_results)  # CSV — отдельно, не в ZIP
         store_result(sid, final_zip)
+        if csv_str:
+            store_csv(sid, csv_str)
         return jsonify({"ok": True, "status": "done", "files": len(all_results)})
     except Exception as e:
         traceback.print_exc()
@@ -78,6 +83,21 @@ def download(sid):
     zip_data = get_result(sid)
     if zip_data is None:
         return jsonify({"ok": False, "error": "Not found"}), 404
-    cleanup(sid)
+    ts = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
     return send_file(io.BytesIO(zip_data), mimetype="application/zip",
-                     as_attachment=True, download_name="drhider_output.zip")
+                     as_attachment=True, download_name=f"drhider_{ts}.zip")
+
+
+@api_bp.route("/csv/<sid>", methods=["GET"])
+def csv_download(sid):
+    """Download CSV separately."""
+    csv_str = get_csv(sid)
+    if csv_str is None:
+        return jsonify({"ok": False, "error": "Not found"}), 404
+    cleanup(sid)
+    ts = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
+    buf = io.BytesIO()
+    buf.write('\ufeff'.encode('utf-8') + csv_str.encode('utf-8'))
+    buf.seek(0)
+    return send_file(buf, mimetype="text/csv",
+                     as_attachment=True, download_name=f"mapping_{ts}.csv")
diff --git a/site/session.py b/site/session.py
index 4b0b7e5..66e1822 100644
--- a/site/session.py
+++ b/site/session.py
@@ -136,3 +136,35 @@ def cleanup(sid: str):
         s = _sessions.pop(sid, None)
         if s and s.get("timer"):
             s["timer"].cancel()
+
+
+def store_csv(sid: str, csv_str: str) -> bool:
+    """Сохранить CSV с таблицей замен.
+
+    Args:
+        sid: Идентификатор сессии
+        csv_str: Строка CSV
+
+    Returns:
+        True если сессия существует, False если нет.
+    """
+    with _lock:
+        s = _sessions.get(sid)
+        if not s:
+            return False
+        s["csv"] = csv_str
+        return True
+
+
+def get_csv(sid: str) -> Optional[str]:
+    """Получить CSV с таблицей замен.
+
+    Args:
+        sid: Идентификатор сессии
+
+    Returns:
+        Строка CSV или None если нет.
+    """
+    with _lock:
+        s = _sessions.get(sid)
+        return s.get("csv") if s else None
diff --git a/site/templates/index.html b/site/templates/index.html
index 2846504..929c465 100644
--- a/site/templates/index.html
+++ b/site/templates/index.html
@@ -3,6 +3,9 @@
 <head>
   <meta charset="UTF-8">
   <meta name="viewport" content="width=device-width, initial-scale=1.0">
+  <meta http-equiv="Cache-Control" content="no-cache, no-store, must-revalidate">
+  <meta http-equiv="Pragma" content="no-cache">
+  <meta http-equiv="Expires" content="0">
   <title>DrHider — обфускация документов</title>
   <style>
     :root {
@@ -13,6 +16,8 @@
       --card: #ffffff;
       --text: #1a1a1a;
       --muted: #6b7280;
+      --red: #ef4444;
+      --green: #22c55e;
     }
     * { box-sizing: border-box; margin: 0; padding: 0; }
     body { font: 14px system-ui, sans-serif; color: var(--text); background: var(--bg); min-height: 100vh; }
@@ -32,6 +37,9 @@
       font-weight: 600; font-size: 14px; border-bottom: 1px solid var(--brand-gray);
     }
     .card-body { padding: 12px; }
+    .sub { font-size: 12px; color: var(--muted); margin-bottom: 10px; line-height: 1.5; }
+    .file-input-wrap { margin-bottom: 10px; }
+    .file-input-wrap input[type="file"] { width: 100%; font-size: 13px; }
     .table-wrap { border: 1px solid var(--brand-gray); border-radius: 8px; overflow: hidden; }
     table { width: 100%; border-collapse: collapse; font-size: 13px; }
     th {
@@ -43,7 +51,15 @@
       padding: 5px 8px; border-right: 1px solid var(--brand-gray);
       border-bottom: 1px solid var(--brand-gray);
     }
+    tr:hover td { background: rgba(37,99,235,.03); }
+    .name-cell { max-width: 340px; overflow: hidden; text-overflow: ellipsis; white-space: nowrap; }
+    .num-cell { text-align: right; white-space: nowrap; }
     .empty-row td { color: var(--muted); text-align: center; padding: 20px; }
+    .remove-btn {
+      cursor: pointer; color: #f87171; background: none; border: none;
+      padding: 2px 4px; font-size: 14px; line-height: 1;
+    }
+    .remove-btn:hover { color: var(--red); }
     .footer-bar {
       margin-top: 10px; display: flex; justify-content: space-between; align-items: center;
       font-size: 12px; color: var(--muted);
@@ -55,6 +71,14 @@
       display: inline-flex; align-items: center; gap: 5px;
     }
     .btn-primary { background: var(--brand-primary); border-color: var(--brand-primary); color: #fff; }
+    .btn-primary:hover { opacity: .9; }
+    .btn:disabled { opacity: .4; cursor: not-allowed; }
+    .status { margin-top: 10px; padding: 8px 12px; border-radius: 8px; font-size: 13px; display: none; }
+    .status.progress { background: #eff6ff; border: 1px solid var(--brand-primary); color: var(--brand-primary); display: block; }
+    .status.done { background: #f0fdf4; border: 1px solid var(--green); color: #16a34a; display: block; }
+    .status.error { background: #fef2f2; border: 1px solid var(--red); color: #991b1b; display: block; }
+    .dl-btns { display: none; gap: 8px; margin-top: 8px; }
+    .dl-btns.show { display: flex; }
   </style>
 </head>
 <body>
@@ -66,6 +90,13 @@
   <div class="card">
     <div class="card-header">Обфускация документов</div>
     <div class="card-body">
+      <p class="sub">
+        Загрузите документы (.docx, .pdf, .txt, .zip) — получите ZIP с обезличенными копиями.<br>
+        CSV с таблицей замен скачивается отдельно.
+      </p>
+      <div class="file-input-wrap">
+        <input type="file" id="fileInput" multiple accept=".docx,.pdf,.txt,.doc,.zip">
+      </div>
       <div class="table-wrap">
         <table>
           <thead>
@@ -76,17 +107,121 @@
               <th style="width:32px;"></th>
             </tr>
           </thead>
-          <tbody>
+          <tbody id="fileList">
             <tr class="empty-row"><td colspan="4">Нет выбранных файлов</td></tr>
           </tbody>
         </table>
       </div>
       <div class="footer-bar">
-        <span>0 файлов</span>
-        <button class="btn btn-primary">🛡️ Обфусцировать</button>
+        <span id="fileCount">0 файлов</span>
+        <button class="btn btn-primary" id="uploadBtn" disabled onclick="uploadFiles()">🛡️ Обфусцировать</button>
+      </div>
+      <div class="status" id="status"></div>
+      <div class="dl-btns" id="dlBtns">
+        <button class="btn btn-primary" id="dlZip" onclick="downloadZip()">📦 Скачать ZIP</button>
+        <button class="btn" id="dlCsv" onclick="downloadCsv()">📋 Скачать CSV</button>
       </div>
     </div>
   </div>
 </div>
+<script>
+const fi = document.getElementById('fileInput');
+const fl = document.getElementById('fileList');
+const fc = document.getElementById('fileCount');
+const ub = document.getElementById('uploadBtn');
+const st = document.getElementById('status');
+const db = document.getElementById('dlBtns');
+let sf = [];
+let currentSid = '';
+
+function fs(b) { return b < 1024 ? b + ' B' : b < 1048576 ? (b / 1024).toFixed(1) + ' KB' : (b / 1048576).toFixed(1) + ' MB'; }
+
+function rr() {
+  if (sf.length === 0) { fl.innerHTML = '<tr class="empty-row"><td colspan="4">Нет выбранных файлов</td></tr>'; }
+  else { fl.innerHTML = sf.map((f, i) => '<tr id="row-' + i + '"><td class="name-cell">' + f.name + '</td><td class="num-cell">' + fs(f.size) + '</td><td class="num-cell" id="st-' + i + '" style="font-size:12px;">—</td><td><button class="remove-btn" onclick="rm(' + i + ')">✕</button></td></tr>').join(''); }
+  fc.textContent = sf.length + ' файлов';
+  ub.disabled = sf.length === 0;
+}
+
+function rm(i) { sf.splice(i, 1); const d = new DataTransfer(); sf.forEach(f => d.items.add(f)); fi.files = d.files; rr(); }
+
+window.addEventListener('load', () => { sf = []; fi.value = ''; rr(); });
+fi.addEventListener('change', () => { sf = Array.from(fi.files); rr(); });
+
+function ss(idx, h) { const e = document.getElementById('st-' + idx); if (e) e.innerHTML = h; }
+
+async function uploadFiles() {
+  if (sf.length === 0) return;
+  ub.disabled = true;
+  db.classList.remove('show');
+  const total = sf.length;
+
+  // Фаза 1: загрузка
+  for (let i = 0; i < total; i++) {
+    const f = sf[i], n = i + 1;
+    ss(i, '<span style="color:#2563eb">⏳ загрузка</span>');
+    st.className = 'status progress';
+    st.textContent = 'Загрузка ' + n + '/' + total + ': ' + f.name;
+    const fd = new FormData();
+    fd.append('files', f, f.name);
+    if (currentSid) fd.append('session', currentSid);
+    try {
+      const resp = await fetch('/api/upload', { method: 'POST', body: fd });
+      const data = await resp.json();
+      if (!data.ok) throw new Error(data.error);
+      currentSid = data.session;
+      ss(i, '<span style="color:#22c55e">✓ загружен</span>');
+    } catch (err) {
+      ss(i, '<span style="color:#ef4444">✗</span>');
+      st.className = 'status error';
+      st.textContent = 'Ошибка загрузки: ' + err.message;
+      ub.disabled = false;
+      return;
+    }
+  }
+
+  // Фаза 2: обработка
+  for (let i = 0; i < total; i++) ss(i, '<span style="color:#2563eb">⏳</span>');
+  st.className = 'status progress';
+  st.textContent = 'Обработка...';
+  const t0 = performance.now();
+  const ptimer = setInterval(() => {
+    const sec = Math.round((performance.now() - t0) / 1000);
+    st.textContent = 'Обработка... ' + sec + 'с';
+  }, 1000);
+
+  try {
+    const resp = await fetch('/api/process/' + currentSid, { method: 'POST' });
+    const data = await resp.json();
+    clearInterval(ptimer);
+    if (!data.ok) throw new Error(data.error);
+    for (let i = 0; i < total; i++) ss(i, '<span style="color:#22c55e">✓</span>');
+    st.className = 'status done';
+    st.textContent = '✅ Обработано ' + data.files + ' файлов за ' + ((performance.now() - t0) / 1000).toFixed(1) + 'с';
+    db.classList.add('show');
+  } catch (err) {
+    clearInterval(ptimer);
+    st.className = 'status error';
+    st.textContent = 'Ошибка: ' + err.message;
+  }
+  ub.disabled = false;
+}
+
+function downloadZip() {
+  if (!currentSid) return;
+  const a = document.createElement('a');
+  a.href = '/api/download/' + currentSid;
+  a.download = 'drhider.zip';
+  a.click();
+}
+
+function downloadCsv() {
+  if (!currentSid) return;
+  const a = document.createElement('a');
+  a.href = '/api/csv/' + currentSid;
+  a.download = 'mapping.csv';
+  a.click();
+}
+</script>
 </body>
 </html>