Update app.py

app.py

@@ -24,9 +24,13 @@ def extract_yt_audio(it, video_url, video_file):
       yt = YouTube(video_url, use_oauth=True)
       a = yt.streams.filter(only_audio=True).first()
       audio_file = a.download()
-      sample = AudioSegment.from_file(audio_file)
+      sample = AudioSegment.from_file(audio_file
+    elif it == 'URL':
+      response = requests.get(video_url)
+      video_data = BytesIO(response.content)
+      sample = AudioSegment.from_file(video_data, format="mp4")                                
     else:
-      sample = AudioSegment.from_file(video_url)
+      sample = AudioSegment.from_file(video_file)
     audio_path = 'audio.wav'
     # display(Audio(audio_path))
     sample.export(audio_path, format="wav")
@@ -99,9 +103,9 @@ def semantic_chunks(segs, max_chunk_length=15.0):
 
 def toggle_input_fields(input_type):
     if input_type == "URL":
-        return gr.update(visible=True), gr.update(visible=False)
+        return gr.update(visible=True), gr.update(visible=False), gr.update(visible=True), gr.update(visible=True)
     else:
-        return gr.update(visible=False), gr.update(visible=True)
+        return gr.update(visible=False), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
 
 def clear_all():
     return (gr.update(visible=True, value=""), gr.update(visible=True, value=""), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False))
@@ -123,13 +127,13 @@ with gr.Blocks() as demo:
     # input_url = gr.Textbox(label="Type-in the URL or File Location of the Video", value='sample.mp4')
     segments = gr.Textbox(visible=False)
     submit_btn_1 = gr.Button("Get the Transcript", visible=True)
-    audio = gr.Audio(visible=True, type="filepath", label='Play Audio')
-    transcript = gr.Textbox(visible=True, label='Transcript')
+    audio = gr.Audio(visible=False, type="filepath", label='Play Audio')
+    transcript = gr.Textbox(visible=False, label='Transcript')
     submit_btn_2 = gr.Button("Get the semantically Chuncked Segments", visible=False)
-    chunks = gr.Dataframe(visible=False, label = 'semantic Chunks')
+    chunks = gr.Dataframe(visible=False, label = 'Semantic Chunks')
     clear_btn = gr.Button("Clear")
 
-    input_type.change(fn=toggle_input_fields, inputs=input_type, outputs=[input_url, video_file])
+    input_type.change(fn=toggle_input_fields, inputs=input_type, outputs=[input_url, video_file, audio, transcript])
     submit_btn_1.click(fn=extract_yt_audio, inputs=[input_type, input_url, video_file], outputs=[transcript, submit_btn_2, segments, audio])
     # submit_btn_1.click(fn=extract_yt_audio, inputs=[input_url], outputs=[transcript, submit_btn_2, segments, audio])
     submit_btn_2.click(fn=semantic_chunks, inputs=[segments], outputs=[chunks])