import logging
import tempfile
from io import BytesIO

import requests
from moviepy.editor import VideoFileClip
from tqdm import tqdm


def extract_audio(video_path):
    try:
        with VideoFileClip(video_path) as video:
            audio = video.audio
            audio_path = video_path.replace('.mp4', '.mp3')
            audio.write_audiofile(audio_path, codec='mp3')
        return audio_path
    except Exception as e:
        logging.error(f"Failed to extract audio: {e}")
        return None


def download_mp4_and_extract_audio(link: str):
    """Download an MP4 file from a given link and return the path to the video and audio files."""
    logging.info("Starting the download of the MP4 file...")
    video_content = BytesIO()
    try:
        r = requests.get(link, stream=True)
        r.raise_for_status()

        total_size = int(r.headers.get('content-length', 0))

        with tqdm(total=total_size, unit='B', unit_scale=True, desc="Downloading...") as bar:
            for data in r.iter_content(chunk_size=1024):
                bar.update(len(data))
                video_content.write(data)
            video_content.seek(0)

        with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as temp_video_file:
            temp_video_file.write(video_content.getvalue())
            temp_video_file_path = temp_video_file.name

        logging.info("Starting audio extraction in a separate process...")
        audio_path = extract_audio(temp_video_file_path)

        logging.info("Download and audio extraction completed")
        return temp_video_file_path, audio_path

    except requests.exceptions.RequestException as e:
        logging.error(f"Request Error: {e}")
        return None, None
    except Exception as e:
        logging.error(f"Failed to download MP4 and extract audio: {e}")
        return None, None
    finally:
        video_content.close()