Update app.py

app.py

@@ -1,85 +1,78 @@
 import os
-import logging
 import subprocess
-from tempfile import mkstemp
-from telegram import Update
-from telegram.ext import Application, CommandHandler, MessageHandler, filters, ContextTypes
+import telebot
+from telebot import types
 
-# Настройки
 TOKEN = "7709328099:AAHvvz2Dqkzb2c0lNh9OViTBnUoW6ZcpreA"
 ALLOWED_EXTENSIONS = {'mp4', 'avi', 'mov', 'mkv'}
 TEMP_DIR = "temp_files"
 
-# Настройка логирования
-logging.basicConfig(
-    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
-    level=logging.INFO
-)
+bot = telebot.TeleBot(TOKEN)
 
-async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    await update.message.reply_text(
-        "Привет! Отправь мне видео, и я интерполирую кадры до нужного FPS. "
-        "Используй команду /help для справки."
-    )
+# Хранилище данных пользователя
+user_data = {}
 
-async def help_command(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    await update.message.reply_text(
-        "Как использовать бота:\n"
-        "1. Отправь мне видео файл\n"
+@bot.message_handler(commands=['start', 'help'])
+def send_welcome(message):
+    bot.reply_to(message, 
+        "Привет! Отправь мне видео, и я интерполирую кадры до нужного FPS.\n"
+        "Инструкция:\n"
+        "1. Отправь видео файл\n"
         "2. Укажи желаемый FPS (целое число)\n"
-        "3. Жди результат обработки\n\n"
-        "Ограничения: максимальный размер файла 20MB"
+        "3. Жди результат обработки"
     )
 
-async def handle_video(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    if not update.message.video:
-        await update.message.reply_text("Пожалуйста, отправьте видео файл")
-        return
+@bot.message_handler(content_types=['video'])
+def handle_video(message):
+    try:
+        # Скачивание файла
+        file_info = bot.get_file(message.video.file_id)
+        downloaded_file = bot.download_file(file_info.file_path)
+        
+        # Создание временной директории
+        os.makedirs(TEMP_DIR, exist_ok=True)
+        
+        # Сохранение файла
+        ext = file_info.file_path.split('.')[-1] if file_info.file_path else 'mp4'
+        input_path = os.path.join(TEMP_DIR, f"input_{message.chat.id}.{ext}")
+        
+        with open(input_path, 'wb') as new_file:
+            new_file.write(downloaded_file)
+        
+        user_data[message.chat.id] = {'input_path': input_path}
+        
+        # Запрос FPS
+        msg = bot.send_message(message.chat.id, "Видео получено. Введите желаемый FPS:")
+        bot.register_next_step_handler(msg, process_fps)
 
-    # Создаем временную директорию
-    os.makedirs(TEMP_DIR, exist_ok=True)
-    
-    # Сохраняем файл
-    video_file = await update.message.video.get_file()
-    ext = video_file.file_path.split('.')[-1] if video_file.file_path else 'mp4'
-    input_path = os.path.join(TEMP_DIR, f"input_{update.update_id}.{ext}")
-    await video_file.download_to_drive(input_path)
-    
-    context.user_data['input_path'] = input_path
-    await update.message.reply_text("Видео получено. Теперь введите желаемый FPS:")
+    except Exception as e:
+        bot.reply_to(message, f"Ошибка: {str(e)}")
 
-async def handle_fps(update: Update, context: ContextTypes.DEFAULT_TYPE):
+def process_fps(message):
     try:
-        target_fps = int(update.message.text)
+        chat_id = message.chat.id
+        target_fps = int(message.text)
+        
         if target_fps <= 0:
-            raise ValueError
-    except ValueError:
-        await update.message.reply_text("Пожалуйста, введите корректное целое число FPS")
-        return
+            raise ValueError("FPS должен быть положительным числом")
+        
+        if chat_id not in user_data or not os.path.exists(user_data[chat_id]['input_path']):
+            bot.send_message(chat_id, "Ошибка: видео не найдено")
+            return
 
-    input_path = context.user_data.get('input_path')
-    if not input_path or not os.path.exists(input_path):
-        await update.message.reply_text("Ошибка: видео не найдено")
-        return
+        input_path = user_data[chat_id]['input_path']
+        output_path = os.path.join(TEMP_DIR, f"output_{chat_id}.mp4")
 
-    try:
         # Получаем исходный FPS
         cmd = f"ffprobe -v error -select_streams v:0 -show_entries stream=r_frame_rate -of default=noprint_wrappers=1:nokey=1 {input_path}"
         original_fps = subprocess.check_output(cmd, shell=True).decode().strip().split('/')
         original_fps = float(original_fps[0])/float(original_fps[1])
         
         if target_fps <= original_fps:
-            await update.message.reply_text(f"Целевой FPS должен быть больше исходного ({original_fps:.2f})")
+            bot.send_message(chat_id, f"Целевой FPS должен быть больше исходного ({original_fps:.2f})")
             return
-    except Exception as e:
-        await update.message.reply_text("Ошибка при определении исходного FPS видео")
-        logging.error(e)
-        return
 
-    # Обработка видео
-    output_path = os.path.join(TEMP_DIR, f"output_{update.update_id}.mp4")
-    
-    try:
+        # Обработка видео
         cmd = [
             'ffmpeg',
             '-i', input_path,
@@ -94,28 +87,22 @@ async def handle_fps(update: Update, context: ContextTypes.DEFAULT_TYPE):
         subprocess.run(cmd, check=True, timeout=300)
         
         # Отправка результата
-        await update.message.reply_video(video=open(output_path, 'rb'))
+        with open(output_path, 'rb') as video_file:
+            bot.send_video(chat_id, video_file)
         
+    except ValueError:
+        bot.send_message(chat_id, "Пожалуйста, введите корректное целое число FPS")
     except subprocess.TimeoutExpired:
-        await update.message.reply_text("Обработка заняла слишком много времени")
+        bot.send_message(chat_id, "Обработка заняла слишком много времени")
     except Exception as e:
-        await update.message.reply_text("Ошибка при обработке видео")
-        logging.error(e)
+        bot.send_message(chat_id, f"Ошибка при обработке: {str(e)}")
     finally:
-        # Удаление временных файлов
-        for path in [input_path, output_path]:
-            if path and os.path.exists(path):
-                os.remove(path)
-
-def main():
-    application = Application.builder().token(TOKEN).build()
-
-    application.add_handler(CommandHandler("start", start))
-    application.add_handler(CommandHandler("help", help_command))
-    application.add_handler(MessageHandler(filters.VIDEO, handle_video))
-    application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_fps))
-
-    application.run_polling()
+        # Очистка временных файлов
+        if chat_id in user_data:
+            for path in [user_data[chat_id].get('input_path'), output_path]:
+                if path and os.path.exists(path):
+                    os.remove(path)
+            del user_data[chat_id]
 
 if __name__ == '__main__':
-    main()
+    bot.polling(none_stop=True)