Update app.py

app.py

@@ -1,9 +1,7 @@
 import streamlit as st
-import os
+import requests
+import json
 import time
-from pytube import YouTube
-from pytube.exceptions import RegexMatchError, VideoUnavailable
-import ffmpeg
 from PIL import Image
 import io
 import re
@@ -30,79 +28,68 @@ app_mode = st.sidebar.radio(
 )
 
 # ==========================================
-# VIEW 1: YOUTUBE DOWNLOADER (Pytube Version)
+# VIEW 1: YOUTUBE DOWNLOADER (API Version)
 # ==========================================
 if app_mode == "YouTube Downloader":
     st.title("🎥 YouTube Media Extractor")
-    st.info("Download Video (720p/1080p) or Audio (MP3).")
+    st.info("High-Speed Downloader (Powered by Cobalt API)")
     
     url = st.text_input("Paste YouTube URL", placeholder="https://youtube.com/watch?v=...")
     
     col1, col2 = st.columns(2)
     with col1:
+        # Cobalt supports various qualities. We'll map them simple.
         format_type = st.radio("Format", ["Video (MP4)", "Audio Only (MP3)"])
     
     if url and st.button("🚀 Process Media"):
         status_area = st.empty()
-        status_area.info("⏳ Connecting to YouTube...")
+        status_area.info("⏳ Contacting processing server...")
         
         try:
-            # 1. Initialize Pytube Object
-            yt = YouTube(url)
+            # Cobalt API Configuration
+            api_url = "https://api.cobalt.tools/api/json"
+            headers = {
+                "Accept": "application/json",
+                "Content-Type": "application/json",
+                "User-Agent": "LexicalTools/1.0"
+            }
             
-            # 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 = ""
+            payload = {
+                "url": url,
+                "vCodec": "h264",
+                "vQuality": "1080",
+                "aFormat": "mp3",
+                "isAudioOnly": True if "Audio" in format_type else False
+            }
 
-            # 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")
-                
-                # 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)
+            # Send Request
+            response = requests.post(api_url, headers=headers, json=payload)
+            data = response.json()
 
-            # 3. Serve File
-            if os.path.exists(final_path):
-                status_area.success("✅ Download Ready!")
+            if "url" in data:
+                download_link = data["url"]
+                status_area.success("✅ Media Ready!")
                 
-                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(final_path),
-                        mime="video/mp4" if "Video" in format_type else "audio/mpeg"
-                    )
+                # Show a Direct Download Link (Fastest method)
+                st.link_button(f"⬇️ Click to Download {format_type}", download_link)
                 
-                # Cleanup
-                os.remove(final_path)
+                # Optional: Preview
+                if "Audio" not in format_type:
+                    st.video(download_link)
+                else:
+                    st.audio(download_link)
+                    
+            elif "text" in data:
+                # API returned an error message
+                st.error(f"Server Error: {data['text']}")
             else:
-                st.error("Processing failed during save.")
+                st.error("Unknown error from processing server.")
 
-        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)}")
+            st.error(f"Connection Error: {str(e)}")
 
 # ==========================================
-# VIEW 2: WEBMASTER TOOLKIT
+# VIEW 2: WEBMASTER TOOLKIT (Existing)
 # ==========================================
 elif app_mode == "Webmaster Toolkit":
     st.title("⚡ Webmaster's Toolkit")