Update app.py

app.py

@@ -3,36 +3,9 @@ select_model ="base" # ['tiny', 'base']
 model = whisper.load_model(select_model)
 
 
-import yt_dlp
-import ffmpeg
-import sys
-import uuid
-import re
-
-def extract_video_id(url):
-    # Regular expression to extract the video ID from different YouTube URL formats
-    pattern = r"(?:youtu\.be/|youtube(?:-nocookie)?\.com/(?:embed/|v/|shorts/|watch\?v=|watch\?.+&v=))([\w-]+)"
-    match = re.search(pattern, url)
-    if match:
-        return match.group(1)
-    return None
-    
-def download_audio(Youtube_Video_Link):
-  video_id = extract_video_id(Youtube_Video_Link)
-  yt_url = f"https://www.youtube.com/watch?v={video_id}"
-  random_uuid = str(uuid.uuid4())[:8]
-  ydl_opts = {
-    'format': 'bestaudio/best',
-  #    'outtmpl': 'output.%(ext)s',
-    'postprocessors': [{
-        'key': 'FFmpegExtractAudio',
-        'preferredcodec': 'mp3',
-    }],
-    "outtmpl": f'{random_uuid}',  # this is where you can edit how you'd like the filenames to be formatted
-  }
-  with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-    ydl.download([yt_url])
-  return f"{random_uuid}.mp3"
+
+
+
 
 def store_path_in_json(path, json_file_path="stored_paths.json"):
     # Create a dictionary with the path and timestamp
@@ -107,41 +80,20 @@ def convert_to_text(audio_path):
   store_path_in_json(audio_path)
   result = model.transcribe(audio_path,fp16=False)
   return result["text"]
-import os
-def audio_to_text(youtube_link,audio_path):
-  if len(youtube_link)>3:
-    audio_file_path=download_audio(youtube_link)
-    audio_file_path=os.getcwd()+"/"+audio_file_path
-    text=convert_to_text(audio_file_path)
-    return text
-  if os.path.exists(audio_path):
-    text=convert_to_text(audio_path)
-    return text
 
 
 import gradio as gr
 import os
 
 
-def transcribe_audio(youtube_link, audio_file):
-    if len(youtube_link)>=4:
-        result = audio_to_text(youtube_link, "None")
-    elif audio_file:
-        if os.path.exists(audio_file):
-            result = audio_to_text("None",audio_file)
-    else:
-        result = "Please provide a YouTube link or upload an audio file."
-
+def transcribe_audio(audio_file):
+    result=convert_to_text(audio_file)
     return result
+demo = gr.Interface(process_audio, 
+                    [gr.File(label="Upload Audio")], 
+                    [gr.Textbox(label="Transcribe")], 
+                   )
+
+demo.launch()
+
 
-iface = gr.Interface(
-    fn=transcribe_audio,
-    inputs=[
-        gr.Textbox(),
-        gr.File()
-    ],
-    outputs="text",
-    live=True
-)
-
-iface.launch()