Update streamlit_app.py

streamlit_app.py

@@ -12,7 +12,7 @@ from googleapiclient.discovery import build
 from TikTokLive import TikTokLiveClient
 
 # ══════════════════════════════════════════════════════════════════════════════ #
-# الإعدادات
+# الإعدادات العامة
 # ══════════════════════════════════════════════════════════════════════════════ #
 TIKTOK_USERNAME = "@vimtoo.2" 
 TOKEN = "/tmp/yt.pkl"
@@ -22,7 +22,7 @@ POLL_INTERVAL = 60
 GRACE_PERIOD_SECONDS = 120 
 
 # ══════════════════════════════════════════════════════════════════════════════ #
-# دالة تنظيف وتجهيز الكوكيز الحساسة فقط لمنع خطأ 400 Bad Request
+# دالة تنظيف وتجهيز الكوكيز الحساسة لمنع خطأ 400 Bad Request
 # ══════════════════════════════════════════════════════════════════════════════ #
 def get_clean_tiktok_cookies():
     raw_cookies = os.getenv("TIKTOK_COOKIES")
@@ -56,32 +56,54 @@ def get_clean_tiktok_cookies():
     
     return raw_cookies[:200] if len(raw_cookies) > 200 else raw_cookies
 
+# ══════════════════════════════════════════════════════════════════════════════ #
+# دالة تصفير وإنهاء أي بث ديناميكي سابق معلق في الحساب لضمان نظافة السيرفر
+# ══════════════════════════════════════════════════════════════════════════════ #
+def cleanup_old_broadcasts(yt):
+    try:
+        res = yt.liveBroadcasts().list(broadcastType="all", mine=True, part="id,status").execute()
+        if res.get("items"):
+            for item in res["items"]:
+                b_id = item["id"]
+                life_status = item["status"].get("lifeCycleStatus")
+                if life_status in ["live", "testing", "ready", "liveStarting"]:
+                    try:
+                        yt.liveBroadcasts().transition(broadcastStatus="complete", id=b_id, part="status").execute()
+                        print(f"[API Cleanup] Force closed old stuck broadcast: {b_id}")
+                    except Exception:
+                        pass
+    except Exception as e:
+        print(f"[API Cleanup] Error cleaning old broadcasts: {e}")
+
 # ══════════════════════════════════════════════════════════════════════════════ #
 # دالة إنشاء البث الحي وتوليد مفتاح البث تلقائياً عبر الـ API
 # ══════════════════════════════════════════════════════════════════════════════ #
 def create_dynamic_youtube_stream(yt):
     try:
+        # تنظيف الحساب أولاً من أي بث عالق قبل البدء بالجديد
+        cleanup_old_broadcasts(yt)
+        
         current_date = datetime.now().strftime("%d-%m-%Y")
         
-        # 1. إنشاء البث (Broadcast) وتحديد العنوان والوصف الافتراضي
+        # 1. إنشاء البث (Broadcast) وتحديد العناوين والخصوصية (عام للجميع)
         broadcast_body = {
             "snippet": {
-                "title": f"بث مباشر ومستمر | {current_date}",
-                "description": f"بث تلقائي مؤرشف بتاريخ: {current_date}\n#بث_مباشر #تيكتوك",
-                "categoryId": "22" # فئة الأشخاص والمدونات العامة
+                "title": f"تغطية مباشرة ومستمر | أحمد عبدالله",
+                "description": f"بث تلقائي آلي مؤرشف بتاريخ: {current_date}\n#بث_مباشر #أحمد_عبدالله",
+                "categoryId": "22"
             },
             "status": {
-                "privacyStatus": "public", # البث عام للجميع
+                "privacyStatus": "public",
                 "selfDeclaredCreativeCommons": False
             }
         }
         broadcast_res = yt.liveBroadcasts().insert(part="snippet,status", body=broadcast_body).execute()
         broadcast_id = broadcast_res["id"]
 
-        # 2. إنشاء نقطة تدفق البيانات (Live Stream) لتوليد مفتاح البث تلقائياً
+        # 2. إنشاء نقطة تدفق البيانات وتوليد الـ Stream Key ديناميكياً
         stream_body = {
             "snippet": {
-                "title": f"Stream Key For - {current_date}"
+                "title": f"Automated Key - {current_date}"
             },
             "cdn": {
                 "frameRate": "variable",
@@ -91,32 +113,62 @@ def create_dynamic_youtube_stream(yt):
         }
         stream_res = yt.liveStreams().insert(part="snippet,cdn", body=stream_body).execute()
         stream_id = stream_res["id"]
-        
-        # استخراج مفتاح البث السري المولد ديناميكياً
         stream_key = stream_res["cdn"]["ingestionInfo"]["streamName"]
 
         # 3. ربط البث بمفتاح البث المولد
         yt.liveBroadcasts().bind(id=broadcast_id, part="id", streamId=stream_id).execute()
         
-        print(f"[API] Created Dynamic Stream Successfully. Key: {stream_key[:5]}...")
+        print(f"[API] Created Dynamic Stream. Broadcast ID: {broadcast_id}, Key Generated.")
         return broadcast_id, stream_key
     except Exception as e:
         print(f"Error in creating dynamic stream: {e}")
         return None, None
 
 # ══════════════════════════════════════════════════════════════════════════════ #
-# دالة إنهاء البث الحالي وحذفه من قائمة البثوث النشطة ليتحول فوراً لأرشيف
+# دالة الانتقال بالبث إلى وضع "الحي والمباشر" (تحل مشكلة عدم الظهور)
+# ══════════════════════════════════════════════════════════════════════════════ #
+def transition_to_live_status(yt, broadcast_id):
+    # نمنح ffmpeg مهلة 5 ثوانٍ لضخ أولى حزم البيانات ليقبل يوتيوب التحويل
+    time.sleep(5)
+    for _ in range(3): # محاولة التكرار في حال تأخر معالجة يوتيوب للإشارة
+        try:
+            yt.liveBroadcasts().transition(broadcastStatus="live", id=broadcast_id, part="status").execute()
+            print(f"[API] Broadcast {broadcast_id} is now successfully LIVE on YouTube!")
+            break
+        except Exception as e:
+            print(f"[API] Waiting for stream data to switch live... Retrying. ({e})")
+            time.sleep(4)
+
+# ══════════════════════════════════════════════════════════════════════════════ #
+# دالة إنهاء البث الحالي وحفظ الأرشيف ملحقاً بالتاريخ في نهايته
 # ══════════════════════════════════════════════════════════════════════════════ #
 def close_dynamic_youtube_stream(yt, broadcast_id):
     try:
         if not broadcast_id: return
+        
+        # جلب البيانات الحالية لإضافة التاريخ قبل الإغلاق التام
+        try:
+            b_data = yt.liveBroadcasts().list(id=broadcast_id, part="snippet").execute()
+            if b_data.get("items"):
+                snippet = b_data["items"][0]["snippet"]
+                current_date = datetime.now().strftime("%d-%m-%Y")
+                
+                # إضافة التاريخ في نهاية السطر المكتوب
+                snippet["title"] = f"{snippet.get('title', 'بث مباشر')} - {current_date}"
+                snippet["description"] = f"{snippet.get('description', '')}\n\nتم إنهاء الأرشفة بنجاح بتاريخ: {current_date}"
+                
+                yt.liveBroadcasts().update(part="snippet", body={"id": broadcast_id, "snippet": snippet}).execute()
+        except Exception:
+            pass
+
+        # تحويل حالة البث إلى مكتمل ليغلق الغرفة تماماً ويحفظ الفيديو بالقناة
         yt.liveBroadcasts().transition(broadcastStatus="complete", id=broadcast_id, part="status").execute()
         print(f"[API] Broadcast {broadcast_id} closed and saved to archive.")
     except Exception as e:
         print(f"Error closing dynamic stream: {e}")
 
 # ══════════════════════════════════════════════════════════════════════════════ #
-# م��رك المراقبة المدمج بنظام المفاتيح الذكية
+# محرك المراقبة المدمج بنظام الجلسات الذكية
 # ══════════════════════════════════════════════════════════════════════════════ #
 def run_monitor():
     lock_file = open(LOCK_FILE, 'w')
@@ -150,9 +202,9 @@ def run_monitor():
             if is_live:
                 offline_start_time = None
                 if not is_streaming:
-                    with open(STATUS, "w") as f: json.dump({"live": False, "msg": "تم رصد بث جديد! جاري إنشاء مفتاح بث ذكي على يوتيوب..."}, f)
+                    with open(STATUS, "w") as f: json.dump({"live": False, "msg": "تم رصد بث جديد! جاري إنشاء غرفة بث وتوليد مفتاح ذكي على يوتيوب..."}, f)
                     
-                    # 🔥 توليد البث والمفتاح ديناميكياً لتجنب التعليق نهائياً
+                    # توليد البث والمفتاح ديناميكياً لتجنب التعليق
                     broadcast_id, dynamic_key = create_dynamic_youtube_stream(yt)
                     
                     if dynamic_key:
@@ -160,7 +212,7 @@ def run_monitor():
                         cookie_string = get_clean_tiktok_cookies()
                         header_option = f'--http-header "Cookie={cookie_string}"' if cookie_string else ""
                         
-                        # تمرير المفتاح الديناميكي المولد ذكياً إلى الـ ffmpeg
+                        # إطلاق عملية سحب وضخ الإشارة إلى يوتيوب
                         cmd = (
                             f'streamlink --loglevel error {header_option} "https://www.tiktok.com/{TIKTOK_USERNAME}/live" best --hls-live-edge 3 --stdout | '
                             f"ffmpeg -hide_banner -loglevel error -thread_queue_size 8192 -i pipe:0 "
@@ -172,8 +224,11 @@ def run_monitor():
                         process = subprocess.Popen(cmd, shell=True)
                         is_streaming = True
                         with open(STATUS, "w") as f: json.dump({"live": True}, f)
+                        
+                        # تفعيل خيط مستقل لتحديث الحالة لـ LIVE على يوتيوب دون تعطيل عملية الضخ الحالية
+                        threading.Thread(target=transition_to_live_status, args=(yt, broadcast_id), daemon=True).start()
                     else:
-                        with open(STATUS, "w") as f: json.dump({"live": False, "msg": "فشل إنشاء مفتاح البث، جاري المحاولة لاحقاً..."}, f)
+                        with open(STATUS, "w") as f: json.dump({"live": False, "msg": "فشل إنشاء البث، جاري المحاولة لاحقاً..."}, f)
             
             else:
                 if is_streaming:
@@ -192,16 +247,16 @@ def run_monitor():
         except Exception:
             time.sleep(POLL_INTERVAL)
 
-# تشغيل الخيط
+# تشغيل خيط الرادار والمراقبة
 if "core_monitor_initialized" not in st.session_state:
     threading.Thread(target=run_monitor, name="CoreTikTokMonitor", daemon=True).start()
     st.session_state.core_monitor_initialized = True
 
 # ══════════════════════════════════════════════════════════════════════════════ #
-# الواجهة
+# واجهة العرض (Streamlit UI)
 # ══════════════════════════════════════════════════════════════════════════════ #
 st.set_page_config(page_title="AutoStream Pro", page_icon="📡")
-st.title("📡 رادار البث الذكي الآلي بالكامل")
+st.title("📡 رادار البث الآلي الذكي بالكامل")
 
 def get_status():
     if not os.path.exists(STATUS) or os.path.getsize(STATUS) == 0:
@@ -212,6 +267,6 @@ def get_status():
 
 s = get_status()
 if s.get("live"):
-    st.success(f"🔴 حساب {TIKTOK_USERNAME} مباشر الآن (نظام المفاتيح الديناميكية الذكية مفعل)")
+    st.success(f"🔴 حساب {TIKTOK_USERNAME} مباشر الآن على يوتيوب بنجاح (النظام المطور)")
 else:
     st.info(s.get("msg", "بانتظار المذيع..."))