Update app.py
Browse files
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,
|
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),
|
94 |
else:
|
95 |
progress(0, desc="β Error during processing.")
|
96 |
-
return None,
|
97 |
|
98 |
except Exception as e:
|
99 |
progress(0, desc="β Error encountered.")
|
100 |
-
return None,
|
101 |
|
102 |
generate_button.click(
|
103 |
generate_subtitles,
|
104 |
inputs=[video_input, language_dropdown],
|
105 |
-
outputs=[output_srt,
|
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()
|