Update app.py

app.py

@@ -7,6 +7,9 @@ def fetch(url, name, ext):
   elif ext == "webm zip":
     opts = ""
     ext = "webm"
+  elif ext == "transcript":
+    opts = "--write-auto-sub --skip-download"
+    ext = "vtt"
   else:
     opts = {
       "mp3": "-f \"ba\" -x --audio-format mp3 --audio-quality 1",
@@ -23,13 +26,21 @@ def fetch(url, name, ext):
   if ext == "webm" or ext == "webm zip":
     os.system(f"zip -9 {filename}.zip {filename}")
     return filename.rstrip(".webm") + ".zip"
+
+  if ext == "transcript":
+    with open(filename, "r") as file:
+      text = file.read()
+      text = text.replace("\n\n", "\n").replace("WEBVTT\n\n", "")
+      with open(f"{name}.txt", "w") as file:
+        file.write(text)
+    return f"{name}.txt"
   return filename
   
 gr.Interface(
   fetch, 
   [gr.Textbox(label="Media link", placeholder="URL goes here..."),
    gr.Textbox(label="File name", placeholder="best-vid"),
-   gr.Dropdown(["mp3", "mp4", "webm", "webm zip"], value="mp4", label="File type")],
+   gr.Dropdown(["mp3", "mp4", "webm", "webm zip", "transcript"], value="mp4", label="File type")],
   gr.outputs.File(label="Download!"),
   description="Download web media! Works on most sites like YouTube, Reddit, Twitter, Instagram, etc.",
   enable_queue=True