Update app.py

app.py

@@ -1,8 +1,9 @@
 import streamlit as st
-import streamlit.components.v1 as components
+import requests
+import json
+import re
 from bs4 import BeautifulSoup
 import markdown
-import re
 
 # --- PAGE CONFIG ---
 st.set_page_config(
@@ -12,7 +13,7 @@ st.set_page_config(
     initial_sidebar_state="expanded"
 )
 
-# --- SIDEBAR ---
+# --- SIDEBAR & SERVER SELECTOR ---
 st.sidebar.title("Navigation")
 query_params = st.query_params
 default_index = 1 if query_params.get("mode") == "downloader" else 0
@@ -23,91 +24,127 @@ 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 (Swarm Mode)
+# VIEW 1: YOUTUBE DOWNLOADER (v10 API)
 # ==========================================
 if app_mode == "YouTube Downloader":
-    # main.py
-# Hugging Face Space: YouTube → MP3 API + UI
-# Works on CPU, no API key, no login
-
-import os
-import uuid
-from fastapi import FastAPI
-from fastapi.responses import FileResponse
-import yt_dlp
-import gradio as gr
-
-OUTPUT_DIR = "downloads"
-os.makedirs(OUTPUT_DIR, exist_ok=True)
-
-app = FastAPI()
-
-
-def download_mp3(youtube_url: str):
-    uid = str(uuid.uuid4())
-    out_file = os.path.join(OUTPUT_DIR, f"{uid}.mp3")
-
-    ydl_opts = {
-        'format': 'bestaudio/best',
-        'outtmpl': out_file.replace('.mp3', '.%(ext)s'),
-        'postprocessors': [{
-            'key': 'FFmpegExtractAudio',
-            'preferredcodec': 'mp3',
-            'preferredquality': '192',
-        }],
-        'quiet': True,
-        'noplaylist': True,
-    }
-
-    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-        ydl.download([youtube_url])
-
-    return out_file
-
-
-# ---------- API ENDPOINT ----------
-@app.get("/mp3")
-def mp3_api(url: str):
-    mp3_path = download_mp3(url)
-    return FileResponse(mp3_path, filename="audio.mp3", media_type="audio/mpeg")
-
-
-# ---------- UI ----------
-def ui_download(url):
-    mp3_path = download_mp3(url)
-    return mp3_path
-
-ui = gr.Interface(
-    fn=ui_download,
-    inputs=gr.Textbox(label="YouTube URL"),
-    outputs=gr.File(label="Download MP3"),
-    title="YouTube → MP3 Converter",
-    description="Paste YouTube link and get MP3 audio"
-)
-
-
-if __name__ == "__main__":
-    ui.launch(server_name="0.0.0.0", server_port=7860)
-
-# -----------------------------
-# requirements.txt (create this file manually):
-# fastapi
-# uvicorn
-# yt-dlp
-# gradio
-# ffmpeg-python
-
+    st.title("🎥 YouTube Media Extractor")
+    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:
+        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...")
         
-        components.html(html_code, height=350)
+        try:
+            # v10 API Payload Structure
+            headers = {
+                "Accept": "application/json",
+                "Content-Type": "application/json",
+                "User-Agent": "LexicalTools/1.0"
+            }
+            
+            payload = {
+                "url": url,
+                # v10 uses 'downloadMode' instead of 'isAudioOnly'
+                "downloadMode": "audio" if "Audio" in format_type else "auto",
+                "videoQuality": "1080",
+                "filenameStyle": "basic"
+            }
+
+            # 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!")
+                
+                # Direct Stream Link (No server storage used)
+                st.link_button(f"⬇️ Click to Download {format_type}", download_link)
+                
+                # 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']}")
+            else:
+                st.error("Unknown error. Try switching the server in the sidebar.")
+
+        except Exception as e:
+            st.error(f"Connection Failed: {str(e)}")
+            st.info("💡 Tip: Try selecting a different server in the Sidebar.")
 
 # ==========================================
-# VIEW 2: WEBMASTER TOOLKIT (Server Side)
+# VIEW 2: WEBMASTER TOOLKIT (Existing)
 # ==========================================
 elif app_mode == "Webmaster Toolkit":
     st.title("⚡ Webmaster's Toolkit")
-    # ... (Keep your existing toolkit logic here if needed) ...
-    tab1, tab2 = st.tabs(["📝 Text Cleaner", "🖼️ Media Optimizer"])
-    # Just a placeholder for the logic you already have
+    
+    tab1, tab2 = st.tabs(["📝 Text Cleaner & SEO", "🖼️ Media Optimizer"])
+    
     with tab1:
-        st.write("Text Cleaner Active")
+        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)
+
+        if processed_text:
+            st.success("Done!")
+            st.text_area("Result", value=processed_text, height=200)
+
+    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+
+            
+            if st.button("Convert to WebP"):
+                buffer = io.BytesIO()
+                original_image.save(buffer, format="WEBP", quality=80, optimize=True)
+                buffer.seek(0)
+                st.success("Converted!")
+                st.download_button("Download WebP", data=buffer, file_name="image.webp")