Update app.py

app.py

@@ -7,27 +7,31 @@ from moviepy import VideoFileClip, TextClip, CompositeVideoClip
 from arabic_reshaper import reshape
 
 # --- الإعدادات ---
-FONT_PATH = "arialbd.ttf" # تأكد من رفع هذا الملف في مجلد المشروع
+FONT_PATH = "arialbd.ttf"
 model = WhisperModel("large-v3", device="cpu", compute_type="int8")
 
 def process_arabic_text(text):
     return reshape(text) + "\n "
 
-def step_1_extract_words(video_path):
+# إضافة gr.Progress لتتبع التقدم في الواجهة
+def step_1_extract_words(video_path, progress=gr.Progress()):
     if not video_path:
         return None, "الرجاء رفع فيديو أولاً."
     
-    print("جاري استخراج الكلمات...")
+    progress(0, desc="جاري تحميل النموذج...")
     segments, _ = model.transcribe(video_path, word_timestamps=True, language="ar")
+    
     words_data = []
+    progress(0.5, desc="جاري تحليل الصوت واستخراج الكلمات...")
+    
     for segment in segments:
         for word in segment.words:
             words_data.append([word.word.strip(), round(word.start, 2), round(word.end, 2)])
     
     df = pd.DataFrame(words_data, columns=["الكلمة", "البداية", "النهاية"])
-    return df, "تم الاستخراج بنجاح! راجع الجدول ثم اضغط إنتاج."
+    return df, "تم الاستخراج بنجاح!"
 
-def step_2_render_video(video_path, df_edited):
+def step_2_render_video(video_path, df_edited, progress=gr.Progress()):
     if video_path is None or df_edited is None or df_edited.empty:
         return None, "بيانات ناقصة."
 
@@ -36,11 +40,10 @@ def step_2_render_video(video_path, df_edited):
     w, h = int(video.w), int(video.h)
     clips = [video]
     words_list = df_edited.values.tolist()
-
     chunk_size = 3 
 
-    print(f"جاري معالجة الفيديو...")
-
+    progress(0.1, desc="جاري تحضير النصوص...")
+    
     for i in range(0, len(words_list), chunk_size):
         current_chunk = words_list[i : i + chunk_size]
         sentence = " ".join([str(r[0]) for r in current_chunk])
@@ -50,10 +53,9 @@ def step_2_render_video(video_path, df_edited):
         c_end = float(current_chunk[-1][2])
         duration = max(0.1, c_end - c_start)
 
-        # إنشاء نص الكابشن ثابت بدون أي أنيميشن
         txt_clip = TextClip(
             text=clean_sentence,
-            font_size=80, 
+            font_size=65, 
             color='yellow',
             stroke_color='black',
             stroke_width=2,
@@ -61,18 +63,27 @@ def step_2_render_video(video_path, df_edited):
             font=FONT_PATH,
             size=(int(w * 0.9), None),
             text_align='center'
-        ).with_start(c_start).with_duration(duration).with_position(('center', int(h * 0.75)))
+        ).with_start(c_start).with_duration(duration).with_position(('center', int(h * 0.65)))
         
         clips.append(txt_clip)
 
     final_video = CompositeVideoClip(clips, size=(w, h))
-    final_video.write_videofile(output_path, codec="libx264", audio_codec="aac", fps=video.fps, logger=None)
+    
+    # هنا أعدنا الـ logger لكي تظهر النسبة المئوية % في الـ Log
+    print("بدء عملية دمج الفيديو والترميز...")
+    final_video.write_videofile(
+        output_path, 
+        codec="libx264", 
+        audio_codec="aac", 
+        fps=video.fps, 
+        logger='bar' # 'bar' هي التي تظهر النسبة المئوية % في الـ Logs
+    )
     
     return output_path, "تم إنتاج الفيديو بنجاح!"
 
 # --- بناء الواجهة ---
-with gr.Blocks(title="Caption Pro Clean") as app:
-    gr.Markdown("# 🎬 Caption Pro - Simple Style")
+with gr.Blocks(title="Caption Pro Progress") as app:
+    gr.Markdown("# 🎬 Caption Pro - With Progress Bar")
     
     with gr.Row():
         video_in = gr.Video(label="فيديو المدخلات")
@@ -84,6 +95,7 @@ with gr.Blocks(title="Caption Pro Clean") as app:
     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])