Update streamlit_app.py

streamlit_app.py

@@ -1,201 +1,122 @@
-import streamlit as st
-import subprocess
-import time
+import asyncio
 import os
-from datetime import datetime
-from pathlib import Path
+import sys
+import time
+import subprocess
+from TikTokLive import TikTokLiveClient
+from TikTokLive.errors import UserOfflineError, UserNotFoundError
 import requests
 
-# Page config MUST be first
-st.set_page_config(page_title="TikTok Live Recorder", page_icon="📹", layout="wide")
+# ==================== إعدادات التشغيل ====================
+# ضع اسم المستخدم الخاص بالمذيع هنا (تأكد من كتابة الـ @)
+TARGET_USERNAME = "@username_here" 
 
-# Loading screen
-@st.cache_resource
-def load_resources():
-    return True
+# عدد الثواني للانتظار بين كل عملية تحقق من البث
+POLL_INTERVAL = 60  
 
-if not load_resources():
-    st.spinner("Loading resources...")
+# المجلد الذي سيتم حفظ الفيديوهات المسجلة بداخله
+SAVE_DIR = "./recordings"
 
-st.markdown("""
-<style>
-.main-header {font-size: 2.5rem; font-weight: bold; background: linear-gradient(90deg, #ff0050, #00f2ea); -webkit-background-clip: text; -webkit-text-fill-color: transparent;}
-.status-box {padding: 1rem; border-radius: 0.5rem; text-align: center; font-weight: bold;}
-.status-recording {background: linear-gradient(90deg, #ff6b6b, #ee5a5a); color: white; animation: pulse 1s infinite;}
-.status-ready {background: linear-gradient(90deg, #00ff88, #00cc66); color: black;}
-.status-offline {background: linear-gradient(90deg, #444444, #222222); color: #aaa;}
-@keyframes pulse {0%, 100% {opacity: 1;} 50% {opacity: 0.7;}}
-</style>
-""", unsafe_allow_html=True)
+# محرك التسجيل المفضل (يفضل streamlink لأنه الأقوى في جلب الجودة التلقائية وبثبات)
+# الخيارات المتاحة: "streamlink" أو "ffmpeg" أو "yt-dlp"
+RECORDING_ENGINE = "streamlink" 
 
-# Session state
-if 'recording' not in st.session_state:
-    st.session_state.recording = False
-if 'process' not in st.session_state:
-    st.session_state.process = None
-if 'stream_url' not in st.session_state:
-    st.session_state.stream_url = None
-if 'username' not in st.session_state:
-    st.session_state.username = None
-if 'video_path' not in st.session_state:
-    st.session_state.video_path = None
-if 'start_time' not in st.session_state:
-    st.session_state.start_time = None
-if 'viewers' not in st.session_state:
-    st.session_state.viewers = 0
-
-OUTPUT_DIR = "/tmp/tiktok_recordings"
-Path(OUTPUT_DIR).mkdir(parents=True, exist_ok=True)
-
-def check_tiktok_live(username):
-    try:
-        from TikTokLive import TikTokLiveClient
-        client = TikTokLiveClient(unique_id=f"@{username}")
-        is_live = client.fetch_is_live()
-        if not is_live:
-            return {'is_live': False, 'error': 'User not live'}
-        room_info = client.fetch_room_info()
-        stream_url = None
-        if hasattr(room_info, 'stream_url') and room_info.stream_url:
-            if hasattr(room_info.stream_url, 'hls_pull_url'):
-                stream_url = room_info.stream_url.hls_pull_url
-            elif hasattr(room_info.stream_url, 'flv_pull_url'):
-                stream_url = room_info.stream_url.flv_pull_url
-        return {'is_live': True, 'stream_url': stream_url, 'room_id': getattr(room_info, 'webcast_room_id', None), 'viewer_count': getattr(room_info, 'user_count', 0)}
-    except Exception as e:
-        return {'is_live': False, 'error': str(e)}
+# إرسال تنبيهات لهاتفك عند بدء البث (اختياري - اتركها فارغة إذا لم تكن تريدها)
+DISCORD_WEBHOOK_URL = ""
+# =======================================================
 
-def start_recording(stream_url, output_path):
-    try:
-        cmd = ['ffmpeg', '-y', '-re', '-i', stream_url, '-c:v', 'copy', '-c:a', 'copy', '-f', 'mp4', output_path]
-        return subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-    except Exception as e:
-        st.error(f"Error: {e}")
-        return None
+os.makedirs(SAVE_DIR, exist_ok=True)
 
-def stop_recording():
-    if st.session_state.process:
+def send_notification(message):
+    """إرسال إشعارات فورية إلى Discord في حال توفر الرابط"""
+    print(f"[تنبيه] {message}")
+    if DISCORD_WEBHOOK_URL:
         try:
-            st.session_state.process.terminate()
-            st.session_state.process.wait(timeout=5)
-        except:
-            st.session_state.process.kill()
-        st.session_state.process = None
-        st.session_state.recording = False
-
-def format_time(seconds):
-    mins = int(seconds // 60)
-    secs = int(seconds % 60)
-    return f"{mins}:{secs:02d}"
-
-def upload_to_telegram(file_path, bot_token, chat_id):
+            requests.post(DISCORD_WEBHOOK_URL, json={"content": message})
+        except Exception as e:
+            print(f"⚠️ فشل إرسال التنبيه: {e}")
+
+async def check_if_live(username):
+    """
+    ✅ الدالة المصلحة للتحقق من أن المذيع متصل بالبث المباشر
+    تم استبدال الدالة الخاطئة fetch_is_live بالدالة الرسمية والمضمونة is_live()
+    """
+    client = TikTokLiveClient(unique_id=username)
     try:
-        url = f"https://api.telegram.org/bot{bot_token}/sendVideo"
-        with open(file_path, 'rb') as video:
-            response = requests.post(url, files={'video': video}, data={'chat_id': chat_id, 'caption': f"📹 TikTok Live - @{st.session_state.username}"}, timeout=300)
-            if response.status_code == 200:
-                result = response.json()
-                if result.get('ok'):
-                    return {'success': True, 'file_id': result['result']['video']['file_id']}
-        return {'success': False, 'error': 'Upload failed'}
+        # الاستدعاء الصحيح والحديث في مكتبة بايثون
+        is_live = await client.is_live()
+        return is_live
+    except UserNotFoundError:
+        print(f"❌ اسم المستخدم {username} غير موجود أو تم كتابته بشكل خاطئ.")
+        return False
     except Exception as e:
-        return {'success': False, 'error': str(e)}
-
-# Main UI
-st.markdown('<h1 class="main-header">📹 TikTok Live Recorder</h1>', unsafe_allow_html=True)
-st.markdown("Record TikTok live streams with free unlimited upload to Telegram")
-
-st.markdown("### 🔗 Connect to Live Stream")
-username_input = st.text_input("TikTok Username (without @)")
-
-col1, col2, col3 = st.columns(3)
-
-with col1:
-    if st.button("🔍 Check Live", use_container_width=True):
-        if username_input:
-            with st.spinner("Checking..."):
-                result = check_tiktok_live(username_input)
-                if result.get('is_live'):
-                    st.success(f"✅ @{username_input} is LIVE!")
-                    st.session_state.stream_url = result.get('stream_url')
-                    st.session_state.username = username_input
-                    st.session_state.viewers = result.get('viewer_count', 0)
-                    st.rerun()
-                else:
-                    st.error(f"❌ Not live: {result.get('error', '')}")
-
-with col2:
-    if st.session_state.stream_url:
-        if st.session_state.recording:
-            if st.button("⏹️ Stop", type="secondary", use_container_width=True):
-                stop_recording()
-                st.success("Recording stopped!")
-                st.rerun()
-        else:
-            if st.button("🔴 Record", type="primary", use_container_width=True):
-                output_path = f"{OUTPUT_DIR}/tiktok_{st.session_state.username}_{datetime.now().strftime('%Y%m%d_%H%M%S')}.mp4"
-                st.session_state.process = start_recording(st.session_state.stream_url, output_path)
-                if st.session_state.process:
-                    st.session_state.recording = True
-                    st.session_state.start_time = time.time()
-                    st.session_state.video_path = output_path
-                    st.success("🎬 Recording started!")
-                    st.rerun()
-
-with col3:
-    if st.session_state.recording:
-        st.markdown('<div class="status-box status-recording">🔴 Recording</div>', unsafe_allow_html=True)
-    elif st.session_state.stream_url:
-        st.markdown('<div class="status-box status-ready">🟢 Ready</div>', unsafe_allow_html=True)
-    else:
-        st.markdown('<div class="status-box status-offline">⚪ Offline</div>', unsafe_allow_html=True)
-
-if st.session_state.recording:
-    st.divider()
-    duration = time.time() - st.session_state.start_time
-    col1, col2, col3 = st.columns(3)
-    col1.metric("Status", "🔴 Recording")
-    col2.metric("Duration", format_time(duration))
-    col3.metric("User", f"@{st.session_state.username}")
-
-if not st.session_state.recording and st.session_state.video_path and os.path.exists(st.session_state.video_path):
-    st.divider()
-    st.markdown("### ☁️ Upload Video")
-    file_size = os.path.getsize(st.session_state.video_path) / (1024 * 1024)
-    st.info(f"📊 File size: {file_size:.1f} MB")
+        # في حال وجود جدار حماية أو مشكلة في الاتصال من السيرفر
+        print(f"⚠️ فشل التحقق من الحالة حالياً: {e}")
+        return False
+
+def start_recording(username):
+    """بدء عملية تسجيل البث وحفظ الفيديو مباشرة"""
+    timestamp = time.strftime("%Y%m%d_%H%M%S")
+    clean_user = username.replace("@", "")
+    output_file = os.path.join(SAVE_DIR, f"{clean_user}_{timestamp}.mp4")
+    live_url = f"https://www.tiktok.com/{username}/live"
     
-    upload_method = st.radio("Choose upload method", ["Telegram (Unlimited Free)", "Internet Archive", "Download Only"], index=0)
+    send_notification(f"🔴 المذيع {username} بدأ البث المباشر الآن! جاري بدء التسجيل...")
     
-    if upload_method == "Telegram (Unlimited Free)":
-        with st.expander("⚙️ Telegram Settings", expanded=True):
-            bot_token = st.text_input("Bot Token", placeholder="123456:ABC-DEF...", type="password")
-            chat_id = st.text_input("Chat ID", placeholder="-1001234567890")
-            st.info("**Get Bot Token:** @BotFather | **Get Chat ID:** @getmyid_bot")
+    # تحديد أمر التسجيل بناءً على المحرك المختار
+    if RECORDING_ENGINE == "streamlink":
+        command = ["streamlink", live_url, "best", "-o", output_file, "--stream-segment-threads", "3"]
+    elif RECORDING_ENGINE == "ffmpeg":
+        command = ["ffmpeg", "-i", live_url, "-c", "copy", "-y", output_file]
+    else:
+        command = ["yt-dlp", "-o", output_file, live_url]
         
-        if st.button("📤 Upload to Telegram", type="primary"):
-            if bot_token and chat_id:
-                with st.spinner("Uploading..."):
-                    result = upload_to_telegram(st.session_state.video_path, bot_token, chat_id)
-                    if result.get('success'):
-                        st.success(f"✅ Uploaded! File ID: {result['file_id']}")
-                    else:
-                        st.error(f"❌ {result.get('error', 'Failed')}")
-            else:
-                st.error("Please enter Bot Token and Chat ID")
+    print(f"🚀 تشغيل أمر التسجيل: {' '.join(command)}")
+    try:
+        # تشغيل التسجيل كعملية فرعية لمتابعة تقدم التحميل في الوقت الفعلي
+        process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True)
+        
+        # طباعة سجلات التحميل في لوحة التحكم (Hugging Face Logs)
+        for line in process.stdout:
+            sys.stdout.write(line)
+            sys.stdout.flush()
+            
+        process.wait()
+        send_notification(f"⏹️ انتهى البث المباشر أو توقف التسجيل لـ {username}. تم حفظ الفيديو في: {output_file}")
+    except Exception as e:
+        print(f"⚠️ حدث خطأ أثناء التسجيل: {e}")
+        send_notification(f"⚠️ حدث خطأ أثناء تسجيل بث {username}: {e}")
+
+async def main():
+    print("=" * 50)
+    print(f"🌟 بدء تشغيل مسجل بث تيك توك التلقائي للمذيع: {TARGET_USERNAME}")
+    print(f"⏱️ فحص دوري كل: {POLL_INTERVAL} ثانية")
+    print(f"📹 مجلد حفظ الفيديوهات: {SAVE_DIR}")
+    print(f"⚙️ محرك التسجيل المعتمد: {RECORDING_ENGINE}")
+    print("=" * 50)
     
-    elif upload_method == "Internet Archive":
-        if st.button("📤 Prepare Archive.org Upload"):
-            identifier = f"tiktok-{st.session_state.username}-{int(time.time())}"
-            st.link_button("🌐 Upload to Archive.org", f"https://archive.org/services/edit/{identifier}")
+    send_notification(f"🚀 تم تشغيل سكربت المراقبة لـ {TARGET_USERNAME}")
     
-    st.divider()
-    with open(st.session_state.video_path, 'rb') as f:
-        st.download_button(label="⬇️ Download MP4", data=f, file_name=f"tiktok_{st.session_state.username}.mp4", mime="video/mp4", use_container_width=True)
-
-st.divider()
-st.markdown("**⚠️ Notes:** Recordings are temporary. User must be LIVE. Respect creators' rights.")
+    while True:
+        # الفحص الذكي باستخدام الدالة الصحيحة المعدلة
+        is_live = await check_if_live(TARGET_USERNAME)
+        
+        if is_live:
+            print(f"🔥 المذيع {TARGET_USERNAME} أونلاين! جاري استدعاء مسجل الفيديو...")
+            start_recording(TARGET_USERNAME)
+        else:
+            print(f"😴 المذيع {TARGET_USERNAME} غير متصل حالياً. إعادة الفحص بعد {POLL_INTERVAL} ثانية...")
+            
+        await asyncio.sleep(POLL_INTERVAL)
 
-with st.sidebar:
-    st.markdown("### ℹ️ Info")
-    st.write("**Status:** Online")
-    st.info("Enter Telegram credentials after recording to upload")
+if __name__ == '__main__':
+    try:
+        asyncio.run(main())
+    except KeyboardInterrupt:
+        print("👋 تم إيقاف البرنامج بواسطة المستخدم.")
+    except Exception as e:
+        print(f"💥 خطأ غير متوقع: {e}")
+        # إعادة تشغيل السكربت تلقائياً في حال حدوث انقطاع مفاجئ للإنترنت بالسيرفر
+        print("🔄 جاري إعادة تشغيل السكربت تلقائياً بعد 10 ثوانٍ...")
+        time.sleep(10)
+        os.execv(sys.executable, ['python'] + sys.argv)