Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,17 +1,24 @@
|
|
1 |
from openai import OpenAI
|
2 |
from pydub import AudioSegment
|
3 |
-
import
|
4 |
import gradio as gr
|
|
|
5 |
|
6 |
client = OpenAI(api_key=os.getenv('OPENAI_API_KEY'))
|
7 |
|
8 |
def extract_audio(video_file):
|
9 |
-
|
10 |
-
|
11 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
12 |
|
13 |
def get_transcript(audio_file):
|
14 |
-
audio_file = open(audio_file, "rb")
|
15 |
transcript = client.audio.transcriptions.create(
|
16 |
file=audio_file,
|
17 |
model="whisper-1",
|
|
|
1 |
from openai import OpenAI
|
2 |
from pydub import AudioSegment
|
3 |
+
import io
|
4 |
import gradio as gr
|
5 |
+
import os
|
6 |
|
7 |
client = OpenAI(api_key=os.getenv('OPENAI_API_KEY'))
|
8 |
|
9 |
def extract_audio(video_file):
|
10 |
+
# Read video file into memory
|
11 |
+
video_data = video_file.read()
|
12 |
+
|
13 |
+
video_audio = AudioSegment.from_file(io.BytesIO(video_data), format="mp4")
|
14 |
+
|
15 |
+
audio_data = io.BytesIO()
|
16 |
+
video_audio.export(audio_data, format="mp3")
|
17 |
+
audio_data.seek(0)
|
18 |
+
|
19 |
+
return audio_data
|
20 |
|
21 |
def get_transcript(audio_file):
|
|
|
22 |
transcript = client.audio.transcriptions.create(
|
23 |
file=audio_file,
|
24 |
model="whisper-1",
|