88 lines
2.4 KiB
Python
88 lines
2.4 KiB
Python
import os
|
||
import io
|
||
import sys
|
||
import json
|
||
import zipfile
|
||
from flask import Flask, render_template, request, send_file, jsonify
|
||
|
||
# Добавляем корень в sys.path чтобы import drhider работал
|
||
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||
|
||
from drhider import obfuscate_files
|
||
|
||
app = Flask(__name__)
|
||
|
||
VERSION = "1.0.0"
|
||
|
||
MAX_BODY = 200 * 1024 * 1024 # 200 MB
|
||
|
||
|
||
class LLMClient:
|
||
"""LLM-клиент для drhider (обнаружение компаний/ФИО)."""
|
||
|
||
def __init__(self):
|
||
import httpx
|
||
self._httpx = httpx
|
||
|
||
def complete(self, prompt: str) -> str:
|
||
key = os.environ.get("LLM_KEY") or os.environ.get("LLM_API_KEY", "")
|
||
r = self._httpx.post(
|
||
os.environ.get("LLM_URL", "https://api.aillm.ru/v1/chat/completions"),
|
||
json={
|
||
"model": os.environ.get("LLM_MODEL", "gpt-oss-120b"),
|
||
"messages": [{"role": "user", "content": prompt}],
|
||
"max_tokens": 8000,
|
||
"temperature": 0.1,
|
||
},
|
||
headers={
|
||
"Authorization": f"Bearer {key}",
|
||
"Content-Type": "application/json",
|
||
},
|
||
timeout=120,
|
||
)
|
||
r.raise_for_status()
|
||
return r.json()["choices"][0]["message"]["content"]
|
||
|
||
|
||
@app.route("/")
|
||
def index():
|
||
return render_template("index.html", version=VERSION)
|
||
|
||
|
||
@app.route("/health")
|
||
def health():
|
||
return jsonify({"ok": True, "version": VERSION})
|
||
|
||
|
||
@app.route("/api/drhider", methods=["POST"])
|
||
def api_drhider():
|
||
"""Обфускация: multipart/form-data с файлами → ZIP."""
|
||
uploaded = request.files.getlist("files")
|
||
if not uploaded:
|
||
return jsonify({"ok": False, "error": "Нет файлов"}), 400
|
||
|
||
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 = LLMClient()
|
||
zip_data, _csv = obfuscate_files(files, llm_client=llm)
|
||
return send_file(
|
||
io.BytesIO(zip_data),
|
||
mimetype="application/zip",
|
||
as_attachment=True,
|
||
download_name="drhider_output.zip",
|
||
)
|
||
except Exception as e:
|
||
import traceback
|
||
traceback.print_exc()
|
||
return jsonify({"ok": False, "error": str(e)}), 500
|
||
|
||
|
||
|