Pushkar0655g commited on
Commit
e219dd2
Β·
verified Β·
1 Parent(s): 7371a9e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -13
app.py CHANGED
@@ -36,11 +36,6 @@ body {
36
  padding: 20px;
37
  border-radius: 8px;
38
  }
39
- .progress-bar {
40
- background: #007acc;
41
- border-radius: 8px;
42
- height: 20px;
43
- }
44
  .progress-text {
45
  color: #00ff00;
46
  font-weight: bold;
@@ -68,7 +63,6 @@ with gr.Blocks(theme=gr.themes.Monochrome(), css=css) as demo:
68
  )
69
 
70
  generate_button = gr.Button("Generate Subtitles πŸš€", elem_classes=["btn-primary"])
71
- progress_bar = gr.ProgressBar(label="Progress", elem_classes=["progress-bar"])
72
  progress_text = gr.Textbox(
73
  label="Status",
74
  interactive=False,
@@ -80,29 +74,29 @@ with gr.Blocks(theme=gr.themes.Monochrome(), css=css) as demo:
80
  try:
81
  # Validate file type
82
  if not video_file.name.lower().endswith(('.mp4', '.mkv', '.avi')):
83
- return None, None, "❌ Invalid file type. Please upload an MP4, MKV, or AVI file."
84
 
85
- # Initialize progress
86
  progress = gr.Progress(track_tqdm=True)
87
- progress(0, desc="Initializing...")
88
 
89
  # Process video
 
90
  srt_path = process_video(video_file.name, language, progress=progress)
91
  if srt_path:
92
  progress(1, desc="βœ… Subtitles generated successfully!")
93
- return gr.File(srt_path), progress(1), "βœ… Subtitles ready for download!"
94
  else:
95
  progress(0, desc="❌ Error during processing.")
96
- return None, progress(0), "❌ Failed to generate subtitles. Check logs."
97
 
98
  except Exception as e:
99
  progress(0, desc="❌ Error encountered.")
100
- return None, progress(0), f"❌ Error: {str(e)}"
101
 
102
  generate_button.click(
103
  generate_subtitles,
104
  inputs=[video_input, language_dropdown],
105
- outputs=[output_srt, progress_bar, progress_text]
106
  )
107
 
108
  demo.launch()
 
36
  padding: 20px;
37
  border-radius: 8px;
38
  }
 
 
 
 
 
39
  .progress-text {
40
  color: #00ff00;
41
  font-weight: bold;
 
63
  )
64
 
65
  generate_button = gr.Button("Generate Subtitles πŸš€", elem_classes=["btn-primary"])
 
66
  progress_text = gr.Textbox(
67
  label="Status",
68
  interactive=False,
 
74
  try:
75
  # Validate file type
76
  if not video_file.name.lower().endswith(('.mp4', '.mkv', '.avi')):
77
+ return None, "❌ Invalid file type. Please upload an MP4, MKV, or AVI file."
78
 
79
+ # Initialize progress tracking
80
  progress = gr.Progress(track_tqdm=True)
 
81
 
82
  # Process video
83
+ progress(0, desc="Initializing...")
84
  srt_path = process_video(video_file.name, language, progress=progress)
85
  if srt_path:
86
  progress(1, desc="βœ… Subtitles generated successfully!")
87
+ return gr.File(srt_path), "βœ… Subtitles ready for download!"
88
  else:
89
  progress(0, desc="❌ Error during processing.")
90
+ return None, "❌ Failed to generate subtitles. Check logs."
91
 
92
  except Exception as e:
93
  progress(0, desc="❌ Error encountered.")
94
+ return None, f"❌ Error: {str(e)}"
95
 
96
  generate_button.click(
97
  generate_subtitles,
98
  inputs=[video_input, language_dropdown],
99
+ outputs=[output_srt, progress_text]
100
  )
101
 
102
  demo.launch()