Update app.py

app.py

@@ -11,17 +11,15 @@ import time
 from datetime import datetime
 import logging
 import threading
+import asyncio
 from huggingface_hub import HfApi, hf_hub_download
 from huggingface_hub.utils import RepositoryNotFoundError
 
-from aiogram import Bot, Dispatcher, executor, types
-import asyncio
-
-import base64
+from aiogram import Bot, Dispatcher, types, F
 from google import genai
 from google.genai import types
 
-BOT_TOKEN = os.getenv("BOT_TOKEN", "7566834146:AAGiG4MaTZZvvbTVsqEJVG5SYK5hUlc_Ewo")
+BOT_TOKEN = os.getenv("BOT_TOKEN", "7566834146:AAGiG4MaTZZvvtVsqEJVG5SYK5hUlc_Ewo")
 HOST = '0.0.0.0'
 PORT = 7860
 DATA_FILE = 'data.json'
@@ -31,6 +29,7 @@ HF_DATA_FILE_PATH = "data.json"
 HF_TOKEN_WRITE = os.getenv("HF_TOKEN_WRITE")
 HF_TOKEN_READ = os.getenv("HF_TOKEN_READ")
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
+GENAI_MODEL = "learnlm-2.0-flash-experimental"
 
 app = Flask(__name__)
 logging.basicConfig(level=logging.INFO)
@@ -38,79 +37,9 @@ app.secret_key = os.urandom(24)
 
 _data_lock = threading.Lock()
 visitor_data_cache = {}
-chat_history_cache = []
-
-bot = Bot(token=BOT_TOKEN)
-dp = Dispatcher(bot)
-
-APP_CONTEXT_TEXT = """
-Morshen Group: Международный IT холдинг.
-Объединяем передовые технологические компании для создания инновационных
-решений мирового уровня. Мы строим будущее технологий сегодня.
-Лидер инноваций 2025.
-Контакт: Telegram @morshenkhan, Телефон +996 500 398 754.
-
-Экосистема инноваций: В состав холдинга входят компании, специализирующиеся на различных
-направлениях передовых технологий, создавая синергию для прорывных решений.
-
-Morshen Alpha: Флагманская компания холдинга.
-Специализация: Искусственный интеллект, Квантовые технологии, Бизнес-решения.
-Деятельность: Разрабатываем передовые бизнес-решения, проводим R&D в сфере AI
-и квантовых технологий. Наши инновации формируют будущее индустрии.
-Статистика: 3+ Страны присутствия, 3K+ Готовых клиентов, 5+ Лет на рынке.
-
-Holmgard Studio: Инновационная студия разработки.
-Специализация: Веб-разработка, Мобильные приложения, ПО на заказ.
-Деятельность: Создает высокотехнологичные веб-сайты,
-мобильные приложения и ПО для бизнеса любого масштаба.
-Использует передовые технологии и гибкие методологии.
-Статистика: 10+ Лет опыта, PRO Любая сложность, FAST Высокая скорость.
-Веб-сайт: https://holmgard.ru. Контакт через @morshenkhan.
-
-Глобальное присутствие: Наши инновационные решения и экспертиза доступны в странах Центральной Азии и за ее пределами:
-Узбекистан
-Казахстан
-Кыргызстан
-Расширяем горизонты...
-
-Сохранить визитку: Телефон +996 500 398 754, Morshen Group, Международный IT Холдинг. Сделайте скриншот экрана.
-"""
-
-def generate_ai_response(prompt_text):
-    if not GEMINI_API_KEY:
-        logging.error("GEMINI_API_KEY not set. Cannot generate AI response.")
-        return "Ошибка: Сервер не настроен для ответа на вопросы (отсутствует API ключ)."
-    try:
-        client = genai.Client(api_key=GEMINI_API_KEY)
-        model = "learnlm-2.0-flash-experimental"
-        contextualized_prompt = f"""
-Основываясь на следующей информации о Morshen Group и ее дочерних компаниях, ответь на вопрос пользователя.
-Если информация отсутствует, так и скажи. Не придумывай информацию, которой нет в тексте.
-Отвечай на русском языке.
-
-Информация о Morshen Group:
-{APP_CONTEXT_TEXT}
-
-Вопрос пользователя: {prompt_text}
-
-Ответ:
-"""
-        contents = [types.Content(role="user", parts=[types.Part.from_text(text=contextualized_prompt)])]
-        generate_content_config = types.GenerateContentConfig(response_mime_type="text/plain")
-
-        response = client.models.generate_content(
-            model=model,
-            contents=contents,
-            config=generate_content_config,
-            stream=False # Use non-streaming for simplicity in this context
-        )
-        return response.text.strip()
-    except Exception as e:
-        logging.error(f"Error generating AI response: {e}")
-        return f"Произошла ошибка при генерации ответа: {e}"
 
 def download_data_from_hf():
-    global visitor_data_cache, chat_history_cache
+    global visitor_data_cache
     if not HF_TOKEN_READ:
         logging.warning("HF_TOKEN_READ not set. Skipping Hugging Face download.")
         return False
@@ -130,14 +59,11 @@ def download_data_from_hf():
         with _data_lock:
             try:
                 with open(DATA_FILE, 'r', encoding='utf-8') as f:
-                    full_data = json.load(f)
-                    visitor_data_cache = full_data.get("visitors", {})
-                    chat_history_cache = full_data.get("chats", [])
+                    visitor_data_cache = json.load(f)
                 logging.info("Successfully loaded downloaded data into cache.")
             except (FileNotFoundError, json.JSONDecodeError) as e:
                  logging.error(f"Error reading downloaded data file: {e}. Starting with empty cache.")
                  visitor_data_cache = {}
-                 chat_history_cache = []
         return True
     except RepositoryNotFoundError:
         logging.error(f"Hugging Face repository '{REPO_ID}' not found. Cannot download data.")
@@ -146,53 +72,34 @@ def download_data_from_hf():
     return False
 
 def load_visitor_data():
-    global visitor_data_cache, chat_history_cache
+    global visitor_data_cache
     with _data_lock:
-        if not visitor_data_cache or not chat_history_cache:
+        if not visitor_data_cache:
             try:
                 with open(DATA_FILE, 'r', encoding='utf-8') as f:
-                    full_data = json.load(f)
-                    visitor_data_cache = full_data.get("visitors", {})
-                    chat_history_cache = full_data.get("chats", [])
-                logging.info("Data loaded from local JSON.")
+                    visitor_data_cache = json.load(f)
+                logging.info("Visitor data loaded from local JSON.")
             except FileNotFoundError:
                 logging.warning(f"{DATA_FILE} not found locally. Starting with empty data.")
                 visitor_data_cache = {}
-                chat_history_cache = []
             except json.JSONDecodeError:
                 logging.error(f"Error decoding {DATA_FILE}. Starting with empty data.")
                 visitor_data_cache = {}
-                chat_history_cache = []
             except Exception as e:
-                logging.error(f"Unexpected error loading data: {e}")
+                logging.error(f"Unexpected error loading visitor data: {e}")
                 visitor_data_cache = {}
-                chat_history_cache = []
-        return {"visitors": visitor_data_cache, "chats": chat_history_cache}
-
+        return visitor_data_cache
 
-def save_visitor_data(data_to_update):
+def save_visitor_data(data):
     with _data_lock:
         try:
-            visitor_data_cache.update(data_to_update)
-            full_data = {"visitors": visitor_data_cache, "chats": chat_history_cache}
+            visitor_data_cache.update(data)
             with open(DATA_FILE, 'w', encoding='utf-8') as f:
-                json.dump(full_data, f, ensure_ascii=False, indent=4)
-            logging.info(f"Data successfully saved to {DATA_FILE}.")
+                json.dump(visitor_data_cache, f, ensure_ascii=False, indent=4)
+            logging.info(f"Visitor data successfully saved to {DATA_FILE}.")
             upload_data_to_hf_async()
         except Exception as e:
-            logging.error(f"Error saving data: {e}")
-
-def save_chat_message(chat_entry):
-     with _data_lock:
-         try:
-             chat_history_cache.append(chat_entry)
-             full_data = {"visitors": visitor_data_cache, "chats": chat_history_cache}
-             with open(DATA_FILE, 'w', encoding='utf-8') as f:
-                 json.dump(full_data, f, ensure_ascii=False, indent=4)
-             logging.info(f"Chat message successfully saved to {DATA_FILE}.")
-             upload_data_to_hf_async()
-         except Exception as e:
-             logging.error(f"Error saving chat message: {e}")
+            logging.error(f"Error saving visitor data: {e}")
 
 def upload_data_to_hf():
     if not HF_TOKEN_WRITE:
@@ -217,9 +124,9 @@ def upload_data_to_hf():
                  repo_id=REPO_ID,
                  repo_type="dataset",
                  token=HF_TOKEN_WRITE,
-                 commit_message=f"Update data {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
+                 commit_message=f"Update visitor data {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
              )
-        logging.info("Data successfully uploaded to Hugging Face.")
+        logging.info("Visitor data successfully uploaded to Hugging Face.")
     except Exception as e:
         logging.error(f"Error uploading data to Hugging Face: {e}")
 
@@ -256,7 +163,7 @@ def verify_telegram_data(init_data_str):
             auth_date = int(parsed_data.get('auth_date', [0])[0])
             current_time = int(time.time())
             if current_time - auth_date > 86400:
-                 logging.warning(f"Telegram InitData is older than 24 hours (Auth Date: {auth_date}, Current: {current_time}).")
+                 logging.warning(f"Telegram InitData is older than 1 hour (Auth Date: {auth_date}, Current: {current_time}).")
             return parsed_data, True
         else:
             logging.warning(f"Data verification failed. Calculated: {calculated_hash}, Received: {received_hash}")
@@ -286,7 +193,7 @@ TEMPLATE = """
             --tg-theme-button-text-color: {{ theme.button_text_color | default('#ffffff') }};
             --tg-theme-secondary-bg-color: {{ theme.secondary_bg_color | default('#1e1e1e') }};
 
-            --bg-gradient: linear-gradient(160deg, #1a232f 0%, var(--tg-theme-bg-color, #121212) 100%);
+            --bg-gradient: linear-gradient(160deg, #1a232f 0%, #121212 100%);
             --card-bg: rgba(44, 44, 46, 0.8);
             --card-bg-solid: #2c2c2e;
             --text-color: var(--tg-theme-text-color);
@@ -465,7 +372,6 @@ TEMPLATE = """
         }
         .save-card-button i { font-size: 1.2em; }
 
-        /* Modal Styles */
         .modal {
             display: none; position: fixed; z-index: 1001;
             left: 0; top: 0; width: 100%; height: 100%;
@@ -497,7 +403,6 @@ TEMPLATE = """
         .modal-text b { color: var(--tg-theme-link-color); font-weight: 600; }
         .modal-instruction { font-size: 1em; color: var(--text-secondary-color); margin-top: var(--padding-m); }
 
-        /* Icons */
         .icon { display: inline-block; width: 1.2em; text-align: center; margin-right: 8px; opacity: 0.9; }
         .icon-save::before { content: '💾'; }
         .icon-web::before { content: '🌐'; }
@@ -519,7 +424,6 @@ TEMPLATE = """
         .icon-leader::before { content: '🏆'; }
         .icon-company::before { content: '🏢'; }
 
-        /* Responsive adjustments */
         @media (max-width: 480px) {
             .section-title { font-size: 1.8em; }
             .logo span { font-size: 1.4em; }
@@ -649,7 +553,6 @@ TEMPLATE = """
         <i class="icon icon-save"></i>Сохранить визитку
     </button>
 
-    <!-- The Modal -->
     <div id="saveModal" class="modal">
       <div class="modal-content">
         <span class="modal-close" id="modal-close-btn">×</span>
@@ -660,7 +563,6 @@ TEMPLATE = """
       </div>
     </div>
 
-
     <script>
         const tg = window.Telegram.WebApp;
 
@@ -796,7 +698,7 @@ ADMIN_TEMPLATE = """
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>Admin - Посетители и Чаты</title>
+    <title>Admin - Посетители</title>
     <link rel="preconnect" href="https://fonts.googleapis.com">
     <link rel="preconnect" href="https://fonts.gstatic.com" crossorigin>
     <link href="https://fonts.googleapis.com/css2?family=Inter:wght@300;400;500;600;700&display=swap" rel="stylesheet">
@@ -812,7 +714,6 @@ ADMIN_TEMPLATE = """
             --admin-success: #198754;
             --admin-danger: #dc3545;
             --admin-warning: #ffc107;
-            --admin-info: #0dcaf0;
             --border-radius: 12px;
             --padding: 1.5rem;
             --font-family: 'Inter', sans-serif;
@@ -826,28 +727,25 @@ ADMIN_TEMPLATE = """
             line-height: 1.6;
         }
         .container { max-width: 1140px; margin: 0 auto; }
-        h1, h2 { text-align: center; color: var(--admin-secondary); margin-bottom: var(--padding); font-weight: 600; }
-         h2 { margin-top: var(--padding); border-top: 1px solid var(--admin-border); padding-top: var(--padding); }
+        h1 { text-align: center; color: var(--admin-secondary); margin-bottom: var(--padding); font-weight: 600; }
         .user-grid {
             display: grid;
             grid-template-columns: repeat(auto-fill, minmax(280px, 1fr));
             gap: var(--padding);
             margin-top: var(--padding);
         }
-        .user-card, .chat-card {
+        .user-card {
             background-color: var(--admin-card-bg);
             border-radius: var(--border-radius);
             padding: var(--padding);
             box-shadow: 0 4px 15px var(--admin-shadow);
             border: 1px solid var(--admin-border);
+            display: flex;
+            flex-direction: column;
+            align-items: center;
+            text-align: center;
             transition: transform 0.2s ease-in-out, box-shadow 0.2s ease-in-out;
         }
-        .user-card {
-             display: flex;
-             flex-direction: column;
-             align-items: center;
-             text-align: center;
-        }
         .user-card:hover {
             transform: translateY(-5px);
             box-shadow: 0 8px 25px rgba(0, 0, 0, 0.08);
@@ -864,7 +762,7 @@ ADMIN_TEMPLATE = """
         .user-card .detail-item { margin-bottom: 0.3rem; }
         .user-card .detail-item strong { color: var(--admin-text); }
         .user-card .timestamp { font-size: 0.8em; color: var(--admin-secondary); margin-top: 1rem; }
-        .no-users, .no-chats { text-align: center; color: var(--admin-secondary); margin-top: 2rem; font-size: 1.1em; }
+        .no-users { text-align: center; color: var(--admin-secondary); margin-top: 2rem; font-size: 1.1em; }
         .alert {
              background-color: #fff3cd; border-left: 6px solid var(--admin-warning);
              margin-bottom: var(--padding); padding: 1rem 1.5rem;
@@ -885,7 +783,6 @@ ADMIN_TEMPLATE = """
         }
          .refresh-btn:hover { background-color: #0b5ed7; }
 
-        /* Admin Controls */
         .admin-controls {
             background: var(--admin-card-bg);
             padding: var(--padding);
@@ -918,102 +815,11 @@ ADMIN_TEMPLATE = """
              display: none;
          }
          @keyframes spin { 0% { transform: rotate(0deg); } 100% { transform: rotate(360deg); } }
-
-        /* Chat History Styles */
-        .chat-history { margin-top: var(--padding); }
-        .chat-message {
-             margin-bottom: var(--padding-s);
-             padding: var(--padding-s) var(--padding-m);
-             border-radius: var(--border-radius-s);
-             max-width: 85%;
-             word-wrap: break-word;
-        }
-        .chat-message.user {
-             background-color: var(--admin-primary);
-             color: white;
-             align-self: flex-start;
-             margin-right: auto;
-        }
-        .chat-message.bot {
-             background-color: #e9ecef;
-             color: var(--admin-text);
-             align-self: flex-end;
-             margin-left: auto;
-        }
-         .chat-message .sender { font-weight: 600; margin-bottom: 4px; font-size: 0.9em;}
-         .chat-message .text { margin-bottom: 4px; font-size: 1em;}
-         .chat-message .time { font-size: 0.75em; color: rgba(255, 255, 255, 0.7); text-align: right; display: block;}
-         .chat-message.bot .time { color: #495057; }
-
-        .chat-thread {
-            background-color: var(--admin-card-bg);
-            border-radius: var(--border-radius);
-            padding: var(--padding);
-            box-shadow: 0 4px 15px var(--admin-shadow);
-            border: 1px solid var(--admin-border);
-            margin-bottom: var(--padding);
-        }
-        .chat-thread-header {
-            font-weight: 600;
-            font-size: 1.1em;
-            margin-bottom: var(--padding-m);
-            padding-bottom: var(--padding-s);
-            border-bottom: 1px solid var(--admin-border);
-             display: flex;
-             justify-content: space-between;
-             align-items: center;
-        }
-         .chat-thread-header span { color: var(--admin-primary); }
-        .chat-thread-messages {
-            display: flex;
-            flex-direction: column;
-            gap: var(--padding-s);
-        }
-
-        /* Send Message Form */
-        .send-message-form {
-             margin-top: var(--padding);
-             padding: var(--padding);
-             background: var(--admin-card-bg);
-             border-radius: var(--border-radius);
-             box-shadow: 0 4px 15px var(--admin-shadow);
-             border: 1px solid var(--admin-border);
-        }
-         .send-message-form h3 { margin-top: 0; margin-bottom: 1rem; color: var(--admin-secondary); font-weight: 600;}
-         .send-message-form label { display: block; margin-bottom: 0.5rem; font-weight: 500;}
-         .send-message-form select, .send-message-form textarea {
-             width: 100%;
-             padding: 0.75rem;
-             margin-bottom: 1rem;
-             border: 1px solid var(--admin-border);
-             border-radius: 8px;
-             font-family: var(--font-family);
-             font-size: 1em;
-             box-sizing: border-box;
-         }
-         .send-message-form textarea { min-height: 100px; resize: vertical; }
-         .send-message-form button {
-             display: inline-block;
-             padding: 10px 20px;
-             font-size: 1em;
-             font-weight: 500;
-             color: #fff;
-             background-color: var(--admin-success);
-             border: none;
-             border-radius: 8px;
-             cursor: pointer;
-             transition: background-color 0.2s ease;
-         }
-         .send-message-form button:hover { background-color: #157347; }
-         .send-message-status { margin-top: 1rem; text-align: center; font-weight: 500;}
-         .status-success { color: var(--admin-success); }
-         .status-error { color: var(--admin-danger); }
-         .status-info { color: var(--admin-info); }
     </style>
 </head>
 <body>
     <div class="container">
-        <h1>Admin Panel</h1>
+        <h1>Посетители Mini App</h1>
         <div class="alert">ВНИМАНИЕ: Этот раздел не защищен! Добавьте аутентификацию для реального использования.</div>
 
         <div class="admin-controls">
@@ -1024,9 +830,8 @@ ADMIN_TEMPLATE = """
             <div class="status" id="status-message"></div>
         </div>
 
-        <button class="refresh-btn" onclick="location.reload()">Обновить страницу</button>
+        <button class="refresh-btn" onclick="location.reload()">Обновить список</button>
 
-        <h2>Посетители Mini App</h2>
         {% if users %}
             <div class="user-grid">
                 {% for user in users|sort(attribute='visited_at', reverse=true) %}
@@ -1051,75 +856,11 @@ ADMIN_TEMPLATE = """
         {% else %}
             <p class="no-users">Данных о посетителях пока нет.</p>
         {% endif %}
-
-         <h2>История чатов с ботом</h2>
-         {% if chats %}
-            <div class="chat-history">
-                {% for chat_id, messages in chats|dictsort %}
-                    <div class="chat-thread">
-                         <div class="chat-thread-header">
-                             <span>Чат ID: {{ chat_id }}</span>
-                             {% set chat_user = users_by_id.get(chat_id|int) %}
-                             {% if chat_user %}
-                                 <span>Пользователь:
-                                     {% if chat_user.username %}
-                                         <a href="https://t.me/{{ chat_user.username }}" target="_blank" style="color: var(--admin-secondary); text-decoration: none;">@{{ chat_user.username }}</a>
-                                     {% else %}
-                                         {{ chat_user.first_name or 'Неизвестный' }}
-                                     {% endif %}
-                                     ({{ chat_user.id }})
-                                 </span>
-                             {% else %}
-                                 <span>Пользователь: Неизвестен ({{ chat_id }})</span>
-                             {% endif %}
-                         </div>
-                        <div class="chat-thread-messages">
-                            {% for message in messages %}
-                                <div class="chat-message {{ 'user' if message.role == 'user' else 'bot' }}">
-                                    <div class="sender">{{ message.sender_name }}</div>
-                                    <div class="text">{{ message.text }}</div>
-                                     <div class="time">{{ message.timestamp_str }}</div>
-                                </div>
-                            {% endfor %}
-                        </div>
-                    </div>
-                {% endfor %}
-            </div>
-         {% else %}
-             <p class="no-chats">История чатов пуста.</p>
-         {% endif %}
-
-         <div class="send-message-form">
-             <h3>Отправить сообщение пользователю</h3>
-             <form id="sendMessageForm">
-                 <label for="chat_id">Выберите пользователя (Chat ID):</label>
-                 <select id="chat_id" name="chat_id" required>
-                     <option value="">-- Выберите --</option>
-                     {% for user in users|sort(attribute='first_name') %}
-                         <option value="{{ user.id }}">
-                             {{ user.first_name or '' }} {{ user.last_name or '' }}
-                             {% if user.username %} (@{{ user.username }}) {% endif %}
-                             (ID: {{ user.id }})
-                         </option>
-                     {% endfor %}
-                 </select>
-
-                 <label for="message_text">Текст сообщения:</label>
-                 <textarea id="message_text" name="message_text" required></textarea>
-
-                 <button type="submit">Отправить</button>
-             </form>
-              <div class="send-message-status" id="sendMessageStatus"></div>
-         </div>
-
-
     </div>
 
     <script>
         const loader = document.getElementById('loader');
         const statusMessage = document.getElementById('status-message');
-        const sendMessageForm = document.getElementById('sendMessageForm');
-        const sendMessageStatus = document.getElementById('sendMessageStatus');
 
         async function handleFetch(url, action) {
             loader.style.display = 'inline-block';
@@ -1133,8 +874,6 @@ ADMIN_TEMPLATE = """
                     statusMessage.style.color = 'var(--admin-success)';
                     if (action === 'скачивание') {
                         setTimeout(() => location.reload(), 1500);
-                    } else {
-                        setTimeout(() => statusMessage.textContent = '', 3000);
                     }
                 } else {
                     throw new Error(data.message || 'Произошла ошибка');
@@ -1143,7 +882,6 @@ ADMIN_TEMPLATE = """
                 statusMessage.textContent = `Ошибка ${action}: ${error.message}`;
                 statusMessage.style.color = 'var(--admin-danger)';
                 console.error(`Error during ${action}:`, error);
-                setTimeout(() => statusMessage.textContent = '', 5000);
             } finally {
                 loader.style.display = 'none';
             }
@@ -1156,111 +894,77 @@ ADMIN_TEMPLATE = """
         function triggerUpload() {
             handleFetch('/admin/upload_data', 'загрузка');
         }
-
-        sendMessageForm.addEventListener('submit', async function(event) {
-            event.preventDefault();
-            const chatId = document.getElementById('chat_id').value;
-            const messageText = document.getElementById('message_text').value;
-
-            if (!chatId || !messageText.trim()) {
-                sendMessageStatus.textContent = 'Выберите пользователя и введите текст сообщения.';
-                sendMessageStatus.className = 'send-message-status status-warning';
-                return;
-            }
-
-            sendMessageStatus.textContent = 'Отправка сообщения...';
-            sendMessageStatus.className = 'send-message-status status-info';
-
-            try {
-                const response = await fetch('/admin/send_message', {
-                    method: 'POST',
-                    headers: { 'Content-Type': 'application/json' },
-                    body: JSON.stringify({ chat_id: chatId, message: messageText })
-                });
-                const data = await response.json();
-
-                if (response.ok && data.status === 'ok') {
-                    sendMessageStatus.textContent = 'Сообщение успешно отправлено!';
-                    sendMessageStatus.className = 'send-message-status status-success';
-                    document.getElementById('message_text').value = '';
-                    setTimeout(() => location.reload(), 1500); // Reload to show sent message
-                } else {
-                    throw new Error(data.message || 'Неизвестная ошибка');
-                }
-            } catch (error) {
-                sendMessageStatus.textContent = `Ошибка отправки: ${error.message}`;
-                sendMessageStatus.className = 'send-message-status status-error';
-                console.error('Error sending message:', error);
-            }
-        });
-
     </script>
 </body>
 </html>
 """
 
-@dp.message_handler()
-async def handle_messages(message: types.Message):
-    user_id = message.from_user.id
-    user_full_name = message.from_user.full_name or f"User {user_id}"
-    chat_id = message.chat.id
-    message_text = message.text
-
-    now = time.time()
-    timestamp_str = datetime.fromtimestamp(now).strftime('%Y-%m-%d %H:%M:%S')
+bot = Bot(token=BOT_TOKEN)
+dp = Dispatcher()
 
-    save_chat_message({
-        'chat_id': chat_id,
-        'role': 'user',
-        'sender_id': user_id,
-        'sender_name': user_full_name,
-        'text': message_text,
-        'timestamp': now,
-        'timestamp_str': timestamp_str
-    })
+def generate_bot_response(query: str, context: str):
+    if not GEMINI_API_KEY:
+        logging.warning("GEMINI_API_KEY not set. Cannot use AI.")
+        return "Извините, функция ответа с использованием AI временно недоступна."
 
-    logging.info(f"Received message from user {user_id} ({user_full_name}): {message_text}")
+    lower_query = query.lower()
+    code_keywords = ["код", "скрипт", "программа", "напиши", "сделай", "код на", "пример кода"]
+    if any(keyword in lower_query for keyword in code_keywords):
+         return "Извините, я не могу предоставить вам код. Могу ли я ответить на другие вопросы о Morshen Group или Holmgard Studio?"
 
     try:
-        ai_response_text = generate_ai_response(message_text)
-        await message.answer(ai_response_text)
-
-        now_bot = time.time()
-        timestamp_str_bot = datetime.fromtimestamp(now_bot).strftime('%Y-%m-%d %H:%M:%S')
-        save_chat_message({
-            'chat_id': chat_id,
-            'role': 'bot',
-            'sender_id': bot.id,
-            'sender_name': "Bot",
-            'text': ai_response_text,
-            'timestamp': now_bot,
-            'timestamp_str': timestamp_str_bot
-        })
-        logging.info(f"Sent AI response to user {user_id}: {ai_response_text}")
+        client = genai.Client(api_key=GEMINI_API_KEY)
+
+        prompt = f"""You are a helpful assistant answering questions about the following website content for "Morshen Group" and "Holmgard Studio".
+Only use information explicitly provided in the CONTENT below. Do not invent or assume information not present in the content.
+If the user asks for code, *you must refuse* and explain that you cannot provide code.
+Keep responses concise and directly related to the query and content.
+
+CONTENT:
+{context}
+
+USER QUERY: {query}
+
+ASSISTANT RESPONSE:"""
+
+        contents = [
+            types.Content(
+                role="user",
+                parts=[
+                    types.Part.from_text(text=prompt),
+                ],
+            ),
+        ]
+        generate_content_config = types.GenerateContentConfig(
+            response_mime_type="text/plain",
+            temperature=0.1,
+            max_output_tokens=500
+        )
+
+        response = client.models.generate_content(
+            model=GENAI_MODEL,
+            contents=contents,
+            config=generate_content_config,
+        )
+        return response.text.strip()
 
     except Exception as e:
-        logging.error(f"Error handling message for user {user_id}: {e}")
-        await message.answer("Произошла ошибка при обработке вашего запроса.")
-        now_bot = time.time()
-        timestamp_str_bot = datetime.fromtimestamp(now_bot).strftime('%Y-%m-%d %H:%M:%S')
-        save_chat_message({
-            'chat_id': chat_id,
-            'role': 'bot',
-            'sender_id': bot.id,
-            'sender_name': "Bot",
-            'text': "Произошла ошибка при обработке вашего запроса.",
-            'timestamp': now_bot,
-            'timestamp_str': timestamp_str_bot
-        })
-
-
-def run_bot():
-    try:
-        logging.info("Starting aiogram bot polling...")
-        executor.start_polling(dp, skip_updates=True)
-    except Exception as e:
-        logging.error(f"Error starting aiogram bot polling: {e}")
+        logging.error(f"Error generating AI response: {e}")
+        return "Извините, произошла ошибка при обработке вашего запроса."
+
+@dp.message()
+async def handle_message(message: types.Message):
+    user_query = message.text
+    if not user_query:
+        return
+
+    await message.answer("Думаю...")
+
+    context = TEMPLATE
+
+    response_text = generate_bot_response(user_query, context)
 
+    await message.answer(response_text)
 
 @app.route('/')
 def index():
@@ -1316,26 +1020,9 @@ def verify_data():
 
 @app.route('/admin')
 def admin_panel():
-    full_data = load_visitor_data()
-    users_list = list(full_data.get("visitors", {}).values())
-    chat_entries = full_data.get("chats", [])
-
-    chats_by_id = {}
-    for chat in chat_entries:
-        chat_id = str(chat.get('chat_id'))
-        if chat_id not in chats_by_id:
-            chats_by_id[chat_id] = []
-        chats_by_id[chat_id].append(chat)
-
-    for chat_id in chats_by_id:
-        chats_by_id[chat_id].sort(key=lambda x: x.get('timestamp', 0))
-
-    users_by_id = {user['id']: user for user in users_list}
-
-    return render_template_string(ADMIN_TEMPLATE,
-                                  users=users_list,
-                                  chats=chats_by_id,
-                                  users_by_id=users_by_id)
+    current_data = load_visitor_data()
+    users_list = list(current_data.values())
+    return render_template_string(ADMIN_TEMPLATE, users=users_list)
 
 @app.route('/admin/download_data', methods=['POST'])
 def admin_trigger_download():
@@ -1352,96 +1039,58 @@ def admin_trigger_upload():
     upload_data_to_hf_async()
     return jsonify({"status": "ok", "message": "Загрузка данных на Hugging Face запущена в фоновом режиме."})
 
-@app.route('/admin/send_message', methods=['POST'])
-def admin_send_message():
+def start_bot_polling():
     try:
-        req_data = request.get_json()
-        chat_id = req_data.get('chat_id')
-        message_text = req_data.get('message')
-
-        if not chat_id or not message_text:
-            return jsonify({"status": "error", "message": "Missing chat_id or message text"}), 400
-
-        try:
-            chat_id = int(chat_id)
-        except ValueError:
-             return jsonify({"status": "error", "message": "Invalid chat ID"}), 400
-
-        loop = asyncio.new_event_loop()
-        asyncio.set_event_loop(loop)
-        try:
-            loop.run_until_complete(bot.send_message(chat_id=chat_id, text=message_text))
-
-            now = time.time()
-            timestamp_str = datetime.fromtimestamp(now).strftime('%Y-%m-%d %H:%M:%S')
-            save_chat_message({
-                'chat_id': chat_id,
-                'role': 'admin_sent',
-                'sender_id': 'admin', # Or a specific admin identifier
-                'sender_name': 'Admin',
-                'text': message_text,
-                'timestamp': now,
-                'timestamp_str': timestamp_str
-            })
-
-            return jsonify({"status": "ok", "message": "Сообщение успешно отправлено!"}), 200
-        except Exception as e:
-            logging.error(f"Error sending message via bot API: {e}")
-            return jsonify({"status": "error", "message": f"Ошибка отправки сообщения через Bot API: {e}"}), 500
-        finally:
-            loop.close()
-
-
+        asyncio.run(dp.start_polling(bot))
     except Exception as e:
-        logging.exception("Error in /admin/send_message endpoint")
-        return jsonify({"status": "error", "message": "Internal server error"}), 500
-
+        logging.error(f"Error in bot polling thread: {e}")
 
 if __name__ == '__main__':
-    logging.info("---")
-    logging.info("--- MORSHEN GROUP MINI APP & BOT SERVER ---")
-    logging.info("---")
-    logging.info(f"Flask server starting on http://{HOST}:{PORT}")
-    logging.info(f"Using Bot Token ID: {BOT_TOKEN.split(':')[0]}")
-    logging.info(f"Visitor data file: {DATA_FILE}")
-    logging.info(f"Hugging Face Repo: {REPO_ID}")
-    logging.info(f"HF Data Path: {HF_DATA_FILE_PATH}")
+    print("---")
+    print("--- MORSHEN GROUP SERVER (FLASK + AIOGRAM) ---")
+    print("---")
+    print(f"Flask server starting on http://{HOST}:{PORT}")
+    print(f"Using Bot Token ID: {BOT_TOKEN.split(':')[0]}")
+    print(f"Visitor data file: {DATA_FILE}")
+    print(f"Hugging Face Repo: {REPO_ID}")
+    print(f"HF Data Path: {HF_DATA_FILE_PATH}")
     if not HF_TOKEN_READ or not HF_TOKEN_WRITE:
-         logging.warning("---")
-         logging.warning("--- WARNING: HUGGING FACE TOKEN(S) NOT SET ---")
-         logging.warning("--- Backup/restore functionality will be limited. Set HF_TOKEN_READ and HF_TOKEN_WRITE environment variables.")
-         logging.warning("---")
+         print("---")
+         print("--- WARNING: HUGGING FACE TOKEN(S) NOT SET ---")
+         print("--- Backup/restore functionality will be limited. Set HF_TOKEN_READ and HF_TOKEN_WRITE environment variables.")
+         print("---")
     else:
-         logging.info("--- Hugging Face tokens found.")
-         logging.info("--- Attempting initial data download from Hugging Face...")
+         print("--- Hugging Face tokens found.")
+         print("--- Attempting initial data download from Hugging Face...")
          download_data_from_hf()
 
     if not GEMINI_API_KEY:
-        logging.warning("---")
-        logging.warning("--- WARNING: GEMINI_API_KEY NOT SET ---")
-        logging.warning("--- AI response functionality will be disabled. Set GEMINI_API_KEY environment variable.")
-        logging.warning("---")
+         print("---")
+         print("--- WARNING: GEMINI_API_KEY NOT SET ---")
+         print("--- AI response functionality will be limited or unavailable. Set GEMINI_API_KEY environment variable.")
+         print("---")
     else:
-         logging.info("--- GEMINI_API_KEY found.")
+        print("--- GEMINI_API_KEY found.")
+        print(f"--- Using AI model: {GENAI_MODEL}")
 
     load_visitor_data()
 
-    logging.warning("---")
-    logging.warning("--- SECURITY WARNING ---")
-    logging.warning("--- The /admin route and its sub-routes are NOT protected.")
-    logging.warning("--- Implement proper authentication before deploying.")
-    logging.warning("---")
+    print("---")
+    print("--- SECURITY WARNING ---")
+    print("--- The /admin route and its sub-routes are NOT protected.")
+    print("--- Implement proper authentication before deploying.")
+    print("---")
 
     if HF_TOKEN_WRITE:
         backup_thread = threading.Thread(target=periodic_backup, daemon=True)
         backup_thread.start()
-        logging.info("--- Periodic backup thread started (every hour).")
+        print("--- Periodic backup thread started (every hour).")
     else:
-        logging.info("--- Periodic backup disabled (HF_TOKEN_WRITE missing).")
+        print("--- Periodic backup disabled (HF_TOKEN_WRITE missing).")
 
-    bot_thread = threading.Thread(target=run_bot, daemon=True)
+    bot_thread = threading.Thread(target=start_bot_polling, daemon=True)
     bot_thread.start()
-    logging.info("--- Telegram bot polling thread started.")
+    print("--- Telegram bot polling thread started.")
 
-    logging.info("--- Server Ready ---")
+    print("--- Server Ready ---")
     app.run(host=HOST, port=PORT, debug=False)