Update app.py

app.py

@@ -1,11 +1,12 @@
-# Dependencies:
-# pip install gradio faster-whisper moviepy torch
-
 import gradio as gr
 import torch
 import os
 from faster_whisper import WhisperModel
-from moviepy.video.io.VideoFileClip import VideoFileClip  # Updated import
+from moviepy.video.io.VideoFileClip import VideoFileClip
+import logging
+
+# Suppress moviepy logs
+logging.getLogger("moviepy").setLevel(logging.ERROR)
 
 # Define the model and device
 MODEL_NAME = "Systran/faster-whisper-large-v3"
@@ -39,7 +40,7 @@ def extract_audio_from_video(video_file):
     """Extract audio from a video file and save it as a WAV file."""
     video = VideoFileClip(video_file)
     audio_file = "extracted_audio.wav"
-    video.audio.write_audiofile(audio_file, fps=16000)
+    video.audio.write_audiofile(audio_file, fps=16000, logger=None)  # Suppress logs
     return audio_file
 
 def generate_subtitles(audio_file, language="Auto Detect"):
@@ -94,46 +95,16 @@ def process_video(video_file, language="Auto Detect"):
     
     return srt_file
 
-# Custom CSS for styling
-custom_css = """
-.gradio-container {
-    background: linear-gradient(135deg, #f5f7fa, #c3cfe2);
-    font-family: 'Arial', sans-serif;
-}
-.header {
-    text-align: center;
-    padding: 20px;
-    background: linear-gradient(135deg, #6a11cb, #2575fc);
-    color: white;
-    border-radius: 10px;
-    margin-bottom: 20px;
-}
-.header h1 {
-    font-size: 2.5rem;
-    margin: 0;
-}
-.header p {
-    font-size: 1.2rem;
-    margin: 10px 0 0;
-}
-.tab {
-    background: white;
-    padding: 20px;
-    border-radius: 10px;
-    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
-}
-"""
-
 # Define the Gradio interface
-with gr.Blocks(css=custom_css, title="AutoSubGen - AI Video Subtitle Generator") as demo:
+with gr.Blocks(title="AutoSubGen - AI Video Subtitle Generator") as demo:
     # Header
-    with gr.Column(elem_classes="header"):
-        gr.Markdown("# AutoSubGen")
+    with gr.Column():
+        gr.Markdown("# 🎥 AutoSubGen")
         gr.Markdown("### AI-Powered Video Subtitle Generator")
-        gr.Markdown("Automatically generate subtitles for your videos in SRT format. Supports 100+ languages and auto-detection.")
+        gr.Markdown("Automatically generate subtitles for your videos in **SRT format**. Supports **100+ languages** and **auto-detection**.")
     
     # Main content
-    with gr.Tab("Generate Subtitles", elem_classes="tab"):
+    with gr.Tab("Generate Subtitles"):
         gr.Markdown("### Upload a video file to generate subtitles.")
         with gr.Row():
             video_input = gr.Video(label="Upload Video File", scale=2)
@@ -153,5 +124,5 @@ with gr.Blocks(css=custom_css, title="AutoSubGen - AI Video Subtitle Generator")
         outputs=subtitle_output
     )
 
-# Launch the Gradio interface
-demo.launch()
+# Launch the Gradio interface with a public link
+demo.launch(share=True)