Update app.py

app.py

@@ -1,23 +1,37 @@
 import os
+os.system('pip uninstall vidfetch')
+os.system('pip install -U https://github.com/dailingx/VidFetch/archive/master.zip')
+os.system('pip install --upgrade google-api-python-client')
+
 import sys
 import gradio as gr
+from vidfetch.website.youtube import YoutubeVideoDataset
 
 
 def fetch(
-    key_word: str
+    key_word: str,
+    dev_key: str
 ):
+    youtube_video_dataset = YoutubeVideoDataset(
+        root_dir="./",
+        google_cloud_developer_key=dev_key,
+        search_keyword=key_word,
+        video_max_num=1
+    )
+    youtube_video_dataset.download()
     return key_word
-    
+
 
 with gr.Blocks() as demo:
     gr.Markdown('''OpenVideo Youtube fetch demo''')
     with gr.Row():
         with gr.Column():
             kw_input_text = gr.Text(label='Keyword')
+            dev_key_input_text = gr.Text(label='Google Cloud Developer Key')
             fetch_btn = gr.Button("Fetch")
         result = gr.Text()
-            
-    fetch_btn.click(fn=fetch, inputs=[kw_input_text], outputs=[result])
+
+    fetch_btn.click(fn=fetch, inputs=[kw_input_text, dev_key_input_text], outputs=[result])
 
 
 if __name__ == "__main__":