Update app.py

app.py

@@ -3,6 +3,10 @@ import time
 import json
 import os
 import gradio as gr
+from pytube import YouTube
+
+YT_embed_html1 = '<iframe width="560" height="315" src="https://www.youtube.com/embed/'
+YT_embed_html2 = '" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>'
 
 SIMBALS_GLOBAL_DB = 1
 SIMBALS_MAIN_DATABASE = 2
@@ -10,6 +14,26 @@ SIMBALS_MAIN_DATABASE = 2
 embed_html1 = '<iframe src="https://open.spotify.com/embed/track/'
 embed_html2 = '" width="300" height="380" frameborder="0" allowtransparency="true" allow="encrypted-media"></iframe>'
 
+
+def download_audio(id_video):
+    video = YouTube(id_video,use_oauth=True, allow_oauth_cache=True)
+    id = id_video.split("?v=")[-1][:11]
+    print("ID youtube ", id)
+    audio_streams = video.streams.filter(only_audio=True)
+    audio_stream = audio_streams[0]
+    audio_file = audio_stream.download()
+    audio_parts = audio_file.split('/')[-1]
+    audio_title = '.'.join(audio_parts.split('.')[:-1])
+    embed_html_all = YT_embed_html1 + id + YT_embed_html2
+    return audio_file, audio_file, embed_html_all
+
+def process_url(input_path):
+    audio_file, audio_file, embed_html_all = download_audio(input_path)
+    return process(audio_file, embed_html_all)
+
+def process_file(input_path):
+    return process(input_path, '')
+
 def process(input_path):
     # setup the client
     client = SimbalsAPIClient(os.environ['TOKEN'], debug=True)
@@ -50,12 +74,46 @@ def process(input_path):
                 return "Not found. Please try again."
         time.sleep(1)
     
-
+'''
 demo = gr.Interface(fn=process,
                     inputs=gr.Audio(type="filepath"),
                     outputs=[gr.outputs.Textbox(label="Identification"),"html"]
                     #examples=example_list,
                     #cache_examples=False
                     )
+'''
+
+with gr.Blocks() as demo:
+    
+    with gr.Row():
+
+        with gr.Column():
+
+            with gr.Row():
+                #gr.HTML(embed_html)
+                html = gr.HTML()
+            
+            with gr.Row():
+                with gr.Column():
+                    audio_url_input = gr.Textbox(placeholder='YouTube video URL', label='YouTube video URL')
+                    analyze_url_btn = gr.Button('Search from URL')
+
+            with gr.Row():
+                with gr.Column():
+                    audio_input_file = gr.Audio(type="filepath", label='Audio Input')
+                    analyze_file_btn = gr.Button('Search from file')
+
+            with gr.Row():
+                with gr.Column():
+                    top1 = gr.Audio(label="top1", show_label=True)
+                    id = gr.outputs.Textbox(label="Identification", show_label=True)        
+                    id_html = html = gr.HTML()
+                    
+    analyze_url_btn.click(process_url, inputs=[audio_url_input], 
+                      outputs=[id, id_html])
+
+    analyze_file_btn.click(process_file, inputs=[audio_input_file], 
+                      outputs=[id, id_html])
+    
 demo.queue(concurrency_count=2)
 demo.launch(debug=True)