salmanmapkar commited on
Commit
bd890fb
1 Parent(s): bb6f760

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -11
app.py CHANGED
@@ -27,11 +27,11 @@ def GetSpeaker(sp):
27
  speaker = SPEAKER_DICT[sp]
28
  else:
29
  speaker = SPEAKER_DICT[sp]
 
30
  return speaker
31
 
32
  def GenerateSpeakerDict(sp):
33
  SPEAKERS = [speaker.strip() for speaker in sp.split(',')]
34
- print(SPEAKERS)
35
 
36
 
37
  def CreateFile(filename):
@@ -142,16 +142,14 @@ def AudioTranscribe(NumberOfSpeakers=None, SpeakerNames="", audio="", retries=5)
142
 
143
  def VideoTranscribe(NumberOfSpeakers=None, SpeakerNames="", video="", retries=5):
144
  if retries:
145
- # command = f"ffmpeg -i {video} -ab 160k -ac 2 -ar 44100 -vn temp_audio.wav"
146
- # subprocess.call(command, shell=True)
147
- clip = mp.VideoFileClip(video)
148
- clip.audio.write_audiofile("temp_audio.wav")
149
- # try:
150
- # command = f"ffmpeg -i {video} -ab 160k -ac 2 -ar 44100 -vn temp_audio.wav"
151
- # subprocess.call(command, shell=True)
152
- # except Exception as ex:
153
- # traceback.print_exc()
154
- # return VideoTranscribe(NumberOfSpeakers, SpeakerNames, video, retries-1)
155
  if not (os.path.isfile("temp_audio.wav")):
156
  return VideoTranscribe(NumberOfSpeakers, SpeakerNames, video, retries-1)
157
  return Transcribe(NumberOfSpeakers, SpeakerNames)
 
27
  speaker = SPEAKER_DICT[sp]
28
  else:
29
  speaker = SPEAKER_DICT[sp]
30
+ prin(sp, speaker)
31
  return speaker
32
 
33
  def GenerateSpeakerDict(sp):
34
  SPEAKERS = [speaker.strip() for speaker in sp.split(',')]
 
35
 
36
 
37
  def CreateFile(filename):
 
142
 
143
  def VideoTranscribe(NumberOfSpeakers=None, SpeakerNames="", video="", retries=5):
144
  if retries:
145
+ try:
146
+ clip = mp.VideoFileClip(video)
147
+ clip.audio.write_audiofile("temp_audio.wav")
148
+ # command = f"ffmpeg -i {video} -ab 160k -ac 2 -ar 44100 -vn temp_audio.wav"
149
+ # subprocess.call(command, shell=True)
150
+ except Exception as ex:
151
+ traceback.print_exc()
152
+ return VideoTranscribe(NumberOfSpeakers, SpeakerNames, video, retries-1)
 
 
153
  if not (os.path.isfile("temp_audio.wav")):
154
  return VideoTranscribe(NumberOfSpeakers, SpeakerNames, video, retries-1)
155
  return Transcribe(NumberOfSpeakers, SpeakerNames)