Update app.py

app.py

@@ -2,29 +2,20 @@ import yt_dlp
 import os
 import gradio as gr
 import re
-from typing import Optional, Dict, Any, List
-import socket
-
-# Force IPv4 (fixes many container DNS issues)
-def force_ipv4():
-    orig_getaddrinfo = socket.getaddrinfo
-
-    def new_getaddrinfo(*args, **kwargs):
-        responses = orig_getaddrinfo(*args, **kwargs)
-        return [r for r in responses if r[0] == socket.AF_INET]
-
-    socket.getaddrinfo = new_getaddrinfo
-
-force_ipv4()
 
 OUTPUT_DIR = "downloads"
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 
-# 1. Modified Engine to return the FILE PATH, not just a list of names
-def core_download(url, progress=gr.Progress()):
+# 1. Modified Engine to accept Proxy
+def core_download(url, proxy_url=None, progress=gr.Progress()):
     filepaths = []
     
-    # Custom hook to update Gradio progress bar
+    # Clean up the proxy string (strip spaces)
+    if proxy_url:
+        proxy_url = proxy_url.strip()
+        if proxy_url == "":
+            proxy_url = None
+    
     def progress_hook(d):
         if d['status'] == 'downloading':
             try:
@@ -44,6 +35,9 @@ def core_download(url, progress=gr.Progress()):
         }],
         'progress_hooks': [progress_hook],
         'noplaylist': True,
+        # KEY FIXES:
+        'force_ipv4': True,  # 1. Force IPv4 to fix DNS resolution errors
+        'proxy': proxy_url,  # 2. Use the user-provided Proxy IP
     }
 
     try:
@@ -53,46 +47,46 @@ def core_download(url, progress=gr.Progress()):
             final_filename = os.path.splitext(filename)[0] + ".mp3"
             
             if os.path.exists(final_filename):
-                return final_filename, f"✅ Success: {os.path.basename(final_filename)}"
+                return final_filename, f"✅ Success: {os.path.basename(final_filename)}", "Download complete!"
             else:
-                return None, "❌ Error: File not found after download."
+                return None, "❌ Error: File not found.", "Failed."
                 
     except Exception as e:
-        return None, f"❌ Error: {str(e)}"
+        # Detailed error logging
+        return None, f"❌ Error: {str(e)}", f"Failed log: {str(e)}"
 
-# 2. UI Wrapper that yields updates
-def ui_logic(url):
+# 2. UI Wrapper
+def ui_logic(url, proxy):
     if not url:
         yield None, "Please enter a URL.", "Waiting..."
         return
 
     yield None, "Starting download...", "Initializing..."
     
-    file_path, status_msg = core_download(url)
+    file_path, status_msg, log_msg = core_download(url, proxy)
     
-    # Return the actual file path to the Gradio File component
-    yield file_path, status_msg, "Complete!"
+    yield file_path, status_msg, log_msg
 
-# 3. Improved UI Layout
+# 3. Improved UI with Proxy Field
 with gr.Blocks(title="UltraMax Downloader") as app:
     gr.Markdown("## 🎵 UltraMax YouTube to MP3")
     
     with gr.Row():
         url_input = gr.Textbox(label="YouTube URL", placeholder="Paste link here...", scale=3)
-        btn = gr.Button("Download", variant="primary", scale=1)
+        # NEW FIELD FOR IP/PROXY
+        proxy_input = gr.Textbox(label="Proxy IP (Optional)", placeholder="http://user:pass@ip:port", scale=1)
+    
+    btn = gr.Button("Download", variant="primary")
 
-    # Output components
     with gr.Row():
-        # This component lets the user actually download the file
         file_output = gr.File(label="Download MP3 Here")
-        
         with gr.Column():
             status_text = gr.Textbox(label="Status")
             log_text = gr.Textbox(label="System Log")
 
     btn.click(
         fn=ui_logic,
-        inputs=url_input,
+        inputs=[url_input, proxy_input],
         outputs=[file_output, status_text, log_text]
     )