Update app.py

app.py

@@ -20,7 +20,7 @@ def step_1_extract_words(video_path, progress=gr.Progress()):
     segments, _ = model.transcribe(video_path, word_timestamps=True, language="ar")
     
     words_data = []
-    progress(0.5, desc="جاري تحليل الصوت...")
+    progress(0.5, desc="جاري تحليل الصوت واستخراج الكلمات...")
     
     for segment in segments:
         for word in segment.words:
@@ -29,19 +29,25 @@ def step_1_extract_words(video_path, progress=gr.Progress()):
     df = pd.DataFrame(words_data, columns=["الكلمة", "البداية", "النهاية"])
     return df, "تم الاستخراج بنجاح!"
 
-# إضافة خيارات اللون والخط كمدخلات للدالة
 def step_2_render_video(video_path, df_edited, font_selection, text_color, font_size, progress=gr.Progress()):
     if video_path is None or df_edited is None or df_edited.empty:
         return None, "بيانات ناقصة."
 
-    output_path = "final_captioned_video.mp4"
+    # حل مشكلة الألوان: التأكد أن اللون بصيغة نصية واضحة
+    if text_color.startswith('rgba'):
+        # تحويل بسيط لضمان قبول اللون
+        final_color = "yellow" 
+    else:
+        final_color = text_color
+
+    output_path = "final_clean_video.mp4"
     video = VideoFileClip(video_path)
     w, h = int(video.w), int(video.h)
     clips = [video]
     words_list = df_edited.values.tolist()
     chunk_size = 3 
 
-    progress(0.1, desc="جاري إنتاج الفيديو بالخيارات المختارة...")
+    progress(0.1, desc="جاري إنتاج الفيديو...")
     
     for i in range(0, len(words_list), chunk_size):
         current_chunk = words_list[i : i + chunk_size]
@@ -55,11 +61,11 @@ def step_2_render_video(video_path, df_edited, font_selection, text_color, font_
         txt_clip = TextClip(
             text=clean_sentence,
             font_size=font_size, 
-            color=text_color,  # اللون المختار من الواجهة
+            color=final_color, 
             stroke_color='black',
             stroke_width=2,
             method='caption',
-            font=font_selection, # الخط المختار من الواجهة
+            font=font_selection,
             size=(int(w * 0.9), None),
             text_align='center'
         ).with_start(c_start).with_duration(duration).with_position(('center', int(h * 0.65)))
@@ -68,7 +74,6 @@ def step_2_render_video(video_path, df_edited, font_selection, text_color, font_
 
     final_video = CompositeVideoClip(clips, size=(w, h))
     
-    print(f"بدء الدمج بلون {text_color} وخط {font_selection}...")
     final_video.write_videofile(
         output_path, 
         codec="libx264", 
@@ -80,21 +85,22 @@ def step_2_render_video(video_path, df_edited, font_selection, text_color, font_
     return output_path, "تم إنتاج الفيديو بنجاح!"
 
 # --- بناء الواجهة ---
-with gr.Blocks(title="Caption Pro Custom") as app:
-    gr.Markdown("# 🎬 محرر الكابشن: تخصيص الألوان والخطوط")
+with gr.Blocks(title="Caption Pro Final") as app:
+    gr.Markdown("# 🎬 Caption Pro - Stable Version")
     
     with gr.Row():
         video_in = gr.Video(label="فيديو المدخلات")
         video_out = gr.Video(label="الفيديو الناتج")
     
     with gr.Row():
-        # خيارات التخصيص
+        # استخدام القائمة المنسدلة للخطوط
         font_dropdown = gr.Dropdown(
             choices=["arialbd.ttf", "Cairo-Bold.ttf", "Almarai-Bold.ttf", "Tajawal-Bold.ttf"], 
             value="arialbd.ttf", 
-            label="اختر الخط (تأكد من رفع الملف)"
+            label="اختر الخط (تأكد من وجود الملف)"
         )
-        color_picker = gr.ColorPicker(value="#FFFF00", label="اختر لون النص") # القيمة الافتراضية أصفر
+        # تحديد لون النص عبر الكلمات أو Hex لضمان الاستقرار
+        color_input = gr.ColorPicker(value="#FFFF00", label="اختر لون النص")
         size_slider = gr.Slider(minimum=30, maximum=120, value=70, step=5, label="حجم الخط")
 
     status = gr.Textbox(label="الحالة", interactive=False)
@@ -103,11 +109,10 @@ with gr.Blocks(title="Caption Pro Custom") as app:
     btn_ex = gr.Button("1. استخراج الكلمات", variant="primary")
     btn_re = gr.Button("2. إنتاج الفيديو", variant="secondary")
 
-    # ربط المدخلات الجديدة بالدالة
     btn_ex.click(step_1_extract_words, inputs=[video_in], outputs=[table, status])
     btn_re.click(
         step_2_render_video, 
-        inputs=[video_in, table, font_dropdown, color_picker, size_slider], 
+        inputs=[video_in, table, font_dropdown, color_input, size_slider], 
         outputs=[video_out, status]
     )