Demiset.ipynb

{
  "cells": [
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "Z5ezJ4DiVPY5"
      },
      "outputs": [],
      "source": [
        "#@title <font color='#018ada'>[1] Clonar repositorio e instalar dependencias</font> { display-mode: \"form\" }\n",
        "%cd /content\n",
        "# Clonar el repositorio\n",
        "!git clone https://github.com/Omarleel/Demiset\n",
        "# Accede al proyecto\n",
        "%cd Demiset\n",
        "# Instalar dependencias\n",
        "!pip install -r requirements.txt"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "5P1nUl3XV0BN"
      },
      "outputs": [],
      "source": [
        "#@title <font color='#018ada'>[2] Parámetros de entrada</font> { display-mode: \"form\" }\n",
        "import os\n",
        "import time\n",
        "from Applications.AudioProcessing import AudioProcessing\n",
        "from Applications.NoiseReducer import NoiseReducer\n",
        "from Applications.RemoteFileDownloader import RemoteFileDownloader\n",
        "from Applications.SilenceRemover import SilenceRemover\n",
        "from Applications.VoiceExtractor import VoiceExtractor\n",
        "from Applications.Zipper import Zipper\n",
        "from Applications.GoogleDriveManager import GoogleDriveManager\n",
        "import zipfile\n",
        "\n",
        "def main():\n",
        "    input_folder = \"/content\" #@param {type:\"string\"}\n",
        "    input_url = 'https://www.twitch.tv/...' #@param {type:\"string\"}\n",
        "    output_folder = \"/content/Outputs\" #@param {type:\"string\"}\n",
        "    noise_threshold = 30 #@param {type:\"slider\", min:0, max:100, step:1}\n",
        "    ms_split = 15000 #@param {type:\"integer\"}\n",
        "    enhance_audio = False #@param {type:\"boolean\"}\n",
        "\n",
        "    remote_file_downloader = RemoteFileDownloader()\n",
        "    audio_processing = AudioProcessing()\n",
        "    noise_reducer = NoiseReducer()\n",
        "    voice_extractor = VoiceExtractor()\n",
        "    silence_remover = SilenceRemover()\n",
        "    zipper = Zipper()\n",
        "    google_drive_manager = GoogleDriveManager()\n",
        "\n",
        "    start_time = time.time()\n",
        "    # Descargar archivo remoto\n",
        "    ruta_archivo_descargado = remote_file_downloader.download(input_url, input_folder)\n",
        "    if ruta_archivo_descargado.endswith('.zip'):\n",
        "        # Obtener la ruta del directorio donde se extraerá el archivo ZIP\n",
        "        output_directory = os.path.dirname(ruta_archivo_descargado)\n",
        "        # Extraer el archivo ZIP en el mismo directorio\n",
        "        with zipfile.ZipFile(ruta_archivo_descargado, 'r') as zip_ref:\n",
        "            zip_ref.extractall(output_directory)\n",
        "    print(f\"Ruta del archivo descargado: {ruta_archivo_descargado} (Tiempo: {time.time() - start_time} segundos)\")\n",
        "\n",
        "    start_time = time.time()\n",
        "    # Combinar todos los audios en un solo archivo\n",
        "    ruta_audio_combinado = audio_processing.combine_audio(input_folder, output_folder)\n",
        "    print(f\"Ruta del audio combinado: {ruta_audio_combinado} (Tiempo: {time.time() - start_time} segundos)\")\n",
        "\n",
        "    start_time = time.time()\n",
        "    # Extraer la voz del audio\n",
        "    ruta_audio_voz = voice_extractor.extract_vocals(ruta_audio_combinado, output_folder)\n",
        "    print(f\"Ruta del audio de voz extraído: {ruta_audio_voz} (Tiempo: {time.time() - start_time} segundos)\")\n",
        "\n",
        "    start_time = time.time()\n",
        "    if noise_threshold > 0:\n",
        "        # Reducir ruido del audio\n",
        "        ruta_audio_sin_ruido = noise_reducer.reduce_noise(ruta_audio_voz, output_folder, noise_threshold)\n",
        "        print(f\"Ruta del audio sin ruido: {ruta_audio_sin_ruido} (Tiempo: {time.time() - start_time} segundos)\")\n",
        "    else:\n",
        "        ruta_audio_sin_ruido = ruta_audio_voz\n",
        "\n",
        "    start_time = time.time()\n",
        "    # Eliminar silencios del audio procesado\n",
        "    ruta_sin_silencio = silence_remover.remove_silence(ruta_audio_sin_ruido, output_folder)\n",
        "    print(f\"Ruta del audio sin silencios: {ruta_sin_silencio} (Tiempo: {time.time() - start_time} segundos)\")\n",
        "\n",
        "    start_time = time.time()\n",
        "    # Mejorar la calidad del audio\n",
        "    if enhance_audio:\n",
        "        # Mejorar la calidad del audio\n",
        "        ruta_audio_mejorado = audio_processing.enhance_audio(ruta_sin_silencio, output_folder)\n",
        "        print(f\"Ruta del audio mejorado: {ruta_audio_mejorado} (Tiempo: {time.time() - start_time} segundos)\")\n",
        "    else:\n",
        "        ruta_audio_mejorado = ruta_sin_silencio\n",
        "\n",
        "    start_time = time.time()\n",
        "    # Dividir archivo en audios de 15 segundos\n",
        "    output_folder_dataset = os.path.join(output_folder, \"dataset\")\n",
        "    ruta_audio_dividido = audio_processing.split_audio(ruta_audio_mejorado, output_folder_dataset, ms_split)\n",
        "    print(f\"Ruta del audio dividido: {ruta_audio_dividido} (Tiempo: {time.time() - start_time} segundos)\")\n",
        "\n",
        "    start_time = time.time()\n",
        "    # Comprimir dataset en un zip\n",
        "    ruta_dataset_comprimido = zipper.zip_files(ruta_audio_dividido, \"dataset.zip\")\n",
        "    output_remote_folder = os.path.join(\"Demiset\", output_folder.lstrip(\"/\"))\n",
        "    google_drive_manager.upload_file(ruta_dataset_comprimido, output_remote_folder)\n",
        "    print(f\"Dataset creado en: {ruta_dataset_comprimido} (Tiempo: {time.time() - start_time} segundos)\")\n",
        "\n",
        "if __name__ == \"__main__\":\n",
        "    main()"
      ]
    }
  ],
  "metadata": {
    "accelerator": "GPU",
    "colab": {
      "authorship_tag": "ABX9TyNS6xExcvDZomISubJg1zbd",
      "gpuType": "T4",
      "include_colab_link": true,
      "provenance": []
    },
    "kernelspec": {
      "display_name": "Python 3",
      "name": "python3"
    },
    "language_info": {
      "name": "python"
    }
  },
  "nbformat": 4,
  "nbformat_minor": 0
}