edchengg commited on
Commit
bf9542e
1 Parent(s): b7bf4d3
Files changed (1) hide show
  1. app.py +35 -0
app.py ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import openai
3
+ import youtube_dl
4
+
5
+
6
+ def asr(url):
7
+ # download audio
8
+ # Options for youtube-dl
9
+ ydl_opts = {
10
+ 'outtmpl': 'my_video.mp4'
11
+ }
12
+
13
+ # Create a youtube-dl object
14
+ ydl = youtube_dl.YoutubeDL(ydl_opts)
15
+
16
+ # Download the video
17
+ info_dict = ydl.extract_info(url, download=True)
18
+
19
+ audio_file= open("my_video.mp4", "rb")
20
+ transcript = openai.Audio.transcribe("whisper-1", audio_file)
21
+ output = openai.ChatCompletion.create(
22
+ model="gpt-3.5-turbo",
23
+ messages=[
24
+ {"role": "user", "content": "Transcript: {transcript}. \n Translate the video conversation transcript into fluent Chinese. Chinese: ".format(transcript=transcript["text"])},
25
+ ]
26
+ )
27
+ return transcript["text"], output['choices'][0]['message']['content']
28
+
29
+ demo = gr.Interface(fn=asr, inputs="text",
30
+ outputs=[
31
+ gr.outputs.Textbox(label="English"),
32
+ gr.outputs.Textbox(label="Chinese"),
33
+ ])
34
+
35
+ demo.launch(share=True)