Upload 2 files

app.py

@@ -0,0 +1,70 @@
+"""TED Talks Summarizer App."""
+
+from re import sub
+
+from gradio import Interface, Series, Textbox
+from requests import get
+
+
+def clean_text(text):
+    """Cleans subtitle text of ted talks.
+
+    Args:
+        text (str): subtitle of ted talk
+
+    Returns:
+        cleaned_text (str): cleaned version of subtitle text
+    """
+    # remove string inside parantheses (i.e appluse)
+    text = sub(r"\(.*\)", "", text)
+    # format text by splitting/removing new lines
+    text = text.split("\n")[1:]
+    # remove empty strings
+    text = list(filter(None, text))
+    # remove timestamps as they contains pattern of "-->"
+    cleaned_text = " ".join([x.strip() for x in text if "-->" not in x])
+    return cleaned_text
+
+
+def ted_talk_transcriber(link):
+    """Creates transcription of ted talks from url.
+
+    Args:
+        link (str): url link of ted talks
+
+    Returns:
+        cleaned_transcript (str): transcription of the ted talk
+    """
+    # request link of the talk
+    page = get(link)
+    # extract unique talk id to reach subtitle file
+    talk_id = str(page.content).split("project_masters/")[1].split("/")[0]
+    raw_text = get(
+        f"https://hls.ted.com/project_masters/{talk_id}/subtitles/en/full.vtt"
+    ).text
+    cleaned_transcript = clean_text(raw_text)
+    return cleaned_transcript
+
+
+transcriber = Interface(
+    ted_talk_transcriber,
+    "text",
+    "text",
+)
+
+summarizer = Interface.load(
+    "huggingface/pszemraj/long-t5-tglobal-base-16384-book-summary"
+)
+
+logo = "<center><img src='file/TED.png' width=180px></center>"
+
+Series(
+    transcriber,
+    summarizer,
+    inputs=Textbox(label="Type the TED Talks link"),
+    examples=[
+    "https://www.ted.com/talks/jen_gunter_the_truth_about_yeast_in_your_body"
+             ],
+    allow_flagging="never",
+    description=logo,
+).launch()