Python-Youtube-Transcriber/main.py at main · m0L3cuL3/Python-Youtube-Transcriber · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83

from youtube_transcript_api import YouTubeTranscriptApi as yta
import re
import dearpygui.dearpygui as dpg
from dearpygui.dearpygui import *


author = 'Sean Baang'
version = '1.0.1'

dpg.create_context()

def get_transcription():
    data = yta.get_transcript(get_value("Youtube##ID"))
    transcript = ''

    for value in data:
        for key,val in value.items():
            if key == 'text':
                transcript += val

    l = transcript.splitlines()
    final_tra = " ".join(l)

    #### SAVE TO FILE ####
    # file = open("transcript.txt", 'w')
    # file.write(final_tra)
    # file.close()
    ######################

    # Display Transcript
    dpg.add_text(final_tra, parent="Main Window", wrap=500, tag="script")

def download_transcript():
    file = open("transcript.txt", 'w')
    file.write(get_value("script"))
    file.close()

def clear_all():
    dpg.delete_item("script")

with dpg.window(tag="Main Window", pos=(100,100)):
    dpg.add_text("Youtube Transcriber - {0}".format(version))
    dpg.add_text("Coded by {0}".format(author))

    dpg.add_spacer()
    dpg.add_separator()
    dpg.add_spacer()

    dpg.add_text("Youtube ID:")
    it1 = dpg.add_input_text(tag="Youtube##ID")
    dpg.add_same_line()
    b1 = dpg.add_button(label="Get Transcript", callback=get_transcription)
    b2 = dpg.add_button(label="Download Transcript", callback=download_transcript)
    dpg.add_same_line()
    bclear = dpg.add_button(label="Clear", callback=clear_all)

    dpg.add_spacer()
    dpg.add_separator()
    dpg.add_spacer()

with dpg.theme() as item_theme:
    with dpg.theme_component(dpg.mvAll):
        dpg.add_theme_color(dpg.mvThemeCol_FrameBg, (135, 135, 135), category=dpg.mvThemeCat_Core)
        dpg.add_theme_style(dpg.mvStyleVar_FrameRounding, 12, category=dpg.mvThemeCat_Core)
        dpg.add_theme_style(dpg.mvStyleVar_FramePadding, 12, 5)

dpg.bind_item_theme(b1, item_theme)
dpg.bind_item_theme(b2, item_theme)
dpg.bind_item_theme(bclear, item_theme)
dpg.bind_item_theme(it1, item_theme)

dpg.create_viewport(
    title="Youtube Transciber",
    width=550,
    height=550,
    resizable=False)  # create viewport takes in config options too!

dpg.setup_dearpygui()
dpg.show_viewport()
dpg.set_primary_window("Main Window", True)
dpg.start_dearpygui()
dpg.destroy_context()