import os
from modules import TTS, log_handler
from requests.exceptions import ReadTimeout
import sys
import io
import time
from config import BOT_SYSTEM_INSTRUCTION, SYSTEM_REFRESH, BARK_SMALL_MODELS, SPEAKER

# TTS Init
au_model = TTS('./assets/audio/', BARK_SMALL_MODELS)
print('TTS initialized successfully 👍')

# Log declaration
log = log_handler('./log/log.txt')

def send_message(chat_id, message):
    print(message)

def send_voice(chat_id, voice_path):
    print("Sending voice message...")

def handle_text(message):
    chat_id = message.chat.id
    if message.text != '/start':
        response = chat.query_bot(f'{message.text}', BOT_SYSTEM_INSTRUCTION, SYSTEM_REFRESH)
        send_message(chat_id, response)
        log.log_write(f'Date:{message.date} Chat_id: {chat_id} User: {message.text} Bot: {response}')
    else:
        send_message(chat_id, 'Hello!')
        log.log_write(f'Date:{message.date} Chat_id: {chat_id} User: {message.text} Bot: Hello!')

def send_audio(text, chat_id):
    try:
        au_model.to_audio(text, SPEAKER)
        send_voice(chat_id, './assets/audio/tts.ogg')
        os.remove("./assets/audio/tts.ogg")
        return
    except Exception as err:
        print('Error in send_audio function:\n'+str(err))
        log.log_write('Something went wrong with send_audio:\n'+str(err))
        send_message(chat_id, text) 
        return
        
if __name__ == "__main__":
    try:
        while True:
            try:
                polling()
            except ReadTimeout:
                print("A timeout occurred. Trying again...")
    except Exception as err:
        print('Error in bot.polling:\n' + str(err))
        log.fn_by_err(str(err))