import gradio as gr
import logging
import os
from TTS.api import TTS
import time

tts = TTS(model_name="voice_conversion_models/multilingual/vctk/freevc24", progress_bar=False, gpu=False)
logging.basicConfig(level=logging.INFO)
count = 0

def main():
    global count
    count += 1
    if count > 50:
        time.sleep(5)
        os.system("rm -R /tmp/*")
        count = 0
        
    with gr.Blocks() as demo:

        with gr.Row():
            with gr.Column(variant="panel"):
                src_audio_mic = gr.Audio(source="microphone", label="Record your voice")
                src_audio_file = gr.Audio(
                    source="upload", type="filepath", label="Or upload audio to convert"
                )

            with gr.Column(variant="panel"):
                tgt_audio_file = gr.Audio(
                    source="upload", type="filepath", label="Select audio with target voice"
                )

        with gr.Row():
            convert_btn = gr.Button("Convert")
        with gr.Row():
            result_audio = gr.Audio()

        
        def voice_conversion(src_from_mic_, src_from_file_, tgt_from_file_):
            """
            helper function which checks where source come from
            """
            src_ = None
            if src_from_mic_:
                src_ = src_from_mic_
            elif src_from_file_:
                src_ = src_from_file_
            tgt_ = tgt_from_file_
            if not src_ or not tgt_:
                logging.warning("source or target are not provided")
                return

            print(src_)
            print(tgt_)
            tts.voice_conversion_to_file(source_wav=src_, target_wav=tgt_, file_path="output.wav")
            return "output.wav"
        
        convert_btn.click(
            voice_conversion,
            inputs=[src_audio_mic, src_audio_file, tgt_audio_file],
            outputs=result_audio,
        )

    demo.queue(concurrency_count=1).launch(show_api=False, show_error=True)


if __name__ == "__main__":
    main()