Update app.py

app.py

@@ -1,76 +1,53 @@
 import gradio as gr
+import telebot
 from transformers import AutoTokenizer, VitsModel
 import torch
 import scipy.io.wavfile
 import threading
-import asyncio
-import os
-from telegram import Update
-from telegram.ext import ApplicationBuilder, ContextTypes, MessageHandler, filters
+import time
 
-# 1. Загрузка модели для чеченского языка
+# 1. Настройка модели
 model_name = "facebook/mms-tts-che"
 model = VitsModel.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
-def tts_chechen_logic(text):
-    # Очистка и синтез
-    inputs = tokenizer(text.strip().lower(), return_tensors="pt")
+def tts_logic(text):
+    inputs = tokenizer(text.lower(), return_tensors="pt")
     with torch.no_grad():
         output = model(**inputs).waveform
-    
-    sampling_rate = model.config.sampling_rate
-    audio_data = output.squeeze().numpy()
-    file_path = "voice.wav"
-    scipy.io.wavfile.write(file_path, rate=sampling_rate, data=audio_data)
-    return sampling_rate, audio_data, file_path
+    file_path = "audio.wav"
+    scipy.io.wavfile.write(file_path, rate=model.config.sampling_rate, data=output.squeeze().numpy())
+    return file_path
 
-# --- БЛОК TELEGRAM БОТА ---
-# Твой новый токен
-TG_TOKEN = "8287698372:AAEwoOWgQWHbeyDmJdN0B-OpKpy7-TARHMg"
+# 2. Настройка бота (теперь на pyTelegramBotAPI)
+TOKEN = "8287698372:AAEwoOWgQWHbeyDmJdN0B-OpKpy7-TARHMg"
+bot = telebot.TeleBot(TOKEN)
 
-async def handle_tg_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    if not update.message or not update.message.text:
-        return
-    
-    print(f"📩 Текст для бота: {update.message.text}")
+@bot.message_handler(func=lambda message: True)
+def echo_all(message):
     try:
-        _, _, file_path = tts_chechen_logic(update.message.text)
-        with open(file_path, 'rb') as audio:
-            await context.bot.send_voice(chat_id=update.effective_chat.id, voice=audio)
+        path = tts_logic(message.text)
+        with open(path, 'rb') as audio:
+            bot.send_voice(message.chat.id, audio)
     except Exception as e:
-        print(f"❌ Ошибка: {e}")
-
-def run_bot_in_background():
-    try:
-        # Настройка асинхронности для контейнера
-        loop = asyncio.new_event_loop()
-        asyncio.set_event_loop(loop)
-        
-        app = ApplicationBuilder().token(TG_TOKEN).build()
-        app.add_handler(MessageHandler(filters.TEXT & (~filters.COMMAND), handle_tg_message))
-        
-        print("🚀 Бот запущен и слушает Telegram!")
-        app.run_polling(drop_pending_updates=True, close_loop=False)
-    except Exception as e:
-        print(f"🔥 Критическая ошибка: {e}")
+        print(f"Ошибка: {e}")
 
 # Запуск бота в отдельном потоке
-threading.Thread(target=run_bot_in_background, daemon=True).start()
-# --------------------------
-
-# Интерфейс Gradio (сайт)
-def gradio_interface(text):
-    sr, audio, _ = tts_chechen_logic(text)
-    return (sr, audio)
-
-demo = gr.Interface(
-    fn=gradio_interface,
-    inputs=gr.Textbox(label="Чеченский текст"),
-    outputs=gr.Audio(label="Голос"),
-    title="NoxchiynAz"
-)
-
-if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
-    
+def run_bot():
+    while True:
+        try:
+            print("🚀 Бот запущен...")
+            bot.polling(none_stop=True, interval=0)
+        except Exception as e:
+            print(f"Перезапуск бота через 5 сек... {e}")
+            time.sleep(5)
+
+threading.Thread(target=run_bot, daemon=True).start()
+
+# 3. Интерфейс сайта (чтобы HF не выключал сервер)
+def web_interface(text):
+    path = tts_logic(text)
+    return path
+
+demo = gr.Interface(fn=web_interface, inputs="text", outputs="audio")
+demo.launch()