main.py

import os, time, threading, logging, multiprocessing, uuid
from logger import create_logger; logger = create_logger('AIStreamer')
from flaskServer import server
from twitch import Bot
from utils import Config
from dotenv import load_dotenv; load_dotenv()
from llmEngine import gen_gpt
from ttsEngine import gen_elevenlabs, gen_RVC, gen_Edge, gen_VITS
import wav2lipEngine, asyncio, random

if __name__ == "__main__":

    twitch_bot = Bot()
    twitch_bot_thread = threading.Thread(target=twitch_bot.run)
    twitch_bot_thread.daemon = True
    twitch_bot_thread.start()

    # flask_server_process = multiprocessing.Process(target=server.runServer)
    # flask_server_process.daemon = True
    # flask_server_process.start()

    while True:
        try:
            message = twitch_bot.message_queue.dequeue()
            if not message:
                time.sleep(1)
                continue

            logger.info(f"Message received: {message.content}")
            logger.info(f"Generating answer...")
            answer = gen_gpt(message)
            logger.info(f"Answer generated: {answer}")
            logger.info(f"Generating audio...")
            ttsType = Config.get_value("tts_type")
            if ttsType == "EdgeTTS":
                asyncio.run(gen_Edge(answer))
            elif ttsType == "VITS":
                gen_VITS(answer)
            elif ttsType == "ElevenLabs":
                gen_elevenlabs(answer)
            if Config.get_value("rvc_name"):
                gen_RVC()
            logger.info(f"Audio generated")
            logger.info(f"Generating video...")
            videoId = str(uuid.uuid4())[:8]
            wav2lipEngine.performInference(videoId)
            with open(f"flaskServer/static/queue/{videoId}.txt", "w") as writer:
                writer.write(f"{message.author.name} : {message.content}")
            logger.info(f"Video generated")
        except Exception as e:
            logger.warning(f"Error while generating {e}")