Update app.py

app.py

@@ -1,6 +1,9 @@
 import streamlit as st
 import requests
 import json
+import time
+from PIL import Image
+import io
 import re
 from bs4 import BeautifulSoup
 import markdown
@@ -13,7 +16,7 @@ st.set_page_config(
     initial_sidebar_state="expanded"
 )
 
-# --- SIDEBAR & SERVER SELECTOR ---
+# --- SIDEBAR NAVIGATION ---
 st.sidebar.title("Navigation")
 query_params = st.query_params
 default_index = 1 if query_params.get("mode") == "downloader" else 0
@@ -24,40 +27,27 @@ app_mode = st.sidebar.radio(
     index=default_index
 )
 
-# List of Public Cobalt v10 Instances (Community Hosted)
-# If one fails, the user can switch to another.
-SERVERS = {
-    "Auto (US)": "https://co.wuk.sh",
-    "Backup (EU)": "https://cobalt.api.timelessnesses.me",
-    "Backup (Asia)": "https://api.cobalt.750.moe",
-    "Backup (US-East)": "https://cobalt-api.hyper.lol"
-}
-
-if app_mode == "YouTube Downloader":
-    st.sidebar.markdown("---")
-    st.sidebar.subheader("⚙️ Server Settings")
-    selected_server_name = st.sidebar.selectbox("Processing Server", list(SERVERS.keys()))
-    api_base_url = SERVERS[selected_server_name]
-
 # ==========================================
-# VIEW 1: YOUTUBE DOWNLOADER (v10 API)
+# VIEW 1: YOUTUBE DOWNLOADER (API Version)
 # ==========================================
 if app_mode == "YouTube Downloader":
     st.title("🎥 YouTube Media Extractor")
-    st.caption(f"Powered by Cobalt v10 • Server: {selected_server_name}")
+    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("⏳ Contacting remote server...")
+        status_area.info("⏳ Contacting processing server...")
         
         try:
-            # v10 API Payload Structure
+            # Cobalt API Configuration
+            api_url = "https://api.cobalt.tools/api/json"
             headers = {
                 "Accept": "application/json",
                 "Content-Type": "application/json",
@@ -66,44 +56,37 @@ if app_mode == "YouTube Downloader":
             
             payload = {
                 "url": url,
-                # v10 uses 'downloadMode' instead of 'isAudioOnly'
-                "downloadMode": "audio" if "Audio" in format_type else "auto",
-                "videoQuality": "1080",
-                "filenameStyle": "basic"
+                "vCodec": "h264",
+                "vQuality": "1080",
+                "aFormat": "mp3",
+                "isAudioOnly": True if "Audio" in format_type else False
             }
 
-            # Send Request to selected instance
-            response = requests.post(f"{api_base_url}", headers=headers, json=payload)
-            
-            try:
-                data = response.json()
-            except:
-                st.error(f"Server Error: {response.text}")
-                st.stop()
+            # Send Request
+            response = requests.post(api_url, headers=headers, json=payload)
+            data = response.json()
 
-            # Handle Success
             if "url" in data:
                 download_link = data["url"]
                 status_area.success("✅ Media Ready!")
                 
-                # Direct Stream Link (No server storage used)
+                # Show a Direct Download Link (Fastest method)
                 st.link_button(f"⬇️ Click to Download {format_type}", download_link)
                 
-                # Preview
+                # Optional: Preview
                 if "Audio" not in format_type:
                     st.video(download_link)
                 else:
                     st.audio(download_link)
-            
-            # Handle API Specific Errors
+                    
             elif "text" in data:
-                st.error(f"API Error: {data['text']}")
+                # API returned an error message
+                st.error(f"Server Error: {data['text']}")
             else:
-                st.error("Unknown error. Try switching the server in the sidebar.")
+                st.error("Unknown error from processing server.")
 
         except Exception as e:
-            st.error(f"Connection Failed: {str(e)}")
-            st.info("💡 Tip: Try selecting a different server in the Sidebar.")
+            st.error(f"Connection Error: {str(e)}")
 
 # ==========================================
 # VIEW 2: WEBMASTER TOOLKIT (Existing)
@@ -113,17 +96,21 @@ elif app_mode == "Webmaster Toolkit":
     
     tab1, tab2 = st.tabs(["📝 Text Cleaner & SEO", "🖼️ Media Optimizer"])
     
+    # --- TEXT CLEANER TAB ---
     with tab1:
         st.write("Paste your raw text or messy HTML below.")
         raw_text = st.text_area("Input Text", height=200)
+        
         c1, c2, c3 = st.columns(3)
         processed_text = ""
         
         if c1.button("Strip HTML"):
             soup = BeautifulSoup(raw_text, "html.parser")
             processed_text = soup.get_text(separator=" ")
+            
         if c2.button("Fix Spacing"):
             processed_text = re.sub(r'\s+', ' ', raw_text).strip()
+            
         if c3.button("MD to HTML"):
             processed_text = markdown.markdown(raw_text)
 
@@ -131,16 +118,14 @@ elif app_mode == "Webmaster Toolkit":
             st.success("Done!")
             st.text_area("Result", value=processed_text, height=200)
 
+    # --- MEDIA OPTIMIZER TAB ---
     with tab2:
         st.write("Convert heavy images to WebP.")
-        # Need to import PIL and io only here to save memory
-        from PIL import Image
-        import io
-        
         uploaded_file = st.file_uploader("Upload Image", type=['png', 'jpg', 'jpeg'])
+        
         if uploaded_file:
             original_image = Image.open(uploaded_file)
-            st.image(original_image, caption="Original", use_container_width=True) # Updated for Streamlit 1.39+
+            st.image(original_image, caption="Original", use_column_width=True)
             
             if st.button("Convert to WebP"):
                 buffer = io.BytesIO()