Update app.py

app.py

@@ -33,33 +33,6 @@ languages = {
 qa_pipeline = pipeline(task="question-answering", model="deepset/roberta-base-squad2")
 
 
-# Function to download audio from YouTube
-def download_audio_from_youtube(youtube_url, output_path="downloaded_audio.mp3"):
-    ydl_opts = {
-        'format': 'bestaudio/best',
-        'outtmpl': 'temp_audio.%(ext)s',
-        'postprocessors': [{
-            'key': 'FFmpegExtractAudio',
-            'preferredcodec': 'mp3',
-            'preferredquality': '192',
-        }],
-        'quiet': True,
-        'no_warnings': True,
-    }
-
-    try:
-        command = [
-            "yt-dlp",
-            "-x",  # extract audio only
-            "--audio-format", "mp3",  # specify mp3 format
-            "-o", output_path,  # specify output path
-            youtube_url  # YouTube URL
-        ]
-        subprocess.run(command, check=True, capture_output=True)
-        return output_path
-    except Exception as e:
-        return f"Error downloading audio: {e}"
-
 # Function to extract audio from video
 def extract_audio_from_video(video_file, output_audio="extracted_audio.mp3"):
     try:
@@ -75,23 +48,18 @@ languageG = None
 
 def content_input_update(content_type):
     visibility_map = {
-        "Audio Upload": (True, False, False),
-        "Video Upload": (False, False, True),
-        "YouTube Link": (False, True, False),
+        "Audio Upload": (True, False),
+        "Video Upload": (False, True),
     }
-    visible_audio, visible_youtube, visible_video = visibility_map.get(content_type, (False, False, False))
+    visible_audio, visible_video = visibility_map.get(content_type, (False, False))
     return (
         gr.update(visible=visible_audio),
-        gr.update(visible=visible_youtube),
         gr.update(visible=visible_video)
     )
 
 def transcribe_content(content_type, audio_path, youtube_link, video):
     if content_type == "Audio Upload" and audio_path:
         return whispermodel.transcribe(audio_path)["text"]
-    elif content_type == "YouTube Link" and youtube_link:
-        audio_file = download_audio_from_youtube(youtube_link)
-        return whispermodel.transcribe(audio_file)["text"]
     elif content_type == "Video Upload" and video:
         audio_file = extract_audio_from_video(video.name)
         return whispermodel.transcribe(audio_file)["text"]
@@ -128,7 +96,7 @@ def create_audio_summary(summary, language):
         return audio_path
     return None
 
-def main(content_type, audio_path, youtube_link, video, language, summarize):
+def main(content_type, audio_path, video, language, summarize):
     global transcription, languageG
     languageG = language
 
@@ -159,26 +127,24 @@ with gr.Blocks() as demo:
     )
 
     content_type = gr.Radio(
-        choices=["Audio Upload", "Video Upload", "YouTube Link"],
+        choices=["Audio Upload", "Video Upload"],
         label="Select Content Type",
         value="Audio Upload"
     )
 
     file_input = gr.Audio(label="Upload an Audio File", visible=True, type="filepath")
-    youtube_input = gr.Textbox(label="Enter YouTube Link", visible=False, placeholder="https://www.youtube.com/watch?v=example")
     video_input = gr.File(label="Upload a Video", visible=False, type="filepath")
 
     language = gr.Radio(choices=["Arabic", "English"], label="Preferred Language", value="English")
     summarize = gr.Checkbox(label="Summarize the content?")
 
     examples = [
-        ["Audio Upload", "audio-example.mp3", None, None, "English", True, True, 5],
-        ["Video Upload", None, None, "video-example.mp4", "Arabic", True, False, 3],
-        ["YouTube Link", None, "https://www.youtube.com/watch?v=J4RqCSD--Dg&ab_channel=LearnFree", None, "English", False, True, 2]
+        ["Audio Upload", "audio-example.mp3",  None, "English", True, True, 5],
+        ["Video Upload", None, "video-example.mp4", "Arabic", True, False, 3],
     ]
     gr.Examples(
         examples=examples,
-        inputs=[content_type, file_input, youtube_input, video_input, language, summarize],
+        inputs=[content_type, file_input, video_input, language, summarize],
         label="Try These Examples"
     )