Update streamlit_app.py

streamlit_app.py

@@ -4,78 +4,287 @@ import os
 import time
 import subprocess
 import threading
-from datetime import datetime
+import signal
+import json
+import base64
+import pickle
+from datetime import datetime, timezone, timedelta
+from google.auth.transport.requests import Request
+from googleapiclient.discovery import build
 from TikTokLive import TikTokLiveClient
 
 # ══════════════════════════════════════════════════════════════════════════════
-#  ⚙️  الإعدادات الثابتة (تعدل مرة واحدة فقط)
+# ⚙️ الإعدادات
 # ══════════════════════════════════════════════════════════════════════════════
-TIKTOK_USERNAME    = "@ah_md107"           # الحساب المراد رصده
-YOUTUBE_STREAM_KEY = "hetf-6m1q-uyaj-sjww-30qc"  # مفتاح بث يوتيوب
-POLL_INTERVAL      = 60                  # فحص كل دقيقة
-YT_RTMP_URL        = f"rtmp://a.rtmp.youtube.com/live2/{YOUTUBE_STREAM_KEY}"
+TIKTOK_USERNAME = "@ah_md107"
+TOKEN = "/tmp/yt.pkl"
+STATUS = "/tmp/status.json"
+HEARTBEAT_FILE = "/tmp/heartbeat.json"
+
+if os.getenv("TOKEN_B64"):
+    open(TOKEN, "wb").write(base64.b64decode(os.getenv("TOKEN_B64")))
+
+# ══════════════════════════════════════════════════════════════════════════════
+# 💓 نبض داخلي
+# ══════════════════════════════════════════════════════════════════════════════
+def internal_heartbeat():
+    while True:
+        try:
+            with open(HEARTBEAT_FILE, "w") as f:
+                json.dump({
+                    "last_ping": datetime.now().isoformat(),
+                    "status": "alive"
+                }, f)
+            time.sleep(30)
+        except:
+            time.sleep(30)
+
+# ══════════════════════════════════════════════════════════════════════════════
+# 🎨 عنوان + وصف فخم
+# ══════════════════════════════════════════════════════════════════════════════
+def generate_fancy_title():
+    now = datetime.now()
+    templates = [
+        "🔴 بث أحمد عبدالله العراق | {date} | {time}",
+        "⚡ LIVE: أحمد عبدالله | العراق | {date} - {time}",
+        "🎥 بث مباشر: أحمد عبدالله العراق | {date} | {time}",
+        "📡 أحمد عبدالله - بث حي | العراق | {date} | {time}",
+        "🌟 بث حصري: أحمد عبدالله | {date} | {time}",
+    ]
+    return random.choice(templates).format(
+        date=now.strftime('%Y-%m-%d'),
+        time=now.strftime('%I:%M %p')
+    )
+
+def generate_fancy_description():
+    now = datetime.now()
+    arabic_tags = [
+        "#أحمد_عبدالله", "#العراق", "#بث_مباشر", "#تيك_توك", "#يوتيوب",
+        "#بث_مباشر_الآن", "#مباشر", "#لايف", "#مشاهير_العراق",
+        "#محتوى_مباشر", "#بث_مباشر_يوتيوب", "#الموصل", "#بغداد",
+        "#بصرة", "#أنبار", "#كردستان", "#العراق_الآن", "#بث_حي",
+        "#محتوى_عراقي", "#شخصيات_عراقية", "#ترند_العراق"
+    ]
+    english_tags = [
+        "#AhmedAbdullah", "#TikTokLive", "#YouTubeLive", "#Iraq",
+        "#LiveStream", "#Streaming", "#Viral", "#Trending",
+        "#LiveNow", "#TikTok", "#YouTube", "#ContentCreator",
+        "#IraqiCreator", "#LiveBroadcast", "#RealTime",
+        "#MustWatch", "#LiveStreaming", "#SocialMedia",
+        "#Entertainment", "#BreakingLive"
+    ]
+    import random
+    selected_arabic = random.sample(arabic_tags, min(8, len(arabic_tags)))
+    selected_english = random.sample(english_tags, min(8, len(english_tags)))
+    return f"""🎥 بث مباشر حصري من تيك توك | تابعوا أحدث اللحظات!
+
+📡 البث مباشر الآن من حساب: @{TIKTOK_USERNAME}
+🕐 تاريخ البث: {now.strftime('%A, %d %B %Y')}
+⏰ توقيت البث: {now.strftime('%I:%M %p')}
+
+━━━━━━━━━━━━━━━━━━━━━━
+🏷️ الهاشتاقات العربية:
+{' '.join(selected_arabic)}
+
+🏷️ Arabic Hashtags:
+{' '.join(selected_english)}
+
+━━━━━━━━━━━━━━━━━━━━━━
+🔗 تابع البث المباشر الآن!
+🔄 يتم تحديث الوصف تلقائياً في كل بث جديد
+"""
 
 # ══════════════════════════════════════════════════════════════════════════════
-# 🚀 محرك العمل في الخلفية (Background Engine)
+# 🚀 YouTube API
 # ══════════════════════════════════════════════════════════════════════════════
+def start_yt_live():
+    creds = pickle.load(open(TOKEN, "rb"))
+    if creds.expired:
+        creds.refresh(Request())
+        pickle.dump(creds, open(TOKEN, "wb"))
+    
+    yt = build('youtube', 'v3', credentials=creds)
+    title = generate_fancy_title()
+    description = generate_fancy_description()
+    start_time = (datetime.now(timezone.utc) - timedelta(seconds=10)).isoformat().replace("+00:00", "Z")
+    
+    bc = yt.liveBroadcasts().insert(
+        part="snippet,status,contentDetails",
+        body={
+            "snippet": {
+                "title": title,
+                "scheduledStartTime": start_time,
+                "description": description
+            },
+            "status": {"privacyStatus": "public", "selfDeclaredMadeForKids": False},
+            "contentDetails": {
+                "enableAutoStart": True,
+                "enableAutoStop": True,
+                "enableDvr": True,
+                "recordFromStart": True,
+                "monitorStream": {
+                    "enableMonitorStream": True,
+                    "broadcastStreamDelayMs": 0
+                }
+            }
+        }
+    ).execute()
+    
+    stream = yt.liveStreams().insert(
+        part="snippet,cdn",
+        body={
+            "snippet": {"title": "TikTok Relay"},
+            "cdn": {
+                "ingestionType": "rtmp",
+                "resolution": "720p",
+                "frameRate": "30fps"
+            }
+        }
+    ).execute()
+    
+    yt.liveBroadcasts().bind(part="id", id=bc["id"], streamId=stream["id"]).execute()
+    
+    rtmp = f"{stream['cdn']['ingestionInfo']['ingestionAddress']}/{stream['cdn']['ingestionInfo']['streamName']}"
+    return bc["id"], rtmp, f"https://youtu.be/{bc['id']}", title
+
+# ══════════════════════════════════════════════════════════════════════════════
+# 🔍 كشف البث + نقل (نفس كودك)
+# ══════════════════════════════════════════════════════════════════════════════
+def kill_process_and_children(process):
+    try:
+        os.killpg(os.getpgid(process.pid), signal.SIGTERM)
+        process.wait(timeout=5)
+    except:
+        try:
+            os.killpg(os.getpgid(process.pid), signal.SIGKILL)
+        except:
+            pass
 
 def run_monitor():
-    """هذه الدالة هي التي تظل تعمل حتى لو أغلقت الصفحة"""
-    print("✅ بدأ محرك المراقبة التلقائي 24/7...")
+    print("✅ بدأ محرك المراقبة 24/7...")
+    
+    current_process = None
+    last_room_id = None
+    current_rtmp = None
+    
     while True:
         try:
-            # فحص حالة البث
             client = TikTokLiveClient(unique_id=TIKTOK_USERNAME)
             is_live = asyncio.run(client.is_live())
-
+            
+            current_room_id = None
+            if is_live:
+                try:
+                    room_info = asyncio.run(client.get_room_info())
+                    current_room_id = room_info.get("room_id") if room_info else "unknown"
+                except:
+                    current_room_id = "unknown"
+            
             if is_live:
-                print(f"[{datetime.now().strftime('%H:%M')}] 🔴 البث بدأ! جاري النقل إلى يوتيوب...")
-
-                # أمر النقل المباشر - محسّن للصوت
-                cmd = (
-                    f"streamlink https://www.tiktok.com/{TIKTOK_USERNAME}/live best --hls-live-edge 3 --stdout | "
-                    f"ffmpeg -thread_queue_size 8192 -i pipe:0 "
-                    f"-c:v copy "
-                    f"-c:a aac -ar 48000 -b:a 192k -ac 2 "
-                    f"-af 'highpass=f=200,lowpass=f=3000,aresample=async=1' "
-                    f"-buffer_size 1024k "
-                    f"-fflags +genpts+igndts "
-                    f"-max_interleave_delta 0 "
-                    f"-f flv {YT_RTMP_URL}"
-                )
-
-                process = subprocess.Popen(cmd, shell=True)
-                process.wait() # سينتظر هنا حتى ين��هي البث فعلياً
-                print("⏹️ انتهى البث، العودة لوضع المراقبة...")
+                print(f"[{datetime.now().strftime('%H:%M')}] 🔴 البث مفتوح! room_id: {current_room_id}")
+                
+                if current_room_id != last_room_id:
+                    print(f"🆕 بث جديد! {last_room_id} → {current_room_id}")
+                    
+                    if current_process and current_process.poll() is None:
+                        print("⏹️ إغلاق البث القديم...")
+                        kill_process_and_children(current_process)
+                        current_process = None
+                        time.sleep(5)
+                    
+                    # إنشاء بث يوتيوب جديد بالعنوان
+                    print("▶️ إنشاء بث يوتيوب جديد...")
+                    bid, rtmp, link, title = start_yt_live()
+                    current_rtmp = rtmp
+                    last_room_id = current_room_id
+                    
+                    # نقل البث (نفس كودك)
+                    cmd = (
+                        f"streamlink https://www.tiktok.com/{TIKTOK_USERNAME}/live best --hls-live-edge 3 --stdout | "
+                        f"ffmpeg -thread_queue_size 8192 -i pipe:0 "
+                        f"-c:v copy "
+                        f"-c:a aac -ar 48000 -b:a 192k -ac 2 "
+                        f"-af 'highpass=f=200,lowpass=f=3000,aresample=async=1' "
+                        f"-buffer_size 1024k "
+                        f"-fflags +genpts+igndts "
+                        f"-max_interleave_delta 0 "
+                        f"-f flv {rtmp}"
+                    )
+                    
+                    current_process = subprocess.Popen(cmd, shell=True, preexec_fn=os.setsid)
+                    
+                    with open(STATUS, "w") as f:
+                        json.dump({
+                            "live": True,
+                            "link": link,
+                            "title": title,
+                            "started_at": datetime.now().isoformat()
+                        }, f)
+                    
+                    print(f"✅ البث بدأ: {link}")
+                    print(f"📝 العنوان: {title}")
+                    
+                else:
+                    if current_process and current_process.poll() is not None:
+                        print("⚠️ ffmpeg توقف! إعادة التشغيل...")
+                        cmd = (
+                            f"streamlink https://www.tiktok.com/{TIKTOK_USERNAME}/live best --hls-live-edge 3 --stdout | "
+                            f"ffmpeg -thread_queue_size 8192 -i pipe:0 "
+                            f"-c:v copy "
+                            f"-c:a aac -ar 48000 -b:a 192k -ac 2 "
+                            f"-af 'highpass=f=200,lowpass=f=3000,aresample=async=1' "
+                            f"-buffer_size 1024k "
+                            f"-fflags +genpts+igndts "
+                            f"-max_interleave_delta 0 "
+                            f"-f flv {current_rtmp}"
+                        )
+                        current_process = subprocess.Popen(cmd, shell=True, preexec_fn=os.setsid)
+                        
             else:
-                #print(f"🔍 فحص: {TIKTOK_USERNAME} غير متصل حالياً.")
-                pass
-
+                print(f"[{datetime.now().strftime('%H:%M')}] 🔍 {TIKTOK_USERNAME} غير متصل")
+                
+                if current_process and current_process.poll() is None:
+                    print("⏹️ أحمد قافل البث، إغلاق...")
+                    kill_process_and_children(current_process)
+                    current_process = None
+                    last_room_id = None
+                    current_rtmp = None
+                    
+                    with open(STATUS, "w") as f:
+                        json.dump({"live": False}, f)
+                else:
+                    last_room_id = None
+                    
         except Exception as e:
             print(f"⚠️ تنبيه: {e}")
+            if current_process:
+                kill_process_and_children(current_process)
+                current_process = None
+        
+        time.sleep(60)
 
-        time.sleep(POLL_INTERVAL)
-
-# تشغيل الخيط الخلفي فور تشغيل التطبيق (مرة واحدة فقط)
+# ══════════════════════════════════════════════════════════════════════════════
+# 🚀 تشغيل المحرك + النبض
+# ══════════════════════════════════════════════════════════════════════════════
 if "background_started" not in st.session_state:
-    # البحث عن أي خيط يعمل مسبقاً لمنع التكرار
     for thread in threading.enumerate():
-        if thread.name == "TikTokMonitorThread":
+        if thread.name in ["TikTokMonitorThread", "HeartbeatThread"]:
             st.session_state.background_started = True
             break
     else:
-        # إذا لم يكن هناك خيط، نبدأ واحد جديد
         monitor_thread = threading.Thread(target=run_monitor, name="TikTokMonitorThread", daemon=True)
         monitor_thread.start()
+        
+        heartbeat_thread = threading.Thread(target=internal_heartbeat, name="HeartbeatThread", daemon=True)
+        heartbeat_thread.start()
+        
         st.session_state.background_started = True
 
 # ══════════════════════════════════════════════════════════════════════════════
-# 🎨 واجهة المستخدم (للمتابعة فقط)
+# 🎨 واجهة المستخدم
 # ══════════════════════════════════════════════════════════════════════════════
-
 st.set_page_config(page_title="AutoStreamer 24/7", page_icon="🔴")
 
-# تنسيق الشكل
 st.markdown("""
 <style>
     .status-box {
@@ -89,20 +298,48 @@ st.markdown("""
 
 st.title("🔴 نظام البث التلقائي المستمر")
 
+try:
+    with open(STATUS, "r") as f:
+        s = json.load(f)
+except:
+    s = {"live": False}
+
+status_color = "#00ff00" if s.get("live") else "#ffaa00"
+status_text = "🔴 يبث الآن" if s.get("live") else "🟡 في وضع المراقبة"
+
 st.markdown(f"""
 <div class="status-box">
-    <h3>حالة النظام: <span style="color: #00ff00;">متصل ويعمل ✅</span></h3>
-    <p>يتم مراقبة الحساب: <b>{TIKTOK_USERNAME}</b></p>
-    <p class="live-tag">النظام يعمل في خلفية السيرفر الآن</p>
+    <h3>حالة النظام: <span style="color: {status_color};">{status_text}</span></h3>
+    <p>الحساب المراقب: <b>{TIKTOK_USERNAME}</b></p>
+    <p class="live-tag">يعمل في الخلفية 24/7</p>
+    {f'<p>📝 العنوان: {s.get("title", "")}</p>' if s.get("title") else ''}
+    {f'<p>🔗 <a href="{s.get("link")}" target="_blank">رابط البث</a></p>' if s.get("link") else ''}
 </div>
 """, unsafe_allow_html=True)
 
-st.success("ℹ️ يمكنك إغلاق هذه الصفحة الآن. السيرفر سيقوم برصد البث ورفعه تلقائياً في أي وقت (اليوم، غداً، أو أي وقت).")
+if s.get("live") and s.get("link"):
+    st.success("🔴 البث مباشر الآن على يوتيوب!")
+    st.video(s["link"])
+
+st.info("💡 يمكنك إغلاق هذه الصفحة. النظام يرصد البث ويرفعه تلقائياً.")
 
-st.info("💡 بمجرد أن يبدأ صاحب الحساب البث، سيقوم النظام بفتحه فوراً ونقله إلى قناتك على يوتيوب.")
+# عرض النبض
+try:
+    if os.path.exists(HEARTBEAT_FILE):
+        with open(HEARTBEAT_FILE, "r") as f:
+            hb = json.load(f)
+        last_ping = datetime.fromisoformat(hb["last_ping"])
+        delta = (datetime.now() - last_ping).total_seconds()
+        if delta < 60:
+            st.success(f"💓 النظام يعمل (آخر نبض: {int(delta)} ثانية)")
+        else:
+            st.warning(f"⚠️ بطيء (آخر نبض: {int(delta)} ثانية)")
+except:
+    pass
 
-# عرض السجلات البسيطة
-if st.button("تحديث الحالة"):
+if st.button("🔄 تحديث الحالة"):
     st.rerun()
 
-st.write(f"آخر فحص تم في: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
+st.write(f"⏱️ آخر تحديث: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
+st.markdown("---")
+st.caption("🚀 TikTok → YouTube Auto Relay | أحمد عبدالله العراق")