Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from datetime import datetime
 import random
+import logging
 from transformers.pipelines.audio_utils import ffmpeg_read
 from moviepy import (
     ImageClip, 
@@ -11,25 +12,41 @@ from moviepy import (
     concatenate_videoclips
 )
 
+# Set up logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(message)s')
+logger = logging.getLogger()
+
 def generate_video_segments(video_path, segment_length=5):
     """Splits the video into segments and returns their paths."""
     video = VideoFileClip(video_path)
     segment_paths = []
     
+    logger.info(f"Video duration: {video.duration} seconds.")
+    
     for i, start in enumerate(range(0, int(video.duration), segment_length)):
         end = min(start + segment_length, video.duration)
-        segment_clip = video.subclip(start, end)
-        segment_filename = f"segment_{i}.mp4"
-        segment_clip.write_videofile(segment_filename, codec="libx264")
-        segment_paths.append(segment_filename)
-
+        logger.info(f"Processing segment {i}: Start time = {start}s, End time = {end}s")
+        
+        try:
+            segment_clip = video.subclip(start, end)
+            segment_filename = f"segment_{i}.mp4"
+            logger.info(f"Writing segment {i} to file: {segment_filename}")
+            segment_clip.write_videofile(segment_filename, codec="libx264")
+            segment_paths.append(segment_filename)
+        except Exception as e:
+            logger.error(f"Error in processing segment {i}: {str(e)}")
+    
+    logger.info(f"Total segments generated: {len(segment_paths)}")
     return segment_paths
 
 def segment_video(file):
     if file is None:
+        logger.error("No file uploaded.")
         return "No file uploaded.", []
     
+    logger.info("Starting video segmentation.")
     segments = generate_video_segments(file.name)
+    logger.info("Video segmentation complete.")
     return "Video segmented successfully!", segments
 
 with gr.Blocks() as demo:
@@ -41,4 +58,4 @@ with gr.Blocks() as demo:
 
     segment_button.click(segment_video, inputs=file_input, outputs=[output_text, video_gallery])
 
-demo.launch()
+demo.launch()