csukuangfj commited on
Commit
b922c32
1 Parent(s): 03f70d9

small fixes

Browse files
Files changed (1) hide show
  1. app.py +21 -5
app.py CHANGED
@@ -83,7 +83,26 @@ def show_file_info(in_filename: str):
83
  _ = os.system(f"ffprobe -hide_banner -i '{in_filename}'")
84
 
85
 
86
- def process_uploaded_file(
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
87
  language: str,
88
  repo_id: str,
89
  in_filename: str,
@@ -115,7 +134,6 @@ def process(language: str, repo_id: str, in_filename: str):
115
  logging.info("Done")
116
 
117
  return (
118
- (in_filename, srt_filename),
119
  srt_filename,
120
  build_html_output("Done! Please download the SRT file", "result_item_success"),
121
  result,
@@ -176,7 +194,6 @@ with demo:
176
  )
177
  upload_audio_button = gr.Button("Submit for recognition")
178
 
179
- output_audio = gr.Audio(label="Output")
180
  output_srt_file_audio = gr.File(
181
  label="Generated subtitles", show_label=True
182
  )
@@ -187,7 +204,7 @@ with demo:
187
  )
188
 
189
  upload_video_button.click(
190
- process_uploaded_file,
191
  inputs=[
192
  language_radio,
193
  model_dropdown,
@@ -209,7 +226,6 @@ with demo:
209
  uploaded_audio_file,
210
  ],
211
  outputs=[
212
- output_audio,
213
  output_srt_file_audio,
214
  output_info_audio,
215
  output_textbox_audio,
 
83
  _ = os.system(f"ffprobe -hide_banner -i '{in_filename}'")
84
 
85
 
86
+ def process_uploaded_video_file(
87
+ language: str,
88
+ repo_id: str,
89
+ in_filename: str,
90
+ ):
91
+ if in_filename is None or in_filename == "":
92
+ return "", build_html_output(
93
+ "Please first upload a file and then click "
94
+ 'the button "submit for recognition"',
95
+ "result_item_error",
96
+ )
97
+
98
+ logging.info(f"Processing uploaded file: {in_filename}")
99
+ show_file_info(in_filename)
100
+
101
+ ans = process(language, repo_id, in_filename)
102
+ return (in_filename, ans[0]), ans[0], ans[1], ans[2]
103
+
104
+
105
+ def process_uploaded_audio_file(
106
  language: str,
107
  repo_id: str,
108
  in_filename: str,
 
134
  logging.info("Done")
135
 
136
  return (
 
137
  srt_filename,
138
  build_html_output("Done! Please download the SRT file", "result_item_success"),
139
  result,
 
194
  )
195
  upload_audio_button = gr.Button("Submit for recognition")
196
 
 
197
  output_srt_file_audio = gr.File(
198
  label="Generated subtitles", show_label=True
199
  )
 
204
  )
205
 
206
  upload_video_button.click(
207
+ process_uploaded_video_file,
208
  inputs=[
209
  language_radio,
210
  model_dropdown,
 
226
  uploaded_audio_file,
227
  ],
228
  outputs=[
 
229
  output_srt_file_audio,
230
  output_info_audio,
231
  output_textbox_audio,