Update app.py

app.py

@@ -2,38 +2,10 @@ import os
 import time
 import uuid
 import threading
-import socket
-import json
-import urllib.request
 from flask import Flask, request, jsonify, send_file, render_template_string
 from flask_cors import CORS
 import yt_dlp
-
-# ==========================================
-# دور زدن محدودیت و قطعی DNS در هاگینگ‌فیس
-# ==========================================
-_orig_getaddrinfo = socket.getaddrinfo
-
-def patched_getaddrinfo(host, port, family=0, type=0, proto=0, flags=0):
-    blocked_domains = ['youtube', 'youtu.be', 'googlevideo', 'instagram', 'cdninstagram']
-    if host and any(d in host.lower() for d in blocked_domains):
-        try:
-            url = f"https://dns.google/resolve?name={host}&type=A"
-            req = urllib.request.Request(url, headers={'User-Agent': 'Mozilla/5.0'})
-            with urllib.request.urlopen(req, timeout=5) as response:
-                data = json.loads(response.read().decode('utf-8'))
-                if 'Answer' in data:
-                    for ans in data['Answer']:
-                        if ans['type'] == 1:
-                            ip = ans['data']
-                            return [(socket.AF_INET, type or socket.SOCK_STREAM, proto or 6, '', (ip, port))]
-        except Exception as e:
-            print(f"DoH fallback failed for {host}: {e}")
-    
-    return _orig_getaddrinfo(host, port, family, type, proto, flags)
-
-socket.getaddrinfo = patched_getaddrinfo
-# ==========================================
+from pytubefix import YouTube
 
 app = Flask(__name__)
 CORS(app)
@@ -105,7 +77,7 @@ HTML_TEMPLATE = """
             if (!url) return alert('لطفا لینک را وارد کنید');
 
             btn.disabled = true;
-            status.innerHTML = "⏳ در حال استخراج و دانلود ویدیو از سرور اصلی...";
+            status.innerHTML = "⏳ در حال استخراج و دانلود ویدیو از سرور... (این فرآیند ممکن است کمی طول بکشد)";
             videoContainer.style.display = 'none';
 
             try {
@@ -153,50 +125,57 @@ def download_video():
         
     url = data['url']
     file_id = str(uuid.uuid4())
-    output_template = os.path.join(DOWNLOAD_FOLDER, f"{file_id}.%(ext)s")
+    final_filename = f"{file_id}.mp4"
+    final_path = os.path.join(DOWNLOAD_FOLDER, final_filename)
     
-    # استفاده از کلاینت TV برای دور زدن محدودیت لاگین
-    ydl_opts = {
-        'format': 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best',
-        'outtmpl': output_template,
-        'quiet': True,
-        'no_warnings': True,
-        'merge_output_format': 'mp4',
-        'force_ipv4': True,
-        'nocheckcertificate': True,
-        # هدرهای شبیه‌سازی مرورگر
-        'http_headers': {
-            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
-        },
-        # استراتژی جدید: استفاده از کلاینت TV که معمولاً نیاز به لاگین ندارد
-        'extractor_args': {
-            'youtube': {
-                'player_client': ['tv', 'android_creator', 'ios'],
-                'player_skip': ['webpage']
-            }
-        },
-    }
+    # تشخیص یوتیوب بودن لینک
+    is_youtube = 'youtube.com' in url.lower() or 'youtu.be' in url.lower()
 
     try:
+        # 🟢 روش اول: استفاده از pytubefix مخصوص دور زدن ربات یوتیوب
+        if is_youtube:
+            try:
+                # گزینه use_po_token باعث فرار از خطای ربات (Sign in to confirm) می‌شود
+                yt = YouTube(url, use_po_token=True)
+                ys = yt.streams.get_highest_resolution()
+                ys.download(output_path=DOWNLOAD_FOLDER, filename=final_filename)
+                
+                if os.path.exists(final_path):
+                    return send_file(final_path, as_attachment=True, download_name=f"youtube_{int(time.time())}.mp4")
+            except Exception as yt_err:
+                print(f"Pytubefix method failed, switching to yt-dlp... Error: {yt_err}")
+                # در صورت شکست، به روش دوم (yt-dlp) هدایت می‌شود
+                pass
+
+        # 🟡 روش دوم: استفاده از yt-dlp برای اینستاگرام و سایت‌های دیگر
+        ydl_opts = {
+            'format': 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best',
+            'outtmpl': os.path.join(DOWNLOAD_FOLDER, f"{file_id}.%(ext)s"),
+            'quiet': True,
+            'no_warnings': True,
+            'merge_output_format': 'mp4',
+            'nocheckcertificate': True,
+        }
+        
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
             info_dict = ydl.extract_info(url, download=True)
             downloaded_ext = info_dict.get('ext', 'mp4')
-            final_filename = f"{file_id}.{downloaded_ext}"
+            dl_filename = f"{file_id}.{downloaded_ext}"
             
-            if not os.path.exists(os.path.join(DOWNLOAD_FOLDER, final_filename)):
-                final_filename = f"{file_id}.mp4"
+            if not os.path.exists(os.path.join(DOWNLOAD_FOLDER, dl_filename)):
+                dl_filename = final_filename
                 
-            final_path = os.path.join(DOWNLOAD_FOLDER, final_filename)
+            dl_path = os.path.join(DOWNLOAD_FOLDER, dl_filename)
             
-            if os.path.exists(final_path):
-                return send_file(final_path, as_attachment=True, download_name=f"video_{int(time.time())}.mp4")
+            if os.path.exists(dl_path):
+                return send_file(dl_path, as_attachment=True, download_name=f"video_{int(time.time())}.mp4")
             else:
-                return jsonify({'error': 'File not found after download'}), 500
+                return jsonify({'error': 'خطا در ذخیره فایل نهایی.'}), 500
 
     except Exception as e:
         error_msg = str(e)
         if "Sign in to confirm" in error_msg or "Private video" in error_msg:
-            return jsonify({'error': 'متاسفانه یوتیوب آی‌پی سرور را محدود کرده است (Sign-in required).'}), 403
+            return jsonify({'error': 'یوتیوب دسترسی به این ویدیو را در سرورهای ابری مسدود کرده است.'}), 403
         return jsonify({'error': error_msg}), 500
 
 if __name__ == '__main__':