David Li commited on
Commit
fb35ac4
·
1 Parent(s): e4d4883

fix: try again

Browse files
Files changed (1) hide show
  1. app.py +11 -5
app.py CHANGED
@@ -1,15 +1,21 @@
1
  import gradio as gr
2
  import re
 
3
  import glob
4
  from openbb_terminal.forecast.whisper_model import transcribe_and_summarize
5
 
 
6
  def get_video_id(url):
 
7
  video_id = re.findall(r"v=([-\w]{11})", url)[0]
8
- # make folder to store output files
9
- # extract video ID from URL using regular expression
10
- transcribe_and_summarize(video=url, output_dir=video_id)
11
- # return files from video_id folder
12
- # find file with video_id/*_summary.txt
 
 
 
13
  summary_file = glob.glob(f"{video_id}/*_summary.txt")[0]
14
  # file .srt file
15
  subtitle_file = glob.glob(f"{video_id}/*.srt")[0] or glob.glob(f"{video_id}/*.vtt")[0]
 
1
  import gradio as gr
2
  import re
3
+ import sys
4
  import glob
5
  from openbb_terminal.forecast.whisper_model import transcribe_and_summarize
6
 
7
+
8
  def get_video_id(url):
9
+
10
  video_id = re.findall(r"v=([-\w]{11})", url)[0]
11
+ if not sys.stdin.isatty():
12
+ sys.stdin = StringIO('-y')
13
+ print("work around for gradio")
14
+ transcribe_and_summarize(video=url, output_dir=video_id)
15
+ else:
16
+ sys.stdin = StringIO('-y')
17
+ print("guessing its not interactive")
18
+ transcribe_and_summarize(video=url, output_dir=video_id)
19
  summary_file = glob.glob(f"{video_id}/*_summary.txt")[0]
20
  # file .srt file
21
  subtitle_file = glob.glob(f"{video_id}/*.srt")[0] or glob.glob(f"{video_id}/*.vtt")[0]