telegram_bot.py

import os
import speech_recognition as sr
import subprocess
from telegram import Update
from telegram.ext import ApplicationBuilder, MessageHandler, CommandHandler, ContextTypes, filters
from bot import bot
import db

TOKEN = '8543722157:AAFbvEtHXBkTUdloiDNIpxov98MrzEwraug'

# Определяем путь к ffmpeg
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
FFMPEG_PATH = os.path.join(BASE_DIR, "ffmpeg", "bin", "ffmpeg.exe")


async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
    await update.message.reply_text("Привет! Я бот по ремонту авто. Напиши или отправь голосовое сообщение.")


async def handle_voice_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
    user = update.message.from_user
    user_db_id = db.save_user(user)  # Сохраняем пользователя

    voice_file = await update.message.voice.get_file()
    file_data = await voice_file.download_as_bytearray()

    # Сохраняем голосовое сообщение
    ogg_file = 'user_voice.ogg'
    wav_file = 'user_voice.wav'

    with open(ogg_file, 'wb') as f:
        f.write(file_data)

    audio_file = None  # Инициализируем переменную заранее

    try:
        # Конвертация ogg в wav с помощью ffmpeg
        print(f"Конвертация {ogg_file} в {wav_file}...")

        # Проверяем, существует ли ffmpeg
        if not os.path.exists(FFMPEG_PATH):
            print(f"FFmpeg не найден по пути: {FFMPEG_PATH}")
            await update.message.reply_text("Ошибка: FFmpeg не установлен. Пожалуйста, напишите текстом.")
            return

        # Команда для конвертации
        cmd = [FFMPEG_PATH, '-i', ogg_file, '-acodec', 'pcm_s16le', '-ar', '16000', '-ac', '1', wav_file]
        print(f"Выполняем команду: {' '.join(cmd)}")

        try:
            result = subprocess.run(cmd, capture_output=True, text=True, timeout=30)
            if result.returncode != 0:
                print(f"Ошибка ffmpeg: {result.stderr}")
                await update.message.reply_text("Ошибка при конвертации аудио. Попробуйте отправить текстом.")
                return

            print("Конвертация успешно завершена!")

        except subprocess.TimeoutExpired:
            print("Таймаут при конвертации аудио")
            await update.message.reply_text("Таймаут при обработке аудио. Попробуйте снова.")
            return
        except Exception as e:
            print(f"Ошибка при запуске ffmpeg: {e}")
            await update.message.reply_text(f"Ошибка обработки: {str(e)}")
            return

        if not os.path.exists(wav_file):
            await update.message.reply_text("Не удалось создать файл .wav. Попробуйте снова.")
            return

        # Распознаем текст
        recognizer = sr.Recognizer()
        try:
            with sr.AudioFile(wav_file) as source:
                # Уменьшаем шум и настраиваем чувствительность
                recognizer.adjust_for_ambient_noise(source, duration=0.5)
                audio_data = recognizer.record(source)

            user_message = recognizer.recognize_google(audio_data, language="ru-RU")
            print(f"Распознанный текст: {user_message}")
            await update.message.reply_text(f"Вы сказали: {user_message}")

            # Получаем ответ от бота
            response, audio_file = bot(user_message)
            db.log_interaction(user_db_id, user_message, response)  # Логируем взаимодействие

            await update.message.reply_text(response)

            # Отправляем аудиоответ, если он создан
            if audio_file and os.path.exists(audio_file):
                with open(audio_file, 'rb') as voice:
                    await update.message.reply_voice(voice)
            else:
                print(f"Аудиофайл не найден: {audio_file}")

        except sr.UnknownValueError:
            await update.message.reply_text("Не удалось распознать речь. Пожалуйста, повторите или напишите текстом.")
        except sr.RequestError as e:
            print(f"Ошибка сервиса распознавания: {e}")
            await update.message.reply_text("Ошибка сервиса распознавания речи. Попробуйте позже.")
        except Exception as e:
            print(f"Ошибка при распознавании: {e}")
            await update.message.reply_text(f"Ошибка при распознавании: {str(e)}")

    except Exception as e:
        print(f"Общая ошибка: {e}")
        await update.message.reply_text(f"Произошла непредвиденная ошибка: {str(e)}")

    finally:
        # Очищаем временные файлы
        cleanup_files = []
        if audio_file:
            cleanup_files.append(audio_file)
        cleanup_files.extend([wav_file, ogg_file])

        for file_path in cleanup_files:
            try:
                if file_path and os.path.exists(file_path):
                    os.remove(file_path)
                    print(f"Удален файл: {file_path}")
            except Exception as e:
                print(f"Ошибка при удалении {file_path}: {e}")


async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
    user = update.message.from_user
    user_db_id = db.save_user(user)  # Сохраняем пользователя
    user_message = update.message.text

    print(f"Получено текстовое сообщение: {user_message}")

    try:
        response, audio_file = bot(user_message)
        db.log_interaction(user_db_id, user_message, response)  # Логируем взаимодействие

        await update.message.reply_text(response)

        if audio_file and os.path.exists(audio_file):
            with open(audio_file, 'rb') as voice:
                await update.message.reply_voice(voice)
            os.remove(audio_file)
            print(f"Аудиоответ отправлен и удален: {audio_file}")
        else:
            await update.message.reply_text("Не удалось сгенерировать аудиоответ.")

    except Exception as e:
        print(f"Ошибка при обработке текстового сообщения: {e}")
        await update.message.reply_text(f"Произошла ошибка: {str(e)}")


def main():
    db.init_db()  # Инициализация БД
    app = ApplicationBuilder().token(TOKEN).build()

    app.add_handler(CommandHandler("start", start))
    app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_message))
    app.add_handler(MessageHandler(filters.VOICE, handle_voice_message))

    print("✅ Бот запущен...")
    print(f"FFmpeg путь: {FFMPEG_PATH}")
    print(f"FFmpeg существует: {os.path.exists(FFMPEG_PATH)}")

    app.run_polling()


if __name__ == '__main__':
    main()