Rename stop to app.py

stop → app.py

@@ -2,14 +2,18 @@ import yt_dlp
 import os
 import gradio as gr
 import random
+import time
 import threading
+import datetime
+import re
 import requests
 from gradio_client import Client
 from huggingface_hub import InferenceClient
 
 # =====================================================
-# CONFIG & CLIENTS
+# CONFIG & SECRETS
 # =====================================================
+# Downloader Config
 PROXY_SPACE_URL = "lexicalspace/Proxy-Server"
 OUTPUT_DIR = "downloads"
 os.makedirs(OUTPUT_DIR, exist_ok=True)
@@ -17,18 +21,26 @@ os.makedirs(OUTPUT_DIR, exist_ok=True)
 MAX_CONCURRENCY = 6   
 MAX_QUEUE = 20        
 
-# SEO Tool Configuration
-REPO_ID = "Qwen/Qwen2.5-Coder-32B-Instruct"
-seo_client = InferenceClient(model=REPO_ID)
+# SEO Config
+# Ensure 'HF_TOKEN' is set in your Space Settings > Secrets
+HF_TOKEN = os.getenv("HF_TOKEN")
+SEO_MODEL_ID = "Qwen/Qwen2.5-Coder-32B-Instruct"
+
+# Initialize SEO Client
+# We wrap this in a try/except to prevent crash if token is missing
+try:
+    seo_client = InferenceClient(api_key=HF_TOKEN)
+except Exception:
+    seo_client = None
 
 # =====================================================
-# GLOBAL STATE
+# GLOBAL STATE (Downloader)
 # =====================================================
 active_jobs = {}
 job_lock = threading.Lock()
 
 # =====================================================
-# DOWNLOADER HELPERS & LOGIC
+# PART 1: DOWNLOADER LOGIC
 # =====================================================
 def get_proxy_batch():
     try:
@@ -78,111 +90,186 @@ def build_format_selector(quality_mode):
     return {}
 
 def fetch_metadata(url):
-    if not url:
-        return None, "⚠️ Please enter a URL first.", None, None, None
+    if not url: return None, "⚠️ Please enter a URL first.", None, None, None
     api_url = f"https://noembed.com/embed?url={url}"
     try:
         response = requests.get(api_url, timeout=10)
         data = response.json()
-        if "error" in data:
-            return None, f"❌ API Error: {data.get('error')}", gr.update(interactive=False), None, None
+        if "error" in data: return None, f"❌ API Error: {data.get('error')}", gr.update(interactive=False), None, None
+        
         title = data.get('title', 'Unknown Title')
         uploader = data.get('author_name', 'Unknown Channel')
         thumb = data.get('thumbnail_url', None)
-        desc = f"### 🎬 {title}\n**👤 Channel:** {uploader}\n**⏱️ Duration:** N/A\n**👁️ Views:** N/A"
+        desc = f"### 🎬 {title}\n**👤 Channel:** {uploader}\n**⏱️ Duration:** N/A (API Limit)\n**👁️ Views:** N/A (API Limit)"
         safe_title = "".join([c for c in title if c.isalnum() or c in " ._-"])
         return thumb, desc, gr.update(interactive=True), gr.update(value=safe_title), "✅ Metadata fetched"
     except Exception as e:
         return None, f"❌ Error: {str(e)}", gr.update(interactive=False), None, None
 
 def run_downloader(engine_name, url, quality, start_time, end_time, custom_name, cookies, progress):
-    log = [f"🚀 Engine: {engine_name}"]
+    log = [f"🚀 Engine: {engine_name} | Quality: {quality}"]
     with job_lock:
         if active_jobs.get(url) == "running": return None, "⛔ Already running", "Blocked"
         active_jobs[url] = "running"
     
     format_opts = build_format_selector(quality)
     filename_tmpl = f"{custom_name}.%(ext)s" if custom_name else '%(title)s.%(ext)s'
+    
     proxies = get_proxy_batch() if "V1" in engine_name else ([get_best_proxy()] if get_best_proxy() else [None])
-    strategies = ["android", "ios", "web"]
+    strategies = ["android", "ios", "web"] if "V1" in engine_name else ["android", "ios", "web", "tv"]
     
     s_sec, e_sec = parse_time(start_time), parse_time(end_time)
     download_ranges = (lambda info, ydl: [{'start_time': s_sec or 0, 'end_time': e_sec or float('inf')}]) if (s_sec or e_sec) else None
 
-    success, final_file = False, None
+    success, final_file, cookie_path = False, None, None
+    if cookies:
+        cookie_path = f"cookies_{random.randint(1000,9999)}.txt"
+        with open(cookie_path, "w") as f: f.write(cookies)
+
     for proxy in proxies:
         if success: break
         for strat in strategies:
+            if success: break
             ydl_opts = {
-                'outtmpl': os.path.join(OUTPUT_DIR, filename_tmpl),
-                'noplaylist': True, 'quiet': True, 'force_ipv4': True,
+                'outtmpl': os.path.join(OUTPUT_DIR, filename_tmpl), 'noplaylist': True, 'quiet': True, 'force_ipv4': True,
                 'extractor_args': {'youtube': {'player_client': [strat]}}, **format_opts
             }
             if proxy: ydl_opts['proxy'] = proxy
             if download_ranges: ydl_opts['download_ranges'] = download_ranges
+            if cookie_path: ydl_opts['cookiefile'] = cookie_path
+            
+            def hook(d):
+                if d['status'] == 'downloading':
+                    total = d.get('total_bytes') or d.get('total_bytes_estimate')
+                    if total: progress(d.get('downloaded_bytes', 0) / total)
+            ydl_opts['progress_hooks'] = [hook]
+
             try:
                 with yt_dlp.YoutubeDL(ydl_opts) as ydl:
                     info = ydl.extract_info(url, download=True)
                     name = ydl.prepare_filename(info)
                     base, _ = os.path.splitext(name)
                     final_file = f"{base}.{format_opts.get('ext', 'mp4')}"
-                    if os.path.exists(final_file): success = True; break
+                    if not os.path.exists(final_file):
+                        for f in os.listdir(OUTPUT_DIR):
+                            if f.startswith(os.path.basename(base)): final_file = os.path.join(OUTPUT_DIR, f); break
+                    if os.path.exists(final_file): success = True
             except: continue
             
+    if cookie_path and os.path.exists(cookie_path): os.remove(cookie_path)
     with job_lock: active_jobs.pop(url, None)
     return (final_file, "✅ Complete", "\n".join(log)) if success else (None, "❌ Failed", "\n".join(log))
 
+def download_router(mode, quality, start, end, custom_name, cookies, url, progress=gr.Progress()):
+    engine = "V2 • Fast Engine" if "V2" in mode else "V1 • Stable Engine"
+    if "Auto" in mode: engine = "V2 • Fast Engine" if (len(url) > 70 or "list=" in url) else "V1 • Stable Engine"
+    return run_downloader(engine, url, quality, start, end, custom_name, cookies, progress)
+
 # =====================================================
-# SEO TOOL LOGIC
+# PART 2: SEO GENERATOR LOGIC
 # =====================================================
 def generate_seo(code_snippet, file_type):
-    if not code_snippet.strip(): return "Please paste some code first."
-    prompt = f"Expert SEO analysis for {file_type} code:\n{code_snippet}\nGenerate Title, Meta Description, Keywords, and JSON-LD."
+    if not seo_client: return "🔒 Error: 'HF_TOKEN' not found in Secrets."
+    if not code_snippet.strip(): return "⚠️ Error: Please paste some code first."
+
+    system_instruction = f"""
+    You are an expert Technical SEO Specialist. Analyze the user's {file_type} code.
+    Your Goal: Generate Google-compliant JSON-LD structured data and SEO meta tags.
+    Output Format (Strict Markdown):
+    ## SEO Metadata
+    **Title:** [Engaging Title, max 60 chars]
+    **Description:** [Summary including keywords, max 160 chars]
+    **Keywords:** [5-8 comma-separated keywords]
+    ## JSON-LD Structured Data
+    ```json
+    [Insert VALID JSON-LD here]
+    ```
+    """
     try:
-        return seo_client.text_generation(prompt, max_new_tokens=1024, temperature=0.3)
+        response = seo_client.chat_completion(
+            model=SEO_MODEL_ID,
+            messages=[{"role": "system", "content": system_instruction}, {"role": "user", "content": f"Analyze:\n{code_snippet}"}],
+            max_tokens=1500, temperature=0.2
+        )
+        return response.choices[0].message.content
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"❌ Error: {str(e)}"
 
 # =====================================================
-# COMBINED GRADIO UI
+# PART 3: COMBINED UI WITH ROUTING
 # =====================================================
-with gr.Blocks(title="Lexical Space Tools", theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🛠️ Lexical Space Multi-Tool")
+def handle_deep_link(request: gr.Request):
+    """
+    Reads the URL parameters to decide which tab to open.
+    Example: ?mode=seo -> Opens SEO Tab
+    Example: ?mode=downloader -> Opens Downloader Tab
+    """
+    params = request.query_params
+    mode = params.get("mode") or params.get("tool")
     
-    with gr.Tabs():
+    if mode == "seo":
+        return gr.Tabs(selected="seo_tab")
+    else:
+        return gr.Tabs(selected="dl_tab")
+
+# MAIN APP BLOCK
+with gr.Blocks(title="Lexical Space Tools", theme=gr.themes.Soft()) as app:
+    
+    # We assign IDs to tabs so we can select them programmatically
+    with gr.Tabs() as main_tabs:
+        
         # --- TAB 1: DOWNLOADER ---
-        with gr.TabItem("🚀 UltraMax Downloader"):
+        with gr.TabItem("🚀 UltraMax Downloader", id="dl_tab"):
+            gr.Markdown("### 📥 Universal Media Downloader")
             with gr.Row(variant="panel"):
-                url_input = gr.Textbox(label="Media URL", placeholder="Paste link here...")
-                fetch_btn = gr.Button("🔍 Fetch Info", variant="primary")
+                url_input = gr.Textbox(label="Media URL", placeholder="Paste YouTube/Insta/TikTok link...", show_label=False, scale=4)
+                fetch_btn = gr.Button("🔍 Fetch Info", variant="primary", scale=1)
+            
             with gr.Row():
                 thumb_preview = gr.Image(label="Thumbnail", interactive=False, height=180)
-                info_display = gr.Markdown("### ⏳ Waiting for link...")
-            
+                with gr.Column(scale=3):
+                    info_display = gr.Markdown("### ⏳ Waiting for link...")
+                    fetch_status = gr.Label(value="Idle", show_label=False)
+
             with gr.Row(variant="panel"):
-                quality_select = gr.Dropdown(["Audio Only (Best)", "Video (4K / Best)", "Video (1080p)", "Video (720p)"], value="Audio Only (Best)", label="Quality")
+                quality_select = gr.Dropdown(["Audio Only (Best)", "Video (4K / Best)", "Video (1080p)", "Video (720p)", "Video (360p / Data Saver)"], value="Audio Only (Best)", label="Quality")
                 engine_mode = gr.Radio(["Auto (Smart)", "V1 (Stable)", "V2 (Fast)"], value="Auto (Smart)", label="Engine")
-                custom_filename = gr.Textbox(label="Rename File", placeholder="MyFile")
-            
-            dl_btn = gr.Button("⬇️ START DOWNLOAD", variant="stop", interactive=False)
-            file_out = gr.File(label="Output")
-            log_out = gr.Textbox(label="Logs", lines=3)
+                
+            with gr.Accordion("⚙️ Advanced (Cut / Rename / Auth)", open=False):
+                with gr.Row():
+                    start_time = gr.Textbox(label="Start", placeholder="00:00")
+                    end_time = gr.Textbox(label="End", placeholder="01:30")
+                    custom_filename = gr.Textbox(label="Filename", placeholder="MyFile")
+                cookies_input = gr.Textbox(label="Cookies", placeholder="Netscape format...", lines=1)
+
+            dl_btn = gr.Button("⬇️ START DOWNLOAD", variant="stop", interactive=False, size="lg")
+            with gr.Accordion("📂 Output & Logs", open=True):
+                with gr.Row():
+                    file_out = gr.File(label="Output")
+                    log_out = gr.Textbox(label="Logs", lines=4)
 
-            fetch_btn.click(fetch_metadata, [url_input], [thumb_preview, info_display, dl_btn, custom_filename])
-            dl_btn.click(run_downloader, [engine_mode, url_input, quality_select, gr.State(""), gr.State(""), custom_filename, gr.State("")], [file_out, gr.State(), log_out])
+            # Downloader Events
+            fetch_btn.click(fetch_metadata, [url_input], [thumb_preview, info_display, dl_btn, custom_filename, fetch_status])
+            dl_btn.click(download_router, [engine_mode, quality_select, start_time, end_time, custom_filename, cookies_input, url_input], [file_out, fetch_status, log_out], concurrency_limit=MAX_CONCURRENCY)
 
         # --- TAB 2: SEO GENERATOR ---
-        with gr.TabItem("🔍 AI Code SEO"):
-            gr.Markdown("### 🚀 AI Code-to-SEO Generator")
+        with gr.TabItem("⚡ SEO Generator", id="seo_tab"):
+            gr.Markdown("### 🔍 AI SEO & JSON-LD Generator")
             with gr.Row():
                 with gr.Column():
                     input_type = gr.Radio(["python", "html"], label="File Type", value="python")
-                    code_input = gr.Code(language="python", label="Paste Code", lines=10)
-                    submit_btn = gr.Button("Generate SEO Strategy", variant="primary")
+                    code_input = gr.Code(language="python", label="Paste Code", lines=15)
+                    seo_btn = gr.Button("✨ Generate SEO Data", variant="primary", size="lg")
                 with gr.Column():
-                    output_markdown = gr.Markdown(label="Generated Result")
+                    output_markdown = gr.Markdown(label="Results")
 
+            # SEO Events
             input_type.change(lambda x: gr.Code(language=x), inputs=input_type, outputs=code_input)
-            submit_btn.click(generate_seo, [code_input, input_type], [output_markdown])
+            seo_btn.click(generate_seo, [code_input, input_type], [output_markdown])
+
+    # --- DEEP LINKING EVENT ---
+    # This runs when the page loads to check the URL
+    app.load(fn=handle_deep_link, inputs=None, outputs=main_tabs)
 
-demo.queue(max_size=MAX_QUEUE).launch()
+if __name__ == "__main__":
+    app.queue(max_size=MAX_QUEUE).launch()