import streamlit as st
import os
import sys
import torch
path_to_add = os.path.join(os.path.dirname(__file__), "Wav2Lip")
if path_to_add not in sys.path:
    sys.path.insert(0, path_to_add)
from avatar import Avatar


if 'is_initialized' not in st.session_state:
    st.session_state.avatar = Avatar()
    st.session_state.avatar.export_video = False
    st.session_state.avatar.load_model("checkpoint/wav2lip_gan.pth")
    print("load model finished")
    st.session_state.avatar.device = 'cuda' if torch.cuda.is_available() else 'cpu'
    print(st.session_state.avatar.device)
    st.session_state.avatar.output_audio_path = "audio/"
    st.session_state.avatar.output_audio_filename = "result.wav"
    st.session_state.avatar.temp_lip_video_no_voice_path = "temp/"
    st.session_state.avatar.temp_lip_video_no_voice_filename = "result.avi"
    st.session_state.avatar.output_video_path = "results/"
    st.session_state.avatar.output_video_name = "result_voice.mp4"

    st.session_state['is_initialized'] = True


from avatar import Avatar
# Create a text input box and store the input in a variable
user_input = st.text_input("Enter your text:")
st.session_state.avatar.dir_clean_up()
# Display the entered text
st.write("You entered:", user_input)