текст из pdf распознается, все работает, добавлены все функции из файла работы с базой данных db_cli.py. Готовность к оптимизации интерфейса.

llm/db_tools.py

@@ -1,16 +1,101 @@
+import json
 import sqlite3
 import logging
 from typing import List, Dict, Any, Optional
+from datetime import datetime, timedelta
 
-# Настройка логирования
 logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(levelname)s] %(message)s")
 logger = logging.getLogger("DB_TOOLS")
 
 DB_PATH = "/home/puh/scud_orion_ai_v2/data/scud_orion_ai.db"
 
+def db_get_current_server_time() -> Dict[str, Any]:
+    now = datetime.now()
+    days_ru = ["понедельник", "вторник", "среда", "четверг", "пятница", "суббота", "воскресенье"]
+    return {
+        "current_date": now.strftime("%d.%m.%Y"),
+        "current_time": now.strftime("%H:%M:%S"),
+        "day_of_week": days_ru[now.weekday()],
+        "iso_date": now.strftime("%Y-%m-%d")
+    }
+
+def smart_parse_date(date_str: Optional[str], original_user_message: str = "") -> Optional[str]:
+    """
+    Дата уже точно подготовлена моделью на основе системного календаря.
+    Возвращаем date_str без повторной тяжелой фильтрации.
+    """
+    return date_str
+
+def db_save_chat_message(session_id: str, role: str, content: str):
+    if not content:
+        return
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    cursor.execute("""
+        INSERT INTO chat_messages (session_id, role, content, created_at)
+        VALUES (?, ?, ?, CURRENT_TIMESTAMP)
+    """, (session_id, role, content))
+    conn.commit()
+    conn.close()
+
+def db_get_chat_history(session_id: str, limit: int = 20) -> List[Dict[str, Any]]:
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    cursor.execute("""
+        SELECT role, content FROM chat_messages 
+        WHERE session_id = ? 
+        ORDER BY id DESC LIMIT ?
+    """, (session_id, limit))
+    rows = cursor.fetchall()
+    conn.close()
+    return [{"role": r["role"], "content": r["content"]} for r in reversed(rows)]
+
+def db_get_snapshots(session_id: str = "web_session_main", date_str: Optional[str] = None, original_user_message: str = "") -> Dict[str, Any]:
+    date_str = smart_parse_date(date_str, original_user_message)
+    
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    
+    query = """
+        SELECT snapshot_id, log_date, snapshot_time, COUNT(*) as record_count 
+        FROM scud_logs 
+    """
+    params = []
+    
+    if date_str:
+        # Приводим дату ДД.ММ.ГГГГ к ISO YYYY-MM-DD
+        iso_date = date_str
+        if "." in date_str:
+            parts = date_str.split(".")
+            if len(parts) == 3:
+                iso_date = f"{parts[2]}-{parts[1]}-{parts[0]}"
+        
+        # Строгий поиск: ищем совпадение строго по log_date или началу snapshot_time/created_at
+        query += " WHERE log_date = ? OR log_date = ? OR snapshot_time LIKE ? OR created_at LIKE ? "
+        params.extend([date_str, iso_date, f"{iso_date}%", f"{iso_date}%"])
+        
+    query += " GROUP BY snapshot_id ORDER BY id DESC LIMIT 20"
+    
+    cursor.execute(query, params)
+    rows = cursor.fetchall()
+    snapshots = [dict(r) for r in rows]
+    
+    result_data = {
+        "query_date": date_str or "все",
+        "snapshots_count": len(snapshots),
+        "snapshots": snapshots
+    }
+    
+    db_set_session_state(
+        session_id=session_id, 
+        state_type="SNAPSHOTS_VIEW", 
+        data=json.dumps(result_data, ensure_ascii=False)
+    )
+    
+    conn.close()
+    return result_data
+
 def get_db_connection():
-    # Увеличиваем таймаут до 30 секунд, чтобы соединения ожидали завершения соседних транзакций,
-    # а также включаем WAL-режим для безопасного параллельного чтения и записи.
     conn = sqlite3.connect(DB_PATH, timeout=30.0)
     conn.row_factory = sqlite3.Row
     conn.execute("PRAGMA journal_mode = WAL;")
@@ -18,17 +103,14 @@ def get_db_connection():
     return conn
 
 def normalize_task_id(task_id_input: str) -> str:
-    """Преобразует 6, '6', 'task-6' в красивый формат TASK-06 или TASK-006"""
     if not task_id_input:
         return ""
     clean_id = str(task_id_input).strip().upper().replace("TASK-", "").replace("TASK", "")
     if clean_id.isdigit():
         num = int(clean_id)
-        return f"TASK-{num:02d}" if num < 100 else f"TASK-{num:03d}"
+        return f"TASK-{(num):02d}" if num < 100 else f"TASK-{(num):03d}"
     return f"TASK-{clean_id}"
 
-# === ЗАДАЧИ С ФИЛЬТРАЦИЕЙ ПО USER_ID ===
-
 def db_get_tasks(user_id: int) -> List[Dict[str, Any]]:
     conn = get_db_connection()
     cursor = conn.cursor()
@@ -105,8 +187,6 @@ def db_delete_task(user_id: int, task_id: str) -> Dict[str, Any]:
     conn.close()
     return {"status": "success", "message": f"Задача {formatted_id} удалена"}
 
-# === ОБЩИЕ СИСТЕМНЫЕ РЕСУРСЫ ===
-
 def db_get_active_system_prompt() -> str:
     conn = get_db_connection()
     cursor = conn.cursor()
@@ -121,18 +201,15 @@ def db_add_system_prompt(name: str, prompt_text: str) -> Dict[str, Any]:
             cursor = conn.cursor()
             cursor.execute("BEGIN IMMEDIATE;")
             
-            # Проверяем, существует ли уже промпт с таким именем
             cursor.execute("SELECT id FROM system_prompts WHERE name = ?", (name,))
             existing = cursor.fetchone()
             
             if existing:
-                # Обновляем существующий активный промпт
                 cursor.execute(
                     "UPDATE system_prompts SET prompt_text = ?, is_active = 1, updated_at = CURRENT_TIMESTAMP WHERE name = ?",
                     (prompt_text, name)
                 )
             else:
-                # Вставляем новый, если его не было
                 cursor.execute(
                     "INSERT INTO system_prompts (name, prompt_text, is_active) VALUES (?, ?, 1)",
                     (name, prompt_text)
@@ -146,7 +223,6 @@ def db_add_system_prompt(name: str, prompt_text: str) -> Dict[str, Any]:
         logger.error(f"Ошибка при сохранении промпта в БД: {e}")
         return {"status": "error", "error": str(e)}
 
-    
 def db_get_rules() -> List[Dict[str, Any]]:
     conn = get_db_connection()
     cursor = conn.cursor()
@@ -182,4 +258,96 @@ def db_clear_session_state(session_id: str):
     cursor = conn.cursor()
     cursor.execute("DELETE FROM session_states WHERE session_id = ?", (session_id,))
     conn.commit()
-    conn.close()
+    conn.close()
+
+def db_get_stats() -> Dict[str, Any]:
+    """Возвращает общую статистику по количеству записей во всех таблицах БД."""
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    tables = ['scud_logs', 'zup_staff', 'zup_absences', 'anomalies_history', 'ai_knowledge_base', 'system_prompts', 'session_states', 'tasks']
+    stats = {}
+    for t in tables:
+        try:
+            cursor.execute(f"SELECT COUNT(*) FROM {t}")
+            stats[t] = cursor.fetchone()[0]
+        except Exception:
+            stats[t] = 0
+    conn.close()
+    return {"status": "success", "tables_stats": stats}
+
+def db_get_anomalies(limit: int = 100, date_str: Optional[str] = None) -> Dict[str, Any]:
+    """Возвращает историю аномалий СКУД с опциональной фильтрацией по дате."""
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    
+    query = "SELECT anomaly_date, fio, anomaly_type, details FROM anomalies_history"
+    params = []
+    
+    if date_str:
+        query += " WHERE anomaly_date = ?"
+        params.append(date_str)
+        
+    query += " ORDER BY id DESC LIMIT ?"
+    params.append(limit)
+    
+    cursor.execute(query, params)
+    rows = cursor.fetchall()
+    conn.close()
+    
+    anomalies_list = [dict(r) for r in rows]
+    return {
+        "status": "success", 
+        "count": len(anomalies_list), 
+        "anomalies": anomalies_list
+    }
+
+def db_get_session_states() -> Dict[str, Any]:
+    """Возвращает список всех активных сессий и состояний превью."""
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    cursor.execute("SELECT session_id, state_type, updated_at FROM session_states")
+    rows = cursor.fetchall()
+    conn.close()
+    return {"status": "success", "active_sessions": [dict(r) for r in rows]}
+
+def db_delete_snapshots(snapshot_id: Optional[str] = None, day_str: Optional[str] = None) -> Dict[str, Any]:
+    """Удаляет снапшот по ID или за конкретную дату."""
+    if not snapshot_id and not day_str:
+        return {"status": "error", "message": "Необходимо указать snapshot_id или day_str (ДД.ММ.ГГГГ)."}
+    
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    
+    if snapshot_id:
+        cursor.execute("DELETE FROM scud_logs WHERE snapshot_id = ?", (snapshot_id,))
+        deleted = cursor.rowcount
+    else:
+        cursor.execute("DELETE FROM scud_logs WHERE log_date = ? OR snapshot_id LIKE ?", (day_str, f"%{day_str.replace('.', '')}%"))
+        deleted = cursor.rowcount
+        
+    conn.commit()
+    conn.close()
+    return {"status": "success", "message": f"Успешно удалено записей: {deleted}"}    
+
+def db_get_reference(category: Optional[str] = None) -> Dict[str, Any]:
+    """Возвращает системные справочники и примеры команд для оператора."""
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    
+    query = "SELECT category, title, example_prompt, description FROM system_reference"
+    params = []
+    
+    if category:
+        query += " WHERE category = ?"
+        params.append(category)
+        
+    query += " ORDER BY id ASC"
+    cursor.execute(query, params)
+    rows = cursor.fetchall()
+    conn.close()
+    
+    return {
+        "status": "success",
+        "count": len(rows),
+        "reference_items": [dict(r) for r in rows]
+    }