themanas021 commited on
Commit
d247bf3
·
verified ·
1 Parent(s): 283cdf9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -5
app.py CHANGED
@@ -1,17 +1,24 @@
1
  from openai import OpenAI
2
  from pydub import AudioSegment
3
- import os
4
  import gradio as gr
 
5
 
6
  client = OpenAI(api_key=os.getenv('OPENAI_API_KEY'))
7
 
8
  def extract_audio(video_file):
9
- output_audio_file = "demo1.mp3"
10
- os.system(f"ffmpeg -i {video_file} -q:a 0 -map a {output_audio_file}")
11
- return output_audio_file
 
 
 
 
 
 
 
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",