import streamlit as st
import os
import subprocess
import ffmpeg  # Import the ffmpeg-python library

# Directories for uploads and processing
UPLOAD_FOLDER = 'uploads'
os.makedirs(UPLOAD_FOLDER, exist_ok=True)

def save_uploaded_file(uploaded_file):
    """Save uploaded file to the uploads folder, keeping its original name or creating a unique name if it already exists."""
    file_path = os.path.join(UPLOAD_FOLDER, uploaded_file.name)
    # If a file with the same name already exists, generate a unique name
    if os.path.exists(file_path):
        base, ext = os.path.splitext(uploaded_file.name)
        i = 1
        while os.path.exists(os.path.join(UPLOAD_FOLDER, f"{base}_{i}{ext}")):
            i += 1
        file_path = os.path.join(UPLOAD_FOLDER, f"{base}_{i}{ext}")
    with open(file_path, "wb") as f:
        f.write(uploaded_file.getbuffer())
    return file_path

def run_ffmpeg(command):
    """Run the FFmpeg command in the uploads folder and capture the output."""
    try:
        # Change working directory to the uploads folder
        result = subprocess.run(command, shell=True, capture_output=True, text=True, cwd=UPLOAD_FOLDER)
        return result.stdout, result.stderr
    except Exception as e:
        return "", str(e)

# Streamlit UI
st.title("FFmpeg Command Runner")

# File uploader
uploaded_file = st.file_uploader("Upload a file", type=None)

# Text area for FFmpeg commands
ffmpeg_command = st.text_area("Enter your FFmpeg command (use '<input_file>' for the uploaded file name)", "")

# Execute button
if st.button("Run FFmpeg Command"):
    if uploaded_file:
        # Save the uploaded file
        uploaded_file_path = save_uploaded_file(uploaded_file)
        st.write(f"File saved at: {uploaded_file_path}")

        # Replace placeholder with the uploaded file name (just the file name, not the full path)
        uploaded_file_name = os.path.basename(uploaded_file_path)
        command_to_run = ffmpeg_command.replace("<input_file>", uploaded_file_name)

        # Run the command
        stdout, stderr = run_ffmpeg(command_to_run)

        # Display the logs
        st.text_area("FFmpeg Output", stdout)
        st.text_area("FFmpeg Errors", stderr)

        # Check if an output file is generated and show the video player
        for file in os.listdir(UPLOAD_FOLDER):
            if file.endswith(".mp4") or file.endswith(".mkv") or file.endswith(".avi"):  # Add other formats as needed
                output_path = os.path.join(UPLOAD_FOLDER, file)
                st.video(output_path)

    else:
        st.write("Please upload a file before running the command.")