Update app.py

app.py

@@ -146,8 +146,7 @@ def parse_srt_from_text(srt_text):
             start_str, end_str = times.split("-->")
             start = srt_time_to_sec(start_str)
             end = srt_time_to_sec(end_str)
-            # захоўваем пераносы
-            text = "\n".join(lines[2:]).strip()
+            text = "\n".join(lines[2:]).strip()  # захоўваем пераносы
             subs.append({"start": start, "end": end, "text": text})
         except Exception:
             continue
@@ -158,13 +157,11 @@ def parse_srt_from_text(srt_text):
 # Wrap: аўтаматычны перанос радкоў па шырыні кадра
 # ----------------------------
 def _line_width_px(draw: ImageDraw.ImageDraw, line: str, font: ImageFont.ImageFont, stroke_width: int = 0) -> int:
-    """Вяртае шырыню радка ў пікселях (з улікам stroke), максімальна сумяшчальна з Pillow."""
     try:
         l, t, r, b = draw.textbbox((0, 0), line, font=font, stroke_width=stroke_width)
         return int(r - l)
     except Exception:
         try:
-            # textlength не ўлічвае stroke, таму дадамо запас
             w = draw.textlength(line, font=font)
             return int(w + 2 * stroke_width)
         except Exception:
@@ -173,7 +170,6 @@ def _line_width_px(draw: ImageDraw.ImageDraw, line: str, font: ImageFont.ImageFo
 
 
 def _break_long_word(draw, word: str, font, max_width: int, stroke_width: int) -> list:
-    """Калі слова вельмі доўгае — разбіваем па сімвалах (greedy)."""
     chunks = []
     cur = ""
     for ch in word:
@@ -190,10 +186,6 @@ def _break_long_word(draw, word: str, font, max_width: int, stroke_width: int) -
 
 def wrap_text_to_width(text: str, draw: ImageDraw.ImageDraw, font: ImageFont.ImageFont,
                        max_width: int, stroke_width: int = 0) -> str:
-    """
-    Аўта-перанос: захоўвае існуючыя '\n' як абзацы,
-    але ўнутры кожнага радка робіць word-wrap пад max_width.
-    """
     if not text:
         return text
 
@@ -210,7 +202,6 @@ def wrap_text_to_width(text: str, draw: ImageDraw.ImageDraw, font: ImageFont.Ima
         line = ""
         for w in words:
             if not line:
-                # калі слова занадта доўгае — разбіваем
                 if _line_width_px(draw, w, font, stroke_width) <= max_width:
                     line = w
                 else:
@@ -224,7 +215,6 @@ def wrap_text_to_width(text: str, draw: ImageDraw.ImageDraw, font: ImageFont.Ima
                 line = test
             else:
                 out_lines.append(line)
-                # пачынаем новы радок
                 if _line_width_px(draw, w, font, stroke_width) <= max_width:
                     line = w
                 else:
@@ -264,6 +254,10 @@ def _measure_multiline_bbox(draw, text, font, stroke_width=0, spacing=4):
     return (0, 0, w, h)
 
 
+def _clamp(v: float, lo: float, hi: float) -> float:
+    return max(lo, min(hi, v))
+
+
 def create_animated_text_clip(
     text,
     duration,
@@ -273,14 +267,14 @@ def create_animated_text_clip(
     stroke_color,
     stroke_width,
     position_type,
-    custom_x,
+    custom_x_shift,  # ЗРУХ X (адносна цэнтра)
     custom_y,
     animation,
     video_width,
     video_height,
     bg_color=None,
     bg_opacity=1.0,
-    wrap_ratio: float = 0.90,  # 90% ад шырыні кадра
+    wrap_ratio: float = 0.90,
 ):
     try:
         txt_rgb = hex_to_rgb(color)
@@ -313,7 +307,6 @@ def create_animated_text_clip(
     spacing = max(0, int(fontsize * 0.15))
 
     # --- AUTO WRAP ---
-    # Макс. шырыня тэксту = wrap_ratio * video_width (але мінімум 200px)
     max_text_width = max(200, int(video_width * float(wrap_ratio)))
     wrapped_text = wrap_text_to_width(text, dummy_draw, pil_font, max_text_width, stroke_width=stroke_width)
 
@@ -332,7 +325,7 @@ def create_animated_text_clip(
     img = Image.new("RGBA", (img_w, img_h), bg_rgb + (bg_a,))
     draw = ImageDraw.Draw(img)
 
-    # галоўнае: улічваем bbox (можа быць адмоўны) -> не рэжа
+    # галоўнае: улічваем bbox (можа быць адмоўны)
     text_x = (pad_x + safe) - l
     text_y = (pad_y + safe) - t
 
@@ -355,19 +348,28 @@ def create_animated_text_clip(
     mask = mpe.ImageClip(alpha, ismask=True).set_duration(duration)
     clip = clip.set_mask(mask)
 
+    # --- Пазіцыя (аптымізавана) ---
+    # верх/ніз ссоўваем да цэнтра на 15% вышыні кадра
+    center_pull = 0.15 * float(video_height)
+    base_x_center = (video_width - clip.w) / 2
+
     if position_type == "bottom":
-        x = (video_width - clip.w) / 2
-        y = video_height - clip.h - 20
+        x = base_x_center
+        y = (video_height - clip.h - 20) - center_pull
     elif position_type == "top":
-        x = (video_width - clip.w) / 2
-        y = 20
+        x = base_x_center
+        y = 20 + center_pull
     elif position_type == "center":
-        x = (video_width - clip.w) / 2
+        x = base_x_center
         y = (video_height - clip.h) / 2
-    else:
-        x = float(custom_x)
+    else:  # custom: Х заўсёды ад цэнтра (як ва ўсіх), custom_x_shift — дадатковы зрух
+        x = base_x_center + float(custom_x_shift)
         y = float(custom_y)
 
+    # clamp, каб не вылятала за кадр
+    x = _clamp(x, 0, max(0, video_width - clip.w))
+    y = _clamp(y, 0, max(0, video_height - clip.h))
+
     anim = (animation or "").lower()
 
     if anim == "fade":
@@ -401,7 +403,7 @@ def apply_subtitles(
     stroke_color,
     stroke_width,
     position_type,
-    custom_x,
+    custom_x_shift,
     custom_y,
     animation,
     export_quality,
@@ -427,7 +429,7 @@ def apply_subtitles(
             stroke_color,
             stroke_width,
             position_type,
-            custom_x,
+            custom_x_shift,
             custom_y,
             animation,
             w,
@@ -476,7 +478,7 @@ def create_single_frame_video(
     stroke_color,
     stroke_width,
     position_type,
-    custom_x,
+    custom_x_shift,
     custom_y,
     animation,
     bg_color=None,
@@ -498,7 +500,7 @@ def create_single_frame_video(
         stroke_color,
         stroke_width,
         position_type,
-        custom_x,
+        custom_x_shift,
         custom_y,
         animation,
         w,
@@ -540,28 +542,81 @@ CSS = f"""
 
 
 # ----------------------------
-# Gradio UI
+# Gradio UI (аптымізаваны layout)
 # ----------------------------
 with gr.Blocks(css=CSS) as demo:
     gr.Markdown("# Аўтаматычнае стварэнне і накладанне субтытраў")
 
-    # адзін плэер: загрузка/передпрагляд/вынік
-    video_input = gr.Video(
-        label="Загрузіце відэа (тут жа будзе перадпрагляд і вынік)",
-        elem_id="main_video",
-    )
-
+    # Стан: арыгінальны файл (каб не губляць пры preview/output)
     uploaded_video = gr.State(None)
     first_frame = gr.State(None)
 
-    with gr.Row():
-        gen_btn = gr.Button("Стварыць субтытры")
-        subs_box = gr.Textbox(label="Тэкст субтытраў", lines=15)
-
+    with gr.Row(equal_height=True):
+        with gr.Column(scale=6):
+            video_input = gr.Video(
+                label="Відэа (тут жа будзе перадпрагляд і вынік)",
+                elem_id="main_video",
+            )
+            with gr.Row():
+                show_original_btn = gr.Button("Паказаць арыгінал")
+                # (кнопка дадатковая, але карысная пасля preview)
+
+        with gr.Column(scale=5):
+            # 1) Стварэнне субтытраў — разварочваецца/згортваецца
+            with gr.Accordion("Стварэнне субтытраў", open=False):
+                gen_btn = gr.Button("Стварыць субтытры")
+                subs_box = gr.Textbox(label="Тэкст субтытраў", lines=14)
+
+            # 2) Стыль
+            with gr.Accordion("Стыль субтытраў", open=True):
+                with gr.Row():
+                    font_dd = gr.Dropdown(list(AVAILABLE_FONTS.keys()), value="DejaVuSans-Bold", label="Шрыфт")
+                    size_sl = gr.Slider(10, 100, 1, value=40, label="Памер")
+
+                with gr.Row():
+                    color_txt = gr.ColorPicker(value="#FFFFFF", label="Колер тэксту")
+                    color_st = gr.ColorPicker(value="#000000", label="Колер абводкі")
+                    width_sl = gr.Slider(0, 10, 1, value=2, label="Абводка")
+
+                with gr.Row():
+                    bg_color = gr.ColorPicker(value="#000000", label="Фон")
+                    bg_opac = gr.Slider(0.0, 1.0, 0.1, value=0.5, label="Празрыстасць")
+
+                with gr.Row():
+                    pos_dd = gr.Dropdown(["bottom", "top", "center", "custom"], value="bottom", label="Пазіцыя")
+                    anim_dd = gr.Dropdown(["None", "fade", "slide", "zoom"], value="None", label="Анімацыя")
+
+                # Custom controls (паказваем толькі калі патрэбна)
+                with gr.Row():
+                    x_shift_sl = gr.Slider(-960, 960, 1, value=0, label="X-зрух ад цэнтра", visible=False)
+                    y_sl = gr.Slider(0, 1080, 1, value=100, label="Y (для custom)", visible=False)
+
+                with gr.Row():
+                    wrap_en = gr.Checkbox(value=True, label="Аўта-перанос радкоў")
+                    wrap_pct = gr.Slider(50, 100, 1, value=90, label="Макс. шырыня радка (%)")
+
+            # 3) Экспарт
+            with gr.Accordion("Экспарт", open=False):
+                exp_dd = gr.Dropdown(
+                    ["мінімальнае", "арыгінальнае", "сярэдняе", "максімальнае"],
+                    value="арыгінальнае",
+                    label="Якасць",
+                )
+                apply_btn = gr.Button("Накласці субтытры (фінальны вынік)")
+
+    # --- Лагічныя апрацоўшчыкі ---
+
+    # Калі значэнне video_input змянілася:
+    # - калі гэта preview/output -> НЕ пераціраем uploaded_video
+    # - калі гэта арыгінал (загрузка) -> абнаўляем uploaded_video і first_frame
     def _on_video_change(v, current_uploaded, current_frame):
         if not v:
             return None, None
 
+        # калі проста вярнуліся да арыгінала праз кнопку і ён ужо вядомы — не пералічваем кадр
+        if current_uploaded and str(v) == str(current_uploaded):
+            return current_uploaded, current_frame
+
         name = os.path.basename(str(v))
         if name in ("preview_video.mp4", "output_video.mp4"):
             return current_uploaded, current_frame
@@ -575,6 +630,13 @@ with gr.Blocks(css=CSS) as demo:
         outputs=[uploaded_video, first_frame],
     )
 
+    # Паказаць арыгінал у тым жа плэеры
+    def _show_original(v_orig):
+        return v_orig
+
+    show_original_btn.click(_show_original, inputs=uploaded_video, outputs=video_input)
+
+    # Генерацыя субтытраў заўсёды па арыгінальным uploaded_video
     def _on_generate_click(v_orig):
         if not v_orig:
             return "Не выбрана відэа"
@@ -583,41 +645,15 @@ with gr.Blocks(css=CSS) as demo:
 
     gen_btn.click(_on_generate_click, inputs=uploaded_video, outputs=subs_box)
 
-    gr.Markdown("## Наладка стылю субтытраў")
-
-    with gr.Row():
-        font_dd = gr.Dropdown(list(AVAILABLE_FONTS.keys()), value="DejaVuSans-Bold", label="Шрыфт")
-        size_sl = gr.Slider(10, 100, 1, value=40, label="Памер шрыфта")
-
-    with gr.Row():
-        color_txt = gr.ColorPicker(value="#FFFFFF", label="Колер тэксту")
-        color_st = gr.ColorPicker(value="#000000", label="Колер абводкі")
-        width_sl = gr.Slider(0, 10, 1, value=2, label="Таўшчыня абводкі")
-
-    with gr.Row():
-        pos_dd = gr.Dropdown(["bottom", "top", "center", "custom"], value="bottom", label="Пазіцыя")
-        anim_dd = gr.Dropdown(["None", "fade", "slide", "zoom"], value="None", label="Анімацыя")
-
-    with gr.Row():
-        x_sl = gr.Slider(0, 1920, 1, value=100, label="X (для custom)")
-        y_sl = gr.Slider(0, 1080, 1, value=100, label="Y (для custom)")
+    # Пераключэнне бачнасці custom-слайдэраў
+    def _toggle_custom(pos):
+        vis = (pos == "custom")
+        return gr.update(visible=vis), gr.update(visible=vis)
 
-    with gr.Row():
-        bg_color = gr.ColorPicker(value="#000000", label="Колер фону")
-        bg_opac = gr.Slider(0.0, 1.0, 0.1, value=0.5, label="Празрыстасць фону")
+    pos_dd.change(_toggle_custom, inputs=pos_dd, outputs=[x_shift_sl, y_sl])
 
-    # ДАДАЛІ: кіраванне wrap
-    with gr.Row():
-        wrap_en = gr.Checkbox(value=True, label="Аўта-перанос радкоў")
-        wrap_pct = gr.Slider(50, 100, 1, value=90, label="Макс. шырыня радка (%)")
-
-    exp_dd = gr.Dropdown(
-        ["мінімальнае", "арыгінальнае", "сярэдняе", "максімальнае"],
-        value="арыгінальнае",
-        label="Якасць",
-    )
-
-    def update_preview(frame, text, font, size, col, colst, wd, pos, x, y, anim, bgc, bgo, do_wrap, w_pct):
+    # Перадпрагляд: 1 секунда на першым кадры, у тым жа video_input
+    def update_preview(frame, text, font, size, col, colst, wd, pos, x_shift, y, anim, bgc, bgo, do_wrap, w_pct):
         if frame is None:
             return None
 
@@ -636,7 +672,7 @@ with gr.Blocks(css=CSS) as demo:
             colst,
             wd,
             pos,
-            x,
+            x_shift,
             y,
             anim,
             bg_color=bgc,
@@ -644,30 +680,29 @@ with gr.Blocks(css=CSS) as demo:
             wrap_ratio=wrap_ratio,
         )
 
-    comps = [font_dd, size_sl, color_txt, color_st, width_sl, pos_dd, anim_dd, subs_box, x_sl, y_sl, bg_color, bg_opac, wrap_en, wrap_pct]
-    for c in comps:
-        c.change(
-            update_preview,
-            inputs=[
-                first_frame,
-                subs_box,
-                font_dd,
-                size_sl,
-                color_txt,
-                color_st,
-                width_sl,
-                pos_dd,
-                x_sl,
-                y_sl,
-                anim_dd,
-                bg_color,
-                bg_opac,
-                wrap_en,
-                wrap_pct,
-            ],
-            outputs=video_input,
-        )
-
+    preview_inputs = [
+        first_frame,
+        subs_box,
+        font_dd,
+        size_sl,
+        color_txt,
+        color_st,
+        width_sl,
+        pos_dd,
+        x_shift_sl,
+        y_sl,
+        anim_dd,
+        bg_color,
+        bg_opac,
+        wrap_en,
+        wrap_pct,
+    ]
+
+    # Любая змена ў стылі/тэксце абнаўляе перадпрагляд
+    for c in [subs_box, font_dd, size_sl, color_txt, color_st, width_sl, pos_dd, x_shift_sl, y_sl, anim_dd, bg_color, bg_opac, wrap_en, wrap_pct]:
+        c.change(update_preview, inputs=preview_inputs, outputs=video_input)
+
+    # Фінальны рэндэр у тым жа плэеры, але па арыгінальным uploaded_video
     def _on_apply(
         v_orig,
         t,
@@ -677,7 +712,7 @@ with gr.Blocks(css=CSS) as demo:
         cs,
         wd,
         p,
-        x,
+        x_shift,
         y,
         a,
         eq,
@@ -699,7 +734,7 @@ with gr.Blocks(css=CSS) as demo:
             cs,
             wd,
             p,
-            x,
+            x_shift,
             y,
             a,
             eq,
@@ -708,7 +743,6 @@ with gr.Blocks(css=CSS) as demo:
             wrap_ratio=wrap_ratio,
         )
 
-    apply_btn = gr.Button("Накласці субтытры (фінальны вынік)")
     apply_btn.click(
         _on_apply,
         inputs=[
@@ -720,7 +754,7 @@ with gr.Blocks(css=CSS) as demo:
             color_st,
             width_sl,
             pos_dd,
-            x_sl,
+            x_shift_sl,
             y_sl,
             anim_dd,
             exp_dd,