Update app.py

app.py

@@ -1,18 +1,20 @@
-from telegram.ext import ApplicationBuilder, CommandHandler, MessageHandler, filters, CallbackContext
+from telegram.ext import ApplicationBuilder, CommandHandler, MessageHandler, filters
 import google.generativeai as genai
 import os
 import logging
 
-# === Логирование ===
+# === Настройка логирования ===
 logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
-# === Настройки Gemini ===
-GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
+# === Проверка переменных окружения ===
 TELEGRAM_TOKEN = os.getenv("TELEGRAM_TOKEN")
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 
 if not TELEGRAM_TOKEN or not GEMINI_API_KEY:
     raise ValueError("TELEGRAM_TOKEN или GEMINI_API_KEY не установлены")
 
+# === Настройка Gemini ===
 genai.configure(api_key=GEMINI_API_KEY)
 
 generation_config = {
@@ -26,4 +28,69 @@ safety_settings = [
     {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
     {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
     {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
-    {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_MEDIUM
+    {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+]
+
+model = genai.GenerativeModel(
+    model_name="gemini-1.0-pro",
+    generation_config=generation_config,
+    safety_settings=safety_settings
+)
+
+# Хранилище чатов для пользователей
+user_chats = {}
+
+# === Команды бота ===
+async def start(update, context):
+    user_id = update.effective_user.id
+    user_chats[user_id] = model.start_chat(history=[])
+    logger.info(f"[{user_id}] Начало нового чата")
+    await update.message.reply_text("Привет! Напишите что-нибудь, и я отвечу.")
+
+async def reset(update, context):
+    user_id = update.effective_user.id
+    if user_id in user_chats:
+        del user_chats[user_id]
+    logger.info(f"[{user_id}] История диалога очищена")
+    await update.message.reply_text("История диалога очищена.")
+
+async def handle_message(update, context):
+    user_id = update.effective_user.id
+    text = update.message.text.strip()
+
+    # Проверки
+    if len(text) == 0:
+        await update.message.reply_text("Вы отправили пустое сообщение.")
+        return
+
+    if len(text) > 2000:
+        await update.message.reply_text("Сообщение слишком длинное. Пожалуйста, сократите его.")
+        return
+
+    # Инициализация чата
+    if user_id not in user_chats:
+        user_chats[user_id] = model.start_chat(history=[])
+
+    chat = user_chats[user_id]
+
+    try:
+        logger.info(f"[{user_id}] Запрос: {text}")
+        response = chat.send_message(text)
+        logger.info(f"[{user_id}] Ответ: {response.text.strip()}")
+        await update.message.reply_text(response.text.strip())
+    except Exception as e:
+        error_msg = f"Ошибка: {str(e)}"
+        logger.error(error_msg)
+        await update.message.reply_text("Произошла ошибка при обработке запроса.")
+
+# === Точка входа ===
+if __name__ == '__main__':
+    application = ApplicationBuilder().token(TELEGRAM_TOKEN).build()
+
+    # Регистрация обработчиков
+    application.add_handler(CommandHandler('start', start))
+    application.add_handler(CommandHandler('reset', reset))
+    application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_message))
+
+    logger.info("Бот запущен...")
+    application.run_polling()