Update app.py

app.py

@@ -5,12 +5,7 @@ import re
 import time
 from moviepy.video.io.VideoFileClip import VideoFileClip
 
-# Function to sanitize the filename
-def sanitize_filename(title):
-    # Replace invalid characters with an underscore
-    return re.sub(r'[<>:"/\\|?*]', '_', title)
-
-# Function to download video and return the file path
+# Function to download video
 def download_video(url):
     download_path = './downloads'  # Folder to store downloads
     if not os.path.exists(download_path):
@@ -18,62 +13,34 @@ def download_video(url):
 
     ydl_opts = {
         'outtmpl': download_path + '/%(title)s.%(ext)s',
-        'format': 'best',  # Download the best available quality
     }
-    
     try:
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            info_dict = ydl.extract_info(url, download=True)  # Set download=True to download the video
+            info_dict = ydl.extract_info(url, download=False)
             title = info_dict.get('title', None)
-            if title:
-                sanitized_title = sanitize_filename(title)
-                file_path = os.path.join(download_path, f"{sanitized_title}.mp4")  # Adjust the extension as needed
-                return file_path if os.path.exists(file_path) else None  # Ensure the file exists before returning
-            return None
-    except Exception as e:
-        return str(e)
-
-# Function to crop the video
-def crop_video(input_path, start_time, end_time):
-    output_path = input_path.replace('.mp4', '_cropped.mp4')
-    try:
-        with VideoFileClip(input_path) as video:
-            cropped_video = video.subclip(start_time, end_time)
-            cropped_video.write_videofile(output_path, codec='libx264')
-        return output_path
+            ydl.download([url])
+            return f"Downloaded successfully: {title}", f"/downloads/{title}.mp4"
     except Exception as e:
-        return str(e)
+        return f"Error: {e}", None
 
 # Create Gradio interface
 def gradio_interface():
     with gr.Blocks() as demo:
-        gr.Markdown("# YouTube Video Downloader and Cropper")
-        gr.Markdown("Enter the YouTube URL to download and crop the video.")
+        gr.Markdown("# YouTube Video Downloader")
+        gr.Markdown("Enter the YouTube URL to download the video.")
 
         # Input for YouTube URL
         url_input = gr.Textbox(label="YouTube URL", placeholder="Enter YouTube video URL here...")
         
-        # Inputs for cropping
-        start_time_input = gr.Number(label="Start Time (seconds)", value=0)
-        end_time_input = gr.Number(label="End Time (seconds)", value=10)
-
-        # File output for downloaded video
-        output_file = gr.File(label="Download Cropped Video")
+        # Output display
+        output = gr.Textbox(label="Status")
+        video_link = gr.File(label="Download Link")
 
-        # Download and Crop button
-        process_btn = gr.Button("Download and Crop Video")
+        # Download button
+        download_btn = gr.Button("Download Video")
 
         # Define the action on button click
-        def process_video(url, start_time, end_time):
-            video_path = download_video(url)
-            if video_path:
-                cropped_video_path = crop_video(video_path, start_time, end_time)
-                if cropped_video_path and os.path.exists(cropped_video_path):
-                    time.sleep(1)  # Add a slight delay to ensure file is fully written
-                    return cropped_video_path
-            return None
-
-        process_btn.click(process_video, inputs=[url_input, start_time_input, end_time_input], outputs=output_file)
+        download_btn.click(download_video, inputs=url_input, outputs=[output, video_link])
 
     return demo