Update streamlit_app.py

streamlit_app.py

@@ -63,19 +63,20 @@ def get_tiktok_stream_url():
         pass
     return None
 
-def analyze_stream(stream_url):
-    """تحليل التدفق باستخدام ffprobe واستخراج معلومات الترميز"""
-    log("🔍 جاري تحليل التدفق من TikTok...")
+def analyze_stream_deep(stream_url):
+    """تحليل عميق للتدفق مع وقت أطول وخيارات إضافية"""
+    log("🔍 جاري تحليل عميق للتدفق...")
     cmd = [
         "ffprobe",
         "-v", "quiet",
         "-print_format", "json",
         "-show_streams",
-        "-analyzeduration", "10000000",  # تحليل 10 ثوانٍ
+        "-analyzeduration", "30000000",  # 30 ثانية تحليل
+        "-probesize", "20000000",
         stream_url
     ]
     try:
-        result = subprocess.run(cmd, capture_output=True, text=True, timeout=20)
+        result = subprocess.run(cmd, capture_output=True, text=True, timeout=45)
         if result.returncode == 0:
             data = json.loads(result.stdout)
             with open(PROBE_LOG, "w") as f:
@@ -90,24 +91,64 @@ def analyze_stream(stream_url):
                     audio_info = s
             return video_info, audio_info
     except Exception as e:
-        log(f"⚠️ فشل التحليل: {e}")
+        log(f"⚠️ فشل التحليل العميق: {e}")
     return None, None
 
 def build_ffmpeg_command(stream_key, stream_url, video_info, audio_info):
-    """بناء أمر ffmpeg بناءً على نتائج التحليل"""
-    if not video_info:
-        log("❌ لم يتم العثور على تدفق فيديو!")
-        return None
-
-    codec_v = video_info.get("codec_name", "unknown")
-    codec_a = audio_info.get("codec_name", "unknown") if audio_info else "none"
-    sample_rate = audio_info.get("sample_rate", "44100") if audio_info else "44100"
-    bitrate_a = int(audio_info.get("bit_rate", 0)) // 1000 if audio_info else 0
-
-    log(f"📊 فيديو: {codec_v} | صوت: {codec_a} @ {bitrate_a}kbps")
-
+    """بناء الأمر بناءً على التحليل العميق أو خطة احتياطية قوية"""
     rtmp_url = f"rtmp://a.rtmp.youtube.com/live2/{stream_key}"
 
+    # ==================== الصوت ====================
+    if audio_info:
+        codec_a = audio_info.get("codec_name", "unknown")
+        # محاولة الحصول على معدل بت حقيقي، إذا كان 0 نعتمد خطة احتياطية
+        bitrate_a = int(audio_info.get("bit_rate", 0)) // 1000 if audio_info.get("bit_rate") else 0
+        log(f"📊 الصوت: {codec_a} | معدل البت: {bitrate_a}kbps")
+        if codec_a == "aac" and bitrate_a >= 128:
+            audio_opts = ["-c:a", "copy"]
+            log("🔊 نسخ الصوت مباشرة (جودة كافية)")
+        else:
+            audio_opts = [
+                "-c:a", "aac",
+                "-b:a", "192k",
+                "-ac", "2",
+                "-ar", "44100",
+                "-async", "1",
+                "-af", "alimiter=limit=0.95:attack=5:release=50,loudnorm=I=-16:TP=-1.5:LRA=11"
+            ]
+            log("🔊 إعادة ترميز الصوت مع فلتر تطبيع واحترافي")
+    else:
+        # لا توجد معلومات صوت، خطة احتياطية
+        audio_opts = [
+            "-c:a", "aac",
+            "-b:a", "192k",
+            "-ac", "2",
+            "-ar", "44100",
+            "-async", "1",
+            "-af", "alimiter=limit=0.95:attack=5:release=50,loudnorm=I=-16:TP=-1.5:LRA=11"
+        ]
+        log("🔊 لم يتم تحليل الصوت – استخدام خطة احتياطية قوية")
+
+    # ==================== الفيديو ====================
+    # نستخدم دائمًا إعدادات الفيديو المثبتة مع فلتر الأبعاد
+    video_opts = [
+        "-vf", "scale=1080:1920:force_original_aspect_ratio=increase,pad=1080:1920:(ow-iw)/2:(oh-ih)/2:black:eval=frame",
+        "-c:v", "libx264",
+        "-preset", "ultrafast",
+        "-crf", "18",
+        "-maxrate", "8000k",
+        "-minrate", "5000k",
+        "-bufsize", "16000k",
+        "-g", "60",
+        "-vsync", "2",
+        "-x264-params", "nal-hrd=cbr",
+        "-pix_fmt", "yuv420p",
+        "-color_range", "tv",
+        "-color_primaries", "bt709",
+        "-color_trc", "bt709",
+        "-colorspace", "bt709",
+    ]
+
     cmd = [
         "nice", "-n", "19", "ffmpeg",
         "-hide_banner", "-loglevel", "verbose",
@@ -119,85 +160,26 @@ def build_ffmpeg_command(stream_key, stream_url, video_info, audio_info):
         "-i", stream_url,
         "-err_detect", "ignore_err",
     ]
-
-    # فيديو
-    if codec_v in ["h264", "hevc"]:
-        # نسخ مباشر إن أمكن، لكننا نحتاج الفلتر للأبعاد، فلا يمكن النسخ الكامل
-        # لذا سنستخدم libx264 دائمًا مع ضبط الجودة
-        video_opts = [
-            "-vf", "scale=1080:1920:force_original_aspect_ratio=increase,pad=1080:1920:(ow-iw)/2:(oh-ih)/2:black:eval=frame",
-            "-c:v", "libx264",
-            "-preset", "ultrafast",
-            "-crf", "18",
-            "-maxrate", "8000k",
-            "-minrate", "5000k",
-            "-bufsize", "16000k",
-            "-g", "60",
-            "-vsync", "2",
-            "-x264-params", "nal-hrd=cbr",
-            "-pix_fmt", "yuv420p",
-            "-color_range", "tv",
-            "-color_primaries", "bt709",
-            "-color_trc", "bt709",
-            "-colorspace", "bt709",
-        ]
-    else:
-        # ترميز غير متوقع، نستخدم نفس الإعدادات الاحتياطية
-        video_opts = [
-            "-c:v", "libx264",
-            "-preset", "ultrafast",
-            "-crf", "18",
-            "-maxrate", "8000k",
-            "-minrate", "5000k",
-            "-bufsize", "16000k",
-            "-g", "60",
-            "-vsync", "2",
-        ]
     cmd.extend(video_opts)
-
-    # صوت
-    if codec_a in ["aac"] and bitrate_a <= 192:
-        # نسخ الصوت مباشرة
-        audio_opts = ["-c:a", "copy"]
-        log("🔊 نسخ الصوت مباشرة (جودة مناسبة)")
-    else:
-        # إعادة ترميز الصوت مع فلتر مناسب
-        audio_opts = [
-            "-c:a", "aac",
-            "-b:a", "192k",
-            "-ac", "2",
-            "-ar", str(sample_rate),
-            "-async", "1",
-        ]
-        # إذا كان الصوت به ذروات عالية (نفترض ذلك إذا كان معدل البت > 192)
-        if bitrate_a > 192:
-            audio_opts.extend(["-af", "alimiter=limit=0.95:attack=5:release=50"])
-            log("🔊 إضافة فلتر الصوت للذروات العالية")
     cmd.extend(audio_opts)
-
-    # إرسال
     cmd.extend([
         "-max_muxing_queue_size", "65535",
         "-max_delay", "5000000",
         "-flvflags", "no_duration_filesize+add_keyframe_index",
         "-f", "flv", rtmp_url
     ])
-
     return cmd
 
 def start_stream_adaptive(stream_key, stream_url):
     if not stream_key or not stream_url:
         return None
 
-    # تحليل التدفق أولاً
-    video_info, audio_info = analyze_stream(stream_url)
-
-    # بناء الأمر
+    video_info, audio_info = analyze_stream_deep(stream_url)
     final_cmd = build_ffmpeg_command(stream_key, stream_url, video_info, audio_info)
     if not final_cmd:
         return None
 
-    log("🎬 بدء البث الذكي...")
+    log("🎬 بدء البث الذكي المُحسَّن...")
     with open(FFMPEG_LOG, "w") as stderr_file:
         proc = subprocess.Popen(final_cmd, stdout=subprocess.DEVNULL,
                                 stderr=stderr_file, preexec_fn=os.setsid)
@@ -222,7 +204,7 @@ def run_monitor():
     was_streaming = False
     consecutive_fails = 0
 
-    log("🚀 محرك البث الذكي (تحليل تلقائي)")
+    log("🚀 محرك البث الذكي المُحسَّن (تحليل عميق)")
 
     while True:
         try:
@@ -263,7 +245,7 @@ def run_monitor():
             time.sleep(10)
 
 if __name__ == "__main__":
-    log("✅ بدء تشغيل المحرك الذكي")
+    log("✅ بدء تشغيل المحرك الذكي المُحسَّن")
     t = threading.Thread(target=run_monitor, daemon=True)
     t.start()
     while True: