Update app.py

app.py

@@ -1,3 +1,5 @@
+from __future__ import unicode_literals
+
 import os
 import glob
 import json
@@ -422,6 +424,152 @@ def use_microphone(microphone):
     else:
         return gr.Audio.update(source="upload")
 
+
+
+# Audio Tool Functions
+
+# cvt audio
+
+from pydub import AudioSegment
+def convert_audio(url,title):
+
+  # Mendefinisikan path untuk file audio
+  input_path = url
+  file_name = os.path.basename(input_path)
+  filename = os.path.splitext(file_name)[0]
+
+
+  parent_dir = os.path.dirname(url)
+  new_path = os.path.relpath(parent_dir, "")
+
+
+
+  output_path = f'youtubeaudio/{title}_converted.mp3'
+
+  # Mengkonversi file audio WAV menjadi MP3 menggunakan pydub
+  sound = AudioSegment.from_wav(input_path)
+  sound.export(output_path, format="mp3")
+
+  # Mengecek apakah file audio MP3 sudah tersimpan
+  if os.path.isfile(output_path):
+      # return output_path
+      return "sukses"
+  else:
+      return "Konversi gagal"
+
+
+# Fungsi play Audio
+def play_audio(url):
+
+  file_path = url
+  file_name = os.path.basename(file_path)
+  filename = os.path.splitext(file_name)[0]
+
+  original_path = f"/content/youtubeaudio/{filename}.wav"
+  vocal_path = f"/content/separated/htdemucs/{filename}/vocals.wav"
+  instrument_path = f"/content/separated/htdemucs/{filename}/no_vocals.wav"
+
+  return url
+
+
+
+# Fungsi download audio
+
+import yt_dlp
+import ffmpeg
+import sys
+
+
+def download_audio(title, url):
+
+  ydl_opts = {
+    'format': 'bestaudio/best',
+    #   'outtmpl': 'output.%(ext)s',
+    'postprocessors': [{
+        'key': 'FFmpegExtractAudio',
+        'preferredcodec': 'wav',
+    }],
+    "outtmpl": f'youtubeaudio/{title}',  # this is where you can edit how you'd like the filenames to be formatted
+  }
+
+  with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+    # url = "https://www.youtube.com/watch?v=LCcNtQuhUgg" #@param {type:"string"}
+    ydl.download([url])
+  # return f"/content/youtubeaudio/{title}.wav"
+  # return f"/content/youtubeaudio/adudio.wav"
+  return "sukses"
+
+#fungsi download video
+def download_video(url, resolution):
+
+  from pytube import YouTube
+
+  yt = YouTube(url)
+  try:
+    stream_check = yt.streams.filter(res=f"{resolution}p")
+    if len(stream_check) > 0:
+      stream = yt.streams.filter(file_extension='mp4', res=f'{resolution}p').first()
+    else:
+      stream = yt.streams.get_highest_resolution()
+  except Exception as e:
+      return "error"
+
+
+
+
+  folder_path = 'youtubevideo'
+  if not os.path.exists(folder_path):
+      os.makedirs(folder_path)
+
+  file_path = os.path.join(folder_path, stream.default_filename)
+  stream.download(output_path=folder_path, filename=stream.default_filename)
+
+  return "sukses"
+
+
+# fungsi split audio
+def split_audio(url):
+  import subprocess
+
+  command = f"demucs --two-stems=vocals {url}"
+  result = subprocess.run(command.split(), stdout=subprocess.PIPE)
+  print(result.stdout.decode())
+  return "sukses"
+
+
+def aio(title, yt_url):
+  download_status = download_audio(title, yt_url)
+
+  audio_url = f"youtubeaudio/{title}.wav"
+  split_status = split_audio(audio_url)
+
+  vocal_url = f"separated/htdemucs/{title}/vocals.wav"
+  no_vocal_url = f"separated/htdemucs/{title}/no_vocals.wav"
+
+  vocal_convert_status = convert_audio(vocal_url, title+"_vocal")
+  no_vocal_convert_status = convert_audio(no_vocal_url, title+"_instrumen")
+
+  import os
+
+  # specify old file path name
+  # old_vocal_name = f"separated/htdemucs/{title}/vocals_converted.mp3"
+  # old_instrumen_name = f"separated/htdemucs/{title}/no_vocals_converted.mp3"
+
+
+  # Specify the new file path and name
+  # new_vocal_name = f"separated/htdemucs/{title}/{title}_vocal.mp3"
+  # new_instrumen_name = f"separated/htdemucs/{title}/{title}_instrumen.mp3"
+
+  # Rename the file separated
+  # os.rename(old_vocal_name, new_vocal_name)
+  # os.rename(old_instrumen_name, new_instrumen_name)
+
+
+  return "sukses"
+
+
+
+
 if __name__ == '__main__':
     load_hubert()
     categories = load_model()
@@ -732,4 +880,17 @@ if __name__ == '__main__':
                                 tts_voice
                             ]
                         )
-        app.queue(concurrency_count=5, max_size=50, api_open=config.api).launch(share=config.share)
+        # Audio tool
+
+        with gr.Tab("AIO"):
+            with gr.Row():
+              with gr.Column():
+                aio_input = [gr.Textbox(label = "title"), gr.Textbox(label = "Youtube Url")]
+                aio_button = gr.Button("Procces")
+              with gr.Column():
+                aio_output =[gr.Textbox(label = "Status Output")]
+
+        aio_button.click(aio, inputs=aio_input, outputs=aio_output)
+
+
+        app.queue(concurrency_count=5, max_size=50, api_open=config.api).launch(share=config.share, debug=True)