Update main.py

main.py

@@ -77,6 +77,10 @@ selected_user_agent = get_desktop_user_agent()
 
 headers = {'User-Agent': selected_user_agent}
 
+# تحديد مسار ملف الكوكيز داخل حاوية Docker
+# تأكد من أن هذا المسار يتطابق مع المكان الذي قمت برفع cookies.txt إليه في Hugging Face Space
+COOKIES_FILE_PATH = '/app/cookies.txt' # افتراض أن cookies.txt في الجذر /app/
+
 ydl_opts_video_info = {
     'quiet': True,
     'skip_download': True,
@@ -85,7 +89,8 @@ ydl_opts_video_info = {
     'no_warnings': True,
     'age_limit': 18,
     'force_generic_extractor': False,
-    'http_headers': headers
+    'http_headers': headers,
+    'cookiefile': COOKIES_FILE_PATH # إضافة خيار ملف الكوكيز لـ yt-dlp
 }
 
 # --- النموذج ---
@@ -116,7 +121,8 @@ def preprocess_text(text):
 
 # --- معالجة فيديو واحد فقط (نسخة 2) ---
 def process_single_video2(video_url, loaded_quality_model, loaded_sentiment_pipeline, max_comments_per_video=50):
-    downloader = YoutubeCommentDownloader()
+    # تمرير مسار ملف الكوكيز إلى YoutubeCommentDownloader
+    downloader = YoutubeCommentDownloader(cookies_file=COOKIES_FILE_PATH)
 
     video_id = extract_video_id(video_url)
     if not video_id:
@@ -228,7 +234,7 @@ async def evaluate_youtube_playlist_individually_same_method2(youtube_url: str,
 
     video_links = []
     try:
-        with yt_dlp.YoutubeDL({'extract_flat': True, 'quiet': True, 'playlist_items': '1:10'}) as ydl:
+        with yt_dlp.YoutubeDL({'extract_flat': True, 'quiet': True, 'playlist_items': '1:10', 'cookiefile': COOKIES_FILE_PATH}) as ydl: # إضافة cookiefile هنا
             playlist_info = ydl.extract_info(youtube_url, download=False)
             if 'entries' in playlist_info:
                 for entry in playlist_info['entries'][:10]:
@@ -255,12 +261,12 @@ async def evaluate_youtube_playlist_individually_same_method2(youtube_url: str,
             ) for video_url in video_links
         ]
 
-        for future in futures: # استخدام futures مباشرة بدون tqdm
+        for future in futures:
             result = future.result()
             if result:
                 individual_results.append(result)
-            # تأخير إضافي بين معالجة نتائج الفيديوهات (اختياري)
-            time.sleep(random.uniform(0.5, 1.5)) # تأخير بين 0.5 و 1.5 ثانية
+            # تأخير إضافي بين معالجة نتائج الفيديوهات
+            time.sleep(random.uniform(0.5, 1.5))
 
     # --- الإحصاء النهائي ---
     num_good_videos = sum(1 for r in individual_results if r and r.get('quality') == 'جيد')