Update app.py

app.py

@@ -1,9 +1,6 @@
 import streamlit as st
 import requests
 import json
-import time
-from PIL import Image
-import io
 import re
 from bs4 import BeautifulSoup
 import markdown
@@ -16,7 +13,7 @@ st.set_page_config(
     initial_sidebar_state="expanded"
 )
 
-# --- SIDEBAR NAVIGATION ---
+# --- SIDEBAR & SERVER SELECTOR ---
 st.sidebar.title("Navigation")
 query_params = st.query_params
 default_index = 1 if query_params.get("mode") == "downloader" else 0
@@ -27,27 +24,40 @@ 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 (API Version)
+# VIEW 1: YOUTUBE DOWNLOADER (v10 API)
 # ==========================================
 if app_mode == "YouTube Downloader":
     st.title("🎥 YouTube Media Extractor")
-    st.info("High-Speed Downloader (Powered by Cobalt API)")
+    st.caption(f"Powered by Cobalt v10 • Server: {selected_server_name}")
     
     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 processing server...")
+        status_area.info("⏳ Contacting remote server...")
         
         try:
-            # Cobalt API Configuration
-            api_url = "https://api.cobalt.tools/api/json"
+            # v10 API Payload Structure
             headers = {
                 "Accept": "application/json",
                 "Content-Type": "application/json",
@@ -56,37 +66,44 @@ if app_mode == "YouTube Downloader":
             
             payload = {
                 "url": url,
-                "vCodec": "h264",
-                "vQuality": "1080",
-                "aFormat": "mp3",
-                "isAudioOnly": True if "Audio" in format_type else False
+                # v10 uses 'downloadMode' instead of 'isAudioOnly'
+                "downloadMode": "audio" if "Audio" in format_type else "auto",
+                "videoQuality": "1080",
+                "filenameStyle": "basic"
             }
 
-            # Send Request
-            response = requests.post(api_url, headers=headers, json=payload)
-            data = response.json()
+            # 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()
 
+            # Handle Success
             if "url" in data:
                 download_link = data["url"]
                 status_area.success("✅ Media Ready!")
                 
-                # Show a Direct Download Link (Fastest method)
+                # Direct Stream Link (No server storage used)
                 st.link_button(f"⬇️ Click to Download {format_type}", download_link)
                 
-                # Optional: Preview
+                # Preview
                 if "Audio" not in format_type:
                     st.video(download_link)
                 else:
                     st.audio(download_link)
-                    
+            
+            # Handle API Specific Errors
             elif "text" in data:
-                # API returned an error message
-                st.error(f"Server Error: {data['text']}")
+                st.error(f"API Error: {data['text']}")
             else:
-                st.error("Unknown error from processing server.")
+                st.error("Unknown error. Try switching the server in the sidebar.")
 
         except Exception as e:
-            st.error(f"Connection Error: {str(e)}")
+            st.error(f"Connection Failed: {str(e)}")
+            st.info("💡 Tip: Try selecting a different server in the Sidebar.")
 
 # ==========================================
 # VIEW 2: WEBMASTER TOOLKIT (Existing)
@@ -96,21 +113,17 @@ 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)
 
@@ -118,14 +131,16 @@ 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.")
-        uploaded_file = st.file_uploader("Upload Image", type=['png', 'jpg', 'jpeg'])
+        # 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_column_width=True)
+            st.image(original_image, caption="Original", use_container_width=True) # Updated for Streamlit 1.39+
             
             if st.button("Convert to WebP"):
                 buffer = io.BytesIO()