Update app.py

app.py

@@ -1,25 +1,9 @@
-import socket
 import streamlit as st
-
-# ==========================================
-# 🛑 CRITICAL NETWORK FIX (Must be at top)
-# ==========================================
-# This hack forces the server to ignore IPv6 completely.
-# It fixes the "[Errno -5] No address associated with hostname" error.
-try:
-    _old_getaddrinfo = socket.getaddrinfo
-    def _new_getaddrinfo(*args, **kwargs):
-        responses = _old_getaddrinfo(*args, **kwargs)
-        # Filter out IPv6 (AF_INET6) and keep only IPv4 (AF_INET)
-        return [r for r in responses if r[0] == socket.AF_INET]
-    socket.getaddrinfo = _new_getaddrinfo
-except Exception as e:
-    pass
-# ==========================================
-
-import yt_dlp
 import os
 import time
+from pytube import YouTube
+from pytube.exceptions import RegexMatchError, VideoUnavailable
+import ffmpeg
 from PIL import Image
 import io
 import re
@@ -46,11 +30,11 @@ app_mode = st.sidebar.radio(
 )
 
 # ==========================================
-# VIEW 1: YOUTUBE DOWNLOADER
+# VIEW 1: YOUTUBE DOWNLOADER (Pytube Version)
 # ==========================================
 if app_mode == "YouTube Downloader":
     st.title("🎥 YouTube Media Extractor")
-    st.info("Download Video (1080p) or Audio (MP3) instantly.")
+    st.info("Download Video (720p/1080p) or Audio (MP3).")
     
     url = st.text_input("Paste YouTube URL", placeholder="https://youtube.com/watch?v=...")
     
@@ -60,53 +44,61 @@ if app_mode == "YouTube Downloader":
     
     if url and st.button("🚀 Process Media"):
         status_area = st.empty()
-        status_area.info("⏳ Fetching metadata... (This may take 10-20s)")
+        status_area.info("⏳ Connecting to YouTube...")
         
         try:
-            timestamp = int(time.time())
-            out_tmpl = f"{timestamp}_%(title)s.%(ext)s"
+            # 1. Initialize Pytube Object
+            yt = YouTube(url)
             
-            # UPDATED OPTIONS: Using built-in cookies and aggressive timeouts
-            ydl_opts = {
-                'outtmpl': out_tmpl,
-                'format': 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best' if format_type == "Video (MP4)" else 'bestaudio/best',
-                'postprocessors': [{
-                    'key': 'FFmpegExtractAudio',
-                    'preferredcodec': 'mp3',
-                    'preferredquality': '192',
-                }] if "Audio" in format_type else [],
-                'quiet': True,
-                'no_warnings': True,
-                'noplaylist': True,
-                'socket_timeout': 10,  # Prevent hanging
-            }
+            # This triggers the network call - if it fails here, the IP is blocked.
+            video_title = yt.title 
+            status_area.info(f"✅ Found: {video_title}")
+            
+            timestamp = int(time.time())
+            filename = f"{timestamp}_{re.sub(r'[^a-zA-Z0-9]', '', video_title)}"
+            final_path = ""
 
-            with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-                info = ydl.extract_info(url, download=True)
-                downloaded_file = ydl.prepare_filename(info)
+            # 2. Download Logic
+            if format_type == "Video (MP4)":
+                # Get highest resolution progressive stream (audio+video combined)
+                # Pytube progressive streams are usually max 720p but safest/most stable
+                stream = yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first()
+                final_path = stream.download(filename=filename + ".mp4")
+                
+            else: # Audio Only
+                # Get audio only stream
+                stream = yt.streams.filter(only_audio=True).first()
+                downloaded_file = stream.download(filename=filename + ".mp4")
                 
-                if "Audio" in format_type:
-                    base, ext = os.path.splitext(downloaded_file)
-                    downloaded_file = base + ".mp3"
+                # Rename to mp3 (Pytube downloads audio as mp4/webm container)
+                base, ext = os.path.splitext(downloaded_file)
+                final_path = base + ".mp3"
+                os.rename(downloaded_file, final_path)
 
-            if os.path.exists(downloaded_file):
-                status_area.success("✅ Done!")
+            # 3. Serve File
+            if os.path.exists(final_path):
+                status_area.success("✅ Download Ready!")
                 
-                with open(downloaded_file, "rb") as f:
+                with open(final_path, "rb") as f:
                     file_bytes = f.read()
                     st.download_button(
                         label=f"⬇️ Download {format_type}",
                         data=file_bytes,
-                        file_name=os.path.basename(downloaded_file),
+                        file_name=os.path.basename(final_path),
                         mime="video/mp4" if "Video" in format_type else "audio/mpeg"
                     )
                 
                 # Cleanup
-                os.remove(downloaded_file)
+                os.remove(final_path)
             else:
-                st.error("File not found. Please try again.")
-                
+                st.error("Processing failed during save.")
+
+        except RegexMatchError:
+            st.error("Invalid URL. Please check the link.")
+        except VideoUnavailable:
+            st.error("Video is unavailable (Private or Region Blocked).")
         except Exception as e:
+            # If this prints a network error again, the server IP is definitely banned.
             st.error(f"Error: {str(e)}")
 
 # ==========================================