Update streamlit_app.py

streamlit_app.py

@@ -4,6 +4,7 @@ import base64
 import pickle
 import subprocess
 import yt_dlp
+import streamlit as st # للوصول للـ Secrets
 from googleapiclient.discovery import build
 
 # ==========================================
@@ -13,56 +14,56 @@ YOUTUBE_STREAM_URL = "rtmp://a.rtmp.youtube.com/live2/skqf-rd92-dfa7-74cx-bxr2"
 TIKTOK_LIVE_URL = "https://www.tiktok.com/@abdulaziz999989/live"
 
 # ==========================================
-# 2. تهيئة YouTube API
+# 2. تجهيز ملف الكوكيز مؤقتاً
 # ==========================================
-def get_youtube_client():
-    b64_token = os.environ.get("YT_TOKEN_B64")
-    pickle_bytes = base64.b64decode(b64_token)
-    credentials = pickle.loads(pickle_bytes)
-    return build('youtube', 'v3', credentials=credentials)
+def setup_cookies():
+    # افترضنا أنك وضعت الكوكيز في الـ Secrets باسم TIKTOK_COOKIES
+    cookies_data = st.secrets.get("TIKTOK_COOKIES")
+    if cookies_data:
+        with open("cookies.txt", "w") as f:
+            f.write(cookies_data)
+        return "cookies.txt"
+    return None
 
 # ==========================================
-# 3. سحب رابط تيك توك المباشر باستخدام yt-dlp
+# 3. سحب الرابط مع الكوكيز
 # ==========================================
-def get_tiktok_stream_url():
+def get_tiktok_stream_url(cookie_file):
     ydl_opts = {
         'format': 'best',
         'quiet': True,
-        'no_warnings': True,
+        'cookiefile': cookie_file, # هنا نقوم بتمرير ملف الكوكيز
+        'user_agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36'
     }
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
         try:
             info = ydl.extract_info(TIKTOK_LIVE_URL, download=False)
             return info.get('url')
         except Exception as e:
-            print(f"[Error] TikTok might be offline or blocked: {e}")
+            print(f"[Error] TikTok connection failed (Cookies might be expired): {e}")
             return None
 
 # ==========================================
-# 4. إعادة البث (Restreaming)
+# 4. التنفيذ
 # ==========================================
 def run_restreamer():
-    print("[System] Starting Restreaming Radar...")
+    cookie_file = setup_cookies()
+    print(f"[System] Starting Restreaming Radar using cookies: {cookie_file}")
     
     while True:
-        tiktok_url = get_tiktok_stream_url()
+        tiktok_url = get_tiktok_stream_url(cookie_file)
         
         if tiktok_url:
-            print("[System] TikTok stream found! Starting FFmpeg...")
-            
-            # أمر FFmpeg لسحب الرابط وإعادة توجيهه ليوتيوب
+            print("[System] Stream link captured! Starting FFmpeg...")
             ffmpeg_cmd = [
                 'ffmpeg', '-re', '-i', tiktok_url,
                 '-c:v', 'libx264', '-preset', 'veryfast', '-b:v', '2000k',
                 '-c:a', 'aac', '-b:a', '128k', '-f', 'flv', YOUTUBE_STREAM_URL
             ]
-            
-            process = subprocess.Popen(ffmpeg_cmd, stdout=subprocess.DEVNULL, stderr=subprocess.STDOUT)
-            process.wait() # انتظر حتى ينتهي البث
-            print("[System] Stream ended. Restarting...")
+            subprocess.run(ffmpeg_cmd)
         else:
-            print("[System] TikTok is not live. Waiting 30 seconds...")
+            print("[System] Waiting for live stream...")
             time.sleep(30)
 
 if __name__ == "__main__":
-    run_restreamer()
+    run_restreamer()