Фаза 2: site/app.py → blueprint'ы (main, health, api), старый drhider.py удалён

site/routes/api_bp.py

@@ -0,0 +1,73 @@
+"""
+Blueprint: API обфускации (POST /api/drhider).
+
+Принимает multipart/form-data с файлами, возвращает ZIP-архив
+с обфусцированными документами.
+"""
+
+import io
+import traceback
+from flask import Blueprint, request, send_file, jsonify
+
+from drhider import obfuscate_files, LLMClient
+
+# ═══════════════════════════════════════════════════════════════════════════
+# Blueprint: API DrHider
+# ═══════════════════════════════════════════════════════════════════════════
+
+api_bp = Blueprint("api", __name__, url_prefix="/api")
+
+# Максимальный размер тела запроса: 200 MB
+MAX_BODY = 200 * 1024 * 1024
+
+
+@api_bp.route("/drhider", methods=["POST"])
+def drhider():
+    """Обфускация документов.
+
+    Принимает multipart/form-data с полем 'files' (один или несколько файлов).
+    Возвращает ZIP-архив с обфусцированными документами + mapping.csv.
+
+    Поддерживаемые форматы:
+        .docx, .pdf, .txt, .doc (бинарный — без изменений)
+        .zip (распаковывается, содержимое обфусцируется)
+
+    Returns:
+        200: ZIP-архив (application/zip)
+        400: {"ok": false, "error": "..."}
+        500: {"ok": false, "error": "..."}
+    """
+    # ── Получаем файлы из запроса ──
+    uploaded = request.files.getlist("files")
+    if not uploaded:
+        return jsonify({"ok": False, "error": "Нет файлов"}), 400
+
+    # Формируем список для obfuscate_files: [(name, bytes, mimetype), ...]
+    files = []
+    for f in uploaded:
+        if f.filename:
+            files.append((f.filename, f.read(), f.mimetype or ""))
+
+    if not files:
+        return jsonify({"ok": False, "error": "Нет файлов"}), 400
+
+    # ── Обфускация ──
+    try:
+        # Создаём LLM-клиент для NER-сканирования
+        llm = LLMClient()
+
+        # Вызываем ядро обфускации
+        zip_data, _csv = obfuscate_files(files, llm_client=llm)
+
+        # Отдаём ZIP-архив
+        return send_file(
+            io.BytesIO(zip_data),
+            mimetype="application/zip",
+            as_attachment=True,
+            download_name="drhider_output.zip",
+        )
+
+    except Exception as e:
+        # Логируем полный трейсбек на сервере
+        traceback.print_exc()
+        return jsonify({"ok": False, "error": str(e)}), 500