Update app.py

app.py

@@ -17,7 +17,9 @@ import spaces
 # Constants and initialization
 HF_TOKEN = os.environ.get("HF_TOKEN")
 REPO_ID = "artificialguybr/video-dubbing"
-MAX_VIDEO_DURATION = 60  # seconds
+
+MAX_VIDEO_DURATION = 900  # 15 minutes
+MAX_FILE_SIZE = 1073741824  # 1GB
 
 api = HfApi(token=HF_TOKEN)
 
@@ -60,6 +62,10 @@ def cleanup_files(*files):
             os.remove(file)
             print(f"Removed file: {file}")
 
+def validate_file_size(video_path):
+    if os.path.getsize(video_path) > MAX_FILE_SIZE:
+        raise ValueError(f"File size exceeds 1GB limit. Please upload a smaller file.")
+
 @spaces.GPU(duration=90)
 def transcribe_audio(file_path):
     print(f"Starting transcription of file: {file_path}")
@@ -116,6 +122,8 @@ def process_video(video, target_language, use_wav2lip):
     try:
         if target_language is None:
             raise ValueError("Please select a Target Language for Dubbing.")
+
+        validate_file_size(video)  # Validate file size first
         
         run_uuid = uuid.uuid4().hex[:6]
         output_filename = f"{run_uuid}_resized_video.mp4"
@@ -177,8 +185,8 @@ def process_video(video, target_language, use_wav2lip):
 # Gradio interface setup
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# AI Video Dubbing")
-    gr.Markdown("This tool uses AI to dub videos into different languages. Upload a video, choose a target language, and get a dubbed version!")
-    
+    gr.Markdown("Upload a video (up to 15 minutes and 1GB), choose a target language, and get a dubbed version!")
+
     with gr.Row():
         with gr.Column(scale=2):
             video_input = gr.Video(label="Upload Video")
@@ -206,19 +214,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     gr.Markdown("""
     ## Notes:
-    - Video limit is 1 minute. The tool will dub all speakers using a single voice.
-    - Processing may take up to 5 minutes.
-    - This is an alpha version using open-source models.
-    - Quality vs. speed trade-off was made for scalability and hardware limitations.
-    - For videos longer than 1 minute, please duplicate this Space and adjust the limit in the code.
-    """)
-
-    gr.Markdown("""
-    ---
-    Developed by [@artificialguybr](https://twitter.com/artificialguybr) using open-source tools. 
-    Special thanks to Hugging Face for GPU support and [@yeswondwer](https://twitter.com/@yeswondwerr) for the original code.
-    
-    Try our [Video Transcription and Translation](https://huggingface.co/spaces/artificialguybr/VIDEO-TRANSLATION-TRANSCRIPTION) tool!
+    - Video limit: 15 minutes, 1GB file size.
+    - Processing may take several minutes depending on length.
+    - Developed using open-source models and tools.
     """)
 
 print("Launching Gradio interface...")