Denis Krivochenko
115 lines
6.2 KiB
Python
115 lines
6.2 KiB
Python
# /opt/trueconf_bot/photo_bot/ai_validator.py
|
|
# Тестовый мод фото-бота: прямая проверка фото ИИ (llama.cpp).
|
|
# Endpoint, модель и промпт — 1:1 из SibCemProfile/PhotoApp/photo_validator.py.
|
|
# Фото никуда не отправляется: только в ИИ, ответ возвращается пользователю.
|
|
import base64
|
|
import io
|
|
import json
|
|
import logging
|
|
import urllib.request
|
|
from PIL import Image
|
|
|
|
import config.config as config
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# Тег рассуждений thinking-моделей (конкатенация без литерала)
|
|
THINK_CLOSE_TAG = '</' + 'think>'
|
|
|
|
|
|
def _build_prompt() -> str:
|
|
"""Промпт проверки фото (без подсказки MediaPipe — в тест-моде геометрия не считается)."""
|
|
cv_hint = 'ℹ️ ПОДСКАЗКА ОТ ALGORITHM: Геометрическая проверка в тестовом режиме не выполняется, проверяй визуально.'
|
|
|
|
return ('''Ты — автоматический эксперт пропускной системы. Проверь фотографию сотрудника на соответствие регламенту.
|
|
|
|
%s
|
|
|
|
ЗАПРЕЩЕНО выводить внутренние рассуждения и теги размышлений. Сразу пиши финальный ответ.
|
|
|
|
БЛОК 0. КРИТИЧЕСКИЕ СТОП-ФАКТОРЫ:
|
|
1. Не реальное фото: ИИ-генерация, 3D-арт, цифровой рисунок, мультфильм, стилизованный портрет, сильная обработка фильтрами.
|
|
2. Графика и текст: Любые надписи, водяные знаки, рамки, декоративный фон с буквами, графическими элементами, коллажи.
|
|
3. Костюмы и атрибуты: Исторические костюмы, короны, маски, каски (кроме строительных).
|
|
4. Посторонние предметы: Оружие, предметы перед лицом.
|
|
5. Неправильный ракурс и взгляд: Строгий боковой профиль, затылок, человек смотрит совсем в сторону.
|
|
6. Гримасы и мимика: Гримасы, смешные рожи, высунутый язык, вытянутые щёки, скривленные губы, смех/улыбка с закрытым ртом, любой невынужденный и неспокойный вид. Выражение лица должно быть нейтральным или слегка спокойным.
|
|
|
|
БЛОК 1. МАСШТАБ И КАДРИРОВАНИЕ:
|
|
- подбородок НИЖЕ середины кадра → тест пройден.
|
|
- «Не прошел», если видны талия, живот, подмышки, одежда ниже линии груди.
|
|
|
|
ИТОГОВЫЙ ВЕРДИКТ:
|
|
- «Прошел» — нормальные портретные фотографии.
|
|
- «Не прошел» — только явные нарушения.
|
|
- «На модерацию» — если фото качественное, но масштаб на грани.
|
|
|
|
ФОРМАТ ОТВЕТА — СТРОГО 2 строки:
|
|
Вердикт: [Прошел / Не прошел / На модерацию]
|
|
Причина: [1–2 предложения с указанием конкретной причины]
|
|
''') % cv_hint
|
|
|
|
|
|
def validate_photo_with_ai(file_bytes: bytes) -> dict:
|
|
"""
|
|
Прямая проверка фото ИИ (тестовый мод).
|
|
|
|
Возвращает {'ok': bool, 'verdict': str, 'reason': str, 'error': str}
|
|
"""
|
|
try:
|
|
# Сжатие как в photo_validator.py (PIL, без сторонних DLL)
|
|
with Image.open(io.BytesIO(file_bytes)) as img:
|
|
img = img.convert('RGB')
|
|
img.thumbnail((768, 768))
|
|
buffer = io.BytesIO()
|
|
img.save(buffer, format='JPEG', quality=85)
|
|
b64_img = base64.b64encode(buffer.getvalue()).decode('utf-8')
|
|
|
|
payload = {
|
|
'model': config.PHOTO_TEST_AI_MODEL,
|
|
'messages': [{
|
|
'role': 'user',
|
|
'content': [
|
|
{'type': 'text', 'text': _build_prompt()},
|
|
{'type': 'image_url', 'image_url': {'url': 'data:image/jpeg;base64,' + b64_img}}
|
|
]
|
|
}],
|
|
'temperature': 0.1,
|
|
'max_tokens': 1000
|
|
}
|
|
|
|
req = urllib.request.Request(
|
|
config.PHOTO_TEST_AI_URL,
|
|
data=json.dumps(payload).encode('utf-8'),
|
|
headers={'Content-Type': 'application/json'}
|
|
)
|
|
|
|
with urllib.request.urlopen(req, timeout=config.PHOTO_TEST_AI_TIMEOUT) as resp:
|
|
res = json.loads(resp.read().decode('utf-8'))
|
|
|
|
full_content = (res['choices'][0]['message'].get('content') or '').strip()
|
|
clean_ans = full_content.split(THINK_CLOSE_TAG)[-1].strip() if THINK_CLOSE_TAG in full_content else full_content.strip()
|
|
|
|
text_lower = clean_ans.lower()
|
|
if 'вердикт: прошел' in text_lower or ('прошел' in text_lower and 'не прошел' not in text_lower and 'модерац' not in text_lower):
|
|
verdict = 'Прошел'
|
|
elif 'на модерацию' in text_lower or 'модерац' in text_lower:
|
|
verdict = 'На модерацию'
|
|
else:
|
|
verdict = 'Не прошел'
|
|
|
|
reason = ''
|
|
for line in clean_ans.splitlines():
|
|
if line.strip().lower().startswith('причина:'):
|
|
reason = line.split(':', 1)[1].strip()
|
|
break
|
|
if not reason:
|
|
reason = clean_ans.replace('\n', ' ').strip()
|
|
|
|
logger.info(f"[TEST MODE] ИИ-вердикт: {verdict} | {reason}")
|
|
return {'ok': True, 'verdict': verdict, 'reason': reason, 'error': ''}
|
|
|
|
except Exception as e:
|
|
logger.error(f"[TEST MODE] Ошибка обращения к ИИ: {e}")
|
|
return {'ok': False, 'verdict': '', 'reason': '', 'error': str(e)}
|