Update app.py

app.py

@@ -7,13 +7,11 @@ from moviepy import VideoFileClip, TextClip, CompositeVideoClip
 from arabic_reshaper import reshape
 
 # --- الإعدادات ---
-FONT_PATH = "arialbd.ttf"
 model = WhisperModel("large-v3", device="cpu", compute_type="int8")
 
 def process_arabic_text(text):
     return reshape(text) + "\n "
 
-# إضافة gr.Progress لتتبع التقدم في الواجهة
 def step_1_extract_words(video_path, progress=gr.Progress()):
     if not video_path:
         return None, "الرجاء رفع فيديو أولاً."
@@ -22,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:
@@ -31,18 +29,19 @@ 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, progress=gr.Progress()):
+# إضافة خيارات اللون والخط كمدخلات للدالة
+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_clean_video.mp4"
+    output_path = "final_captioned_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,12 +54,12 @@ def step_2_render_video(video_path, df_edited, progress=gr.Progress()):
 
         txt_clip = TextClip(
             text=clean_sentence,
-            font_size=65, 
-            color='yellow',
+            font_size=font_size, 
+            color=text_color,  # اللون المختار من الواجهة
             stroke_color='black',
             stroke_width=2,
             method='caption',
-            font=FONT_PATH,
+            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)))
@@ -69,35 +68,48 @@ def step_2_render_video(video_path, df_edited, progress=gr.Progress()):
 
     final_video = CompositeVideoClip(clips, size=(w, h))
     
-    # هنا أعدنا الـ logger لكي تظهر النسبة المئوية % في الـ Log
-    print("بدء عملية دمج الفيديو والترميز...")
+    print(f"بدء الدمج بلون {text_color} وخط {font_selection}...")
     final_video.write_videofile(
         output_path, 
         codec="libx264", 
         audio_codec="aac", 
         fps=video.fps, 
-        logger='bar' # 'bar' هي التي تظهر النسبة المئوية % في الـ Logs
+        logger='bar'
     )
     
     return output_path, "تم إنتاج الفيديو بنجاح!"
 
 # --- بناء الواجهة ---
-with gr.Blocks(title="Caption Pro Progress") as app:
-    gr.Markdown("# 🎬 Caption Pro - With Progress Bar")
+with gr.Blocks(title="Caption Pro Custom") as app:
+    gr.Markdown("# 🎬 محرر الكابشن: تخصيص الألوان والخطوط")
     
     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="اختر الخط (تأكد من رفع الملف)"
+        )
+        color_picker = gr.ColorPicker(value="#FFFF00", label="اختر لون النص") # القيمة الافتراضية أصفر
+        size_slider = gr.Slider(minimum=30, maximum=120, value=70, step=5, label="حجم الخط")
+
     status = gr.Textbox(label="الحالة", interactive=False)
     table = gr.Dataframe(headers=["الكلمة", "البداية", "النهاية"], datatype=["str", "number", "number"], interactive=True)
     
     btn_ex = gr.Button("1. استخراج الكلمات", variant="primary")
     btn_re = gr.Button("2. إنتاج الفيديو", variant="secondary")
 
-    # نلاحظ إضافة progress=True في الربط
+    # ربط المدخلات الجديدة بالدالة
     btn_ex.click(step_1_extract_words, inputs=[video_in], outputs=[table, status])
-    btn_re.click(step_2_render_video, inputs=[video_in, table], outputs=[video_out, status])
+    btn_re.click(
+        step_2_render_video, 
+        inputs=[video_in, table, font_dropdown, color_picker, size_slider], 
+        outputs=[video_out, status]
+    )
 
 if __name__ == "__main__":
-    app.launch()
+    app.launch()