import streamlit as st
import tempfile

st.set_page_config(
    page_title="PC-AVS",
    layout="wide",
    initial_sidebar_state="collapsed"
)

st.header("PC-AVS")

col1, col2 = st.columns(2)
result = None

# Upload video and audio
with col1:
    st.header("Video Input")
    video_file = st.file_uploader("Upload video", type=["mp4", "avi", "mov"])
    if video_file is not None:
        # Show video
        temp_video = tempfile.NamedTemporaryFile(delete=False)
        temp_video.write(video_file.read())
        st.video(temp_video.name)
    else:
        st.write("No video have been uploaded yet.")
    
    st.header("Audio Input")
    audio_file = st.file_uploader("Upload audio", type=["mp3", "wav", "ogg"])
    if audio_file is not None:
        temp_audio = tempfile.NamedTemporaryFile(delete=False)
        temp_audio.write(audio_file.read())
        st.audio(temp_audio.name)
    else:
        st.write("No audio have been uploaded yet.")
    if st.button("Submit", key="replay-button"):
        if video_file is not None and audio_file is not None:
            result = video_file
        else:
            st.write("Please upload video and audio")
# Show result
with col2:
    st.header("Result")
    if result is not None:
        # Show video
        result_video = tempfile.NamedTemporaryFile(delete=False)
        result_video.write(result.read())
        st.video(result_video.name)
    else:
        st.write("No result")