Update app.py

app.py

@@ -2,10 +2,14 @@ import os
 import shutil
 import subprocess
 import uuid
+import json
 from datetime import timedelta
-from fastapi import FastAPI, UploadFile, File, Form
+from fastapi import FastAPI, UploadFile, File, Form, Body
 from fastapi.responses import HTMLResponse, FileResponse, JSONResponse
+from fastapi.staticfiles import StaticFiles
 from faster_whisper import WhisperModel
+from pydantic import BaseModel
+from typing import List
 
 app = FastAPI()
 
@@ -18,28 +22,35 @@ print("Loading Whisper AI...")
 model = WhisperModel("small", device="cpu", compute_type="int8")
 print("Model Loaded!")
 
-# --- تعریف استایل‌ها و فونت‌ها ---
+# --- مدل‌های داده (برای دریافت اطلاعات از سمت کاربر) ---
+class SegmentData(BaseModel):
+    start: float
+    end: float
+    text: str
 
+class BurnRequest(BaseModel):
+    file_id: str
+    style: str
+    font: str
+    segments: List[SegmentData]
+
+# --- تعریف استایل‌ها ---
 STYLES_CONFIG = {
     "hormozi": {
         "name": "هرمزی (زرد)",
-        # زرد جیغ پشت متن، متن مشکی
-        "ass_style": "BackColour=&H0000FFFF,PrimaryColour=&H00000000,BorderStyle=3,Outline=0,Shadow=0"
+        "ass_header": "Style: Default,{font_name},60,&H00000000,&H000000FF,&H00000000,&H0000FFFF,1,0,0,0,100,100,0,0,3,2,0,2,10,10,180,1"
     },
     "cinema": {
-        "name": "سینمایی (مشکی)",
-        # مشکی نیمه شفاف پشت متن، متن سفید
-        "ass_style": "BackColour=&H80000000,PrimaryColour=&H00FFFFFF,BorderStyle=3,Outline=0,Shadow=0"
+        "name": "سینمایی (کادر تیره)",
+        "ass_header": "Style: Default,{font_name},50,&H00FFFFFF,&H000000FF,&H00000000,&H80000000,0,0,0,0,100,100,0,0,3,0,0,2,10,10,50,1"
     },
     "neon": {
         "name": "نئون (آبی)",
-        # متن سفید با حاشیه ضخیم آبی، بدون کادر
-        "ass_style": "BackColour=&H00000000,PrimaryColour=&H00FFFFFF,OutlineColour=&H00FF0000,BorderStyle=1,Outline=3,Shadow=1"
+        "ass_header": "Style: Default,{font_name},60,&H00FFFFFF,&H000000FF,&H00FF0000,&H00000000,1,0,0,0,100,100,0,0,1,3,1,2,10,10,150,1"
     },
     "clean": {
-        "name": "ساده (سفید)",
-        # متن سفید با حاشیه مشکی باریک
-        "ass_style": "BackColour=&H00000000,PrimaryColour=&H00FFFFFF,OutlineColour=&H00000000,BorderStyle=1,Outline=2,Shadow=1"
+        "name": "ساده (سفید با حاشیه)",
+        "ass_header": "Style: Default,{font_name},55,&H00FFFFFF,&H000000FF,&H00000000,&H00000000,1,0,0,0,100,100,0,0,1,2,1,2,10,10,100,1"
     }
 }
 
@@ -48,7 +59,7 @@ FONTS_CONFIG = {
     "lalezar": "Lalezar"
 }
 
-# --- توابع ---
+# --- توابع کمکی ---
 
 def format_timestamp(seconds: float):
     td = timedelta(seconds=seconds)
@@ -59,58 +70,12 @@ def format_timestamp(seconds: float):
     centisecs = int(td.microseconds / 10000)
     return f"{hours:01d}:{minutes:02d}:{secs:02d}.{centisecs:02d}"
 
-def create_custom_ass(segments, output_file, style_key, font_key):
-    """ساخت فایل زیرنویس با تنظیمات کاربر"""
-    
-    # دریافت تنظیمات انتخاب شده
-    style_settings = STYLES_CONFIG.get(style_key, STYLES_CONFIG["hormozi"])["ass_style"]
+def create_ass_file(segments, output_file, style_key, font_key):
     font_name = FONTS_CONFIG.get(font_key, "Vazirmatn")
-    
-    # هدر فایل ASS
-    ass_content = f"""[Script Info]
-ScriptType: v4.00+
-PlayResX: 1080
-PlayResY: 1920
-WrapStyle: 0
+    style_template = STYLES_CONFIG.get(style_key, STYLES_CONFIG["hormozi"])["ass_header"]
+    final_style = style_template.replace("{font_name}", font_name)
 
-[V4+ Styles]
-Format: Name, Fontname, Fontsize, PrimaryColour, SecondaryColour, OutlineColour, BackColour, Bold, Italic, Underline, StrikeOut, ScaleX, ScaleY, Spacing, Angle, BorderStyle, Outline, Shadow, Alignment, MarginL, MarginR, MarginV, Encoding
-Style: Default,{font_name},70,&H00FFFFFF,&H000000FF,&H00000000,&H00000000,-1,0,0,0,100,100,0,0,1,2,0,2,10,10,180,1
-
-[Events]
-Format: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text
-"""
-    # جایگزینی استایل انتخاب شده در هدر (Trick برای ساده سازی)
-    # ما خط استایل پیش فرض را با تنظیمات کاربر آپدیت میکنیم
-    base_style = f"Style: Default,{font_name},65,{{primary}},&H000000FF,{{outline}},{{back}},-1,0,0,0,100,100,0,0,{{border}},{{out_width}},{{shadow}},2,10,10,180,1"
-    
-    # پارس کردن تنظیمات ساده به فرمت کامل ASS
-    # این یک روش ساده‌سازی شده است. برای کنترل دقیق‌تر، مقادیر را مستقیم جاگذاری می‌کنیم:
-    
-    final_style_line = f"Style: Default,{font_name},60,PrimaryColour,SecondaryColour,OutlineColour,BackColour,-1,0,0,0,100,100,0,0,BorderStyle,Outline,Shadow,2,10,10,150,1"
-    
-    # اعمال تغییرات استایل روی رشته
-    # مقادیر پیش فرض
-    params = {
-        "PrimaryColour": "&H00FFFFFF", "OutlineColour": "&H00000000", "BackColour": "&H00000000",
-        "BorderStyle": "1", "Outline": "2", "Shadow": "0"
-    }
-    
-    # آپدیت با انتخاب کاربر
-    user_settings = style_settings.split(",")
-    for setting in user_settings:
-        key, value = setting.split("=")
-        params[key] = value
-        
-    real_style_line = f"Style: Default,{font_name},65,{params['PrimaryColour']},{params['PrimaryColour']},{params['OutlineColour']},{params['BackColour']},-1,0,0,0,100,100,0,0,{params['BorderStyle']},{params['Outline']},{params['Shadow']},2,10,10,150,1"
-    
-    # بازنویسی هدر با استایل نهایی
-    ass_content = ass_content.replace(
-        "Style: Default,Noto Sans Arabic UI,60,&H00000000,&H000000FF,&H00000000,&H0000D7FF,1,0,0,0,100,100,0,0,3,2,0,2,10,10,180,1",
-        real_style_line
-    )
-    # روش مطمئن‌تر: ساخت هدر جدید
-    ass_content = f"""[Script Info]
+    header = f"""[Script Info]
 ScriptType: v4.00+
 PlayResX: 1080
 PlayResY: 1920
@@ -118,18 +83,17 @@ WrapStyle: 0
 
 [V4+ Styles]
 Format: Name, Fontname, Fontsize, PrimaryColour, SecondaryColour, OutlineColour, BackColour, Bold, Italic, Underline, StrikeOut, ScaleX, ScaleY, Spacing, Angle, BorderStyle, Outline, Shadow, Alignment, MarginL, MarginR, MarginV, Encoding
-{real_style_line}
+{final_style}
 
 [Events]
 Format: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text
 """
-
     with open(output_file, "w", encoding="utf-8") as f:
-        f.write(ass_content)
-        for segment in segments:
-            start = format_timestamp(segment.start)
-            end = format_timestamp(segment.end)
-            text = segment.text.strip()
+        f.write(header)
+        for seg in segments:
+            start = format_timestamp(seg.start)
+            end = format_timestamp(seg.end)
+            text = seg.text.strip().replace("\n", " ")
             f.write(f"Dialogue: 0,{start},{end},Default,,0,0,0,,{text}\n")
 
 def burn_subtitles(video_path, ass_path, output_path):
@@ -138,13 +102,13 @@ def burn_subtitles(video_path, ass_path, output_path):
             "ffmpeg", "-y",
             "-i", video_path,
             "-vf", f"ass={ass_path}",
-            "-c:v", "libx264", "-preset", "ultrafast", "-crf", "28",
+            "-c:v", "libx264", "-preset", "ultrafast", "-crf", "26",
             "-c:a", "copy",
             output_path
         ], check=True)
         return True
     except Exception as e:
-        print(f"Error: {e}")
+        print(f"FFmpeg Error: {e}")
         return False
 
 # --- روت‌ها ---
@@ -157,122 +121,193 @@ async def home():
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>استودیو زیرنویس هوشمند</title>
+    <title>ویرایشگر زیرنویس هوشمند</title>
     <link href="https://fonts.googleapis.com/css2?family=Vazirmatn:wght@400;700;900&display=swap" rel="stylesheet">
     <style>
-        :root { --primary: #6C63FF; --bg: #F4F7FE; --card: #ffffff; }
-        body { font-family: 'Vazirmatn', sans-serif; background: var(--bg); margin: 0; padding: 20px; display: flex; justify-content: center; min-height: 100vh; }
-        .container { background: var(--card); width: 100%; max-width: 500px; padding: 2rem; border-radius: 24px; box-shadow: 0 10px 40px rgba(0,0,0,0.05); }
+        :root { --primary: #4361ee; --bg: #f8f9fa; --card: #ffffff; --border: #e9ecef; }
+        body { font-family: 'Vazirmatn', sans-serif; background: var(--bg); margin: 0; padding: 20px; }
         
-        h1 { text-align: center; color: #2B3674; margin-bottom: 30px; }
+        .container { max-width: 800px; margin: 0 auto; background: var(--card); padding: 25px; border-radius: 20px; box-shadow: 0 10px 30px rgba(0,0,0,0.05); }
+        h1 { text-align: center; color: #2b2d42; margin-bottom: 30px; }
         
-        .upload-area { border: 2px dashed #E0E5F2; border-radius: 16px; padding: 40px; text-align: center; cursor: pointer; transition: 0.3s; margin-bottom: 20px; }
-        .upload-area:hover { border-color: var(--primary); background: #F4F7FE; }
-        .upload-icon { font-size: 40px; margin-bottom: 10px; }
+        /* آپلودر */
+        .upload-area { border: 2px dashed var(--primary); border-radius: 16px; padding: 40px; text-align: center; cursor: pointer; background: #f0f4ff; transition: 0.3s; }
+        .upload-area:hover { background: #e0e7ff; }
         
-        .options-grid { display: grid; grid-template-columns: 1fr 1fr; gap: 15px; margin-bottom: 20px; }
+        /* ادیتور */
+        .editor-area { display: none; margin-top: 30px; }
+        .segment-box { background: #fff; border: 1px solid var(--border); padding: 15px; border-radius: 12px; margin-bottom: 10px; display: flex; gap: 10px; align-items: center; box-shadow: 0 2px 5px rgba(0,0,0,0.02); }
+        .time-badge { background: #e9ecef; padding: 5px 10px; border-radius: 8px; font-size: 0.8rem; color: #495057; white-space: nowrap; direction: ltr; }
+        .text-input { flex-grow: 1; padding: 10px; border: 1px solid #ced4da; border-radius: 8px; font-family: 'Vazirmatn'; font-size: 1rem; }
+        .text-input:focus { border-color: var(--primary); outline: none; }
         
-        label { display: block; margin-bottom: 8px; font-weight: bold; color: #2B3674; font-size: 0.9rem; }
-        select { width: 100%; padding: 12px; border-radius: 12px; border: 1px solid #E0E5F2; font-family: 'Vazirmatn'; font-size: 1rem; outline: none; }
+        /* تنظیمات */
+        .settings-grid { display: grid; grid-template-columns: 1fr 1fr; gap: 15px; margin-bottom: 20px; background: #f8f9fa; padding: 15px; border-radius: 12px; }
+        select { width: 100%; padding: 10px; border-radius: 8px; border: 1px solid #ced4da; font-family: 'Vazirmatn'; }
         
-        .btn { background: var(--primary); color: white; border: none; width: 100%; padding: 15px; border-radius: 16px; font-size: 1.1rem; font-weight: bold; cursor: pointer; transition: 0.2s; margin-top: 10px; display: none; }
-        .btn:hover { background: #5751D1; transform: translateY(-2px); }
+        .btn { background: var(--primary); color: white; border: none; width: 100%; padding: 15px; border-radius: 12px; font-size: 1.1rem; font-weight: bold; cursor: pointer; transition: 0.2s; margin-top: 10px; }
+        .btn:disabled { background: #ccc; cursor: not-allowed; }
+        .btn:hover:not(:disabled) { transform: translateY(-2px); box-shadow: 0 5px 15px rgba(67, 97, 238, 0.3); }
         
-        .loader { display: none; text-align: center; margin-top: 20px; }
+        .loader { display: none; text-align: center; margin: 20px 0; }
         .spinner { width: 30px; height: 30px; border: 3px solid #f3f3f3; border-top: 3px solid var(--primary); border-radius: 50%; animation: spin 1s linear infinite; margin: 0 auto 10px; }
         @keyframes spin { 0% { transform: rotate(0deg); } 100% { transform: rotate(360deg); } }
         
-        .result-box { display: none; margin-top: 30px; text-align: center; animation: fadeIn 0.5s; }
-        video { width: 100%; border-radius: 12px; margin-bottom: 15px; box-shadow: 0 10px 20px rgba(0,0,0,0.1); }
-        .dl-btn { background: #05CD99; color: white; text-decoration: none; padding: 12px 30px; border-radius: 12px; font-weight: bold; display: inline-block; }
-        
-        @keyframes fadeIn { from{opacity:0; transform:translateY(10px)} to{opacity:1; transform:translateY(0)} }
+        #finalResult { display: none; text-align: center; margin-top: 30px; }
+        video { width: 100%; max-height: 500px; border-radius: 12px; background: #000; }
+        .dl-btn { display: inline-block; background: #2ec4b6; color: white; text-decoration: none; padding: 12px 30px; border-radius: 10px; font-weight: bold; margin-top: 10px; }
     </style>
 </head>
 <body>
     <div class="container">
-        <h1>✨ زیرنویس‌ساز جادویی</h1>
+        <h1>✂️ ویرایشگر زیرنویس هوشمند</h1>
         
-        <div class="upload-area" onclick="document.getElementById('fileInput').click()">
-            <div class="upload-icon">📹</div>
-            <div id="fileName">برای انتخاب ویدیو کلیک کنید</div>
-            <input type="file" id="fileInput" accept="video/*" hidden onchange="handleFile(this)">
+        <!-- مرحله ۱: آپلود -->
+        <div class="upload-area" id="uploadArea" onclick="document.getElementById('fileInput').click()">
+            <div style="font-size: 40px;">📂</div>
+            <p id="uploadText">برای انتخاب ویدیو کلیک کنید</p>
+            <input type="file" id="fileInput" accept="video/*" hidden onchange="handleUpload(this)">
         </div>
 
-        <div class="options-grid">
-            <div>
-                <label>🎨 استایل زیرنویس</label>
-                <select id="styleSelect">
-                    <option value="hormozi">هرمزی (کادر زرد)</option>
-                    <option value="cinema">سینمایی (کادر مشکی)</option>
-                    <option value="neon">نئون (آبی)</option>
-                    <option value="clean">ساده (سفید)</option>
-                </select>
-            </div>
-            <div>
-                <label>✒️ فونت</label>
-                <select id="fontSelect">
-                    <option value="vazir">وزیر (استاندارد)</option>
-                    <option value="lalezar">لاله زار (ضخیم)</option>
-                </select>
-            </div>
+        <div id="analyzeLoader" class="loader">
+            <div class="spinner"></div>
+            <span>در حال استخراج متن از ویدیو... (لطفا صبر کنید)</span>
         </div>
 
-        <button id="processBtn" class="btn" onclick="startProcess()">شروع پردازش ⚡</button>
+        <!-- مرحله ۲: ویرایش -->
+        <div id="editorArea" class="editor-area">
+            <div class="settings-grid">
+                <div>
+                    <label>استایل:</label>
+                    <select id="styleSelect">
+                        <option value="hormozi">هرمزی (زرد)</option>
+                        <option value="cinema">سینمایی (تیره)</option>
+                        <option value="neon">نئون (آبی)</option>
+                        <option value="clean">ساده</option>
+                    </select>
+                </div>
+                <div>
+                    <label>فونت:</label>
+                    <select id="fontSelect">
+                        <option value="vazir">وزیر (رسمی)</option>
+                        <option value="lalezar">لاله زار (تیتر)</option>
+                    </select>
+                </div>
+            </div>
 
-        <div id="loader" class="loader">
+            <h3>📝 متن‌ها را ویرایش کنید:</h3>
+            <div id="segmentsContainer"></div>
+            
+            <button id="burnBtn" class="btn" onclick="burnVideo()">🔥 ساخت ویدیو نهایی</button>
+        </div>
+
+        <div id="burnLoader" class="loader">
             <div class="spinner"></div>
-            <span id="statusText">در حال گوش دادن و نوشتن...</span>
+            <span>در حال چسباندن زیرنویس‌ها...</span>
         </div>
 
-        <div id="result" class="result-box">
-            <video id="finalVideo" controls></video>
+        <!-- مرحله ۳: دانلود -->
+        <div id="finalResult">
+            <h3>✅ ویدیو آماده شد!</h3>
+            <video id="previewVideo" controls></video>
             <br>
-            <a id="dlLink" class="dl-btn" download>دانلود ویدیو نهایی</a>
+            <a id="downloadLink" class="dl-btn" download>دانلود ویدیو</a>
             <br><br>
-            <a href="javascript:location.reload()" style="color: #A3AED0; text-decoration: none; font-size: 0.9rem;">شروع دوباره</a>
+            <button onclick="location.reload()" style="background:none; border:none; color:#666; cursor:pointer;">شروع دوباره</button>
         </div>
     </div>
 
     <script>
-        let selectedFile = null;
+        let currentFileId = null;
+        let currentSegments = [];
+
+        async function handleUpload(input) {
+            const file = input.files[0];
+            if (!file) return;
+
+            // UI Update
+            document.getElementById('uploadArea').style.display = 'none';
+            document.getElementById('analyzeLoader').style.display = 'block';
+
+            const formData = new FormData();
+            formData.append("file", file);
 
-        function handleFile(input) {
-            if (input.files[0]) {
-                selectedFile = input.files[0];
-                document.getElementById('fileName').innerText = selectedFile.name;
-                document.getElementById('processBtn').style.display = 'block';
+            try {
+                const res = await fetch("/api/analyze", { method: "POST", body: formData });
+                const data = await res.json();
+
+                if (data.error) throw new Error(data.error);
+
+                currentFileId = data.file_id;
+                currentSegments = data.segments;
+                
+                renderEditor();
+            } catch (e) {
+                alert("خطا: " + e.message);
+                location.reload();
             }
         }
 
-        async function startProcess() {
-            if (!selectedFile) return;
+        function renderEditor() {
+            document.getElementById('analyzeLoader').style.display = 'none';
+            const container = document.getElementById('segmentsContainer');
+            container.innerHTML = "";
 
-            // UI setup
-            document.querySelector('.upload-area').style.display = 'none';
-            document.querySelector('.options-grid').style.display = 'none';
-            document.getElementById('processBtn').style.display = 'none';
-            document.getElementById('loader').style.display = 'block';
+            currentSegments.forEach((seg, index) => {
+                const div = document.createElement('div');
+                div.className = 'segment-box';
+                
+                const time = document.createElement('div');
+                time.className = 'time-badge';
+                // نمایش زمان به فرمت ساده (ثانیه)
+                time.innerText = `${Math.floor(seg.start)}s - ${Math.floor(seg.end)}s`;
+                
+                const input = document.createElement('input');
+                input.type = 'text';
+                input.className = 'text-input';
+                input.value = seg.text.trim();
+                input.oninput = (e) => { currentSegments[index].text = e.target.value; };
 
-            const formData = new FormData();
-            formData.append("file", selectedFile);
-            formData.append("style", document.getElementById('styleSelect').value);
-            formData.append("font", document.getElementById('fontSelect').value);
+                div.appendChild(time);
+                div.appendChild(input);
+                container.appendChild(div);
+            });
+
+            document.getElementById('editorArea').style.display = 'block';
+        }
+
+        async function burnVideo() {
+            document.getElementById('editorArea').style.display = 'none';
+            document.getElementById('burnLoader').style.display = 'block';
+
+            const payload = {
+                file_id: currentFileId,
+                style: document.getElementById('styleSelect').value,
+                font: document.getElementById('fontSelect').value,
+                segments: currentSegments
+            };
 
             try {
-                const res = await fetch("/api/process", { method: "POST", body: formData });
-                if (!res.ok) throw new Error("خطا در سرور");
-                
+                const res = await fetch("/api/burn", {
+                    method: "POST",
+                    headers: { "Content-Type": "application/json" },
+                    body: JSON.stringify(payload)
+                });
                 const data = await res.json();
+
+                if (data.error) throw new Error(data.error);
+
+                document.getElementById('burnLoader').style.display = 'none';
+                document.getElementById('finalResult').style.display = 'block';
                 
-                document.getElementById('loader').style.display = 'none';
-                document.getElementById('result').style.display = 'block';
-                document.getElementById('finalVideo').src = data.url;
-                document.getElementById('dlLink').href = data.url;
+                const videoEl = document.getElementById('previewVideo');
+                videoEl.src = data.url;
+                document.getElementById('downloadLink').href = data.url;
 
             } catch (e) {
-                alert("مشکلی پیش آمد: " + e.message);
-                location.reload();
+                alert("خطا در ساخت ویدیو: " + e.message);
+                document.getElementById('burnLoader').style.display = 'none';
+                document.getElementById('editorArea').style.display = 'block';
             }
         }
     </script>
@@ -280,45 +315,59 @@ async def home():
 </html>
     """
 
-@app.post("/api/process")
-async def process_video(
-    file: UploadFile = File(...),
-    style: str = Form("hormozi"),
-    font: str = Form("vazir")
-):
+@app.post("/api/analyze")
+async def analyze_video(file: UploadFile = File(...)):
     try:
-        job_id = str(uuid.uuid4())
-        input_path = f"{TEMP_DIR}/{job_id}_in.mp4"
-        ass_path = f"{TEMP_DIR}/{job_id}.ass"
-        output_path = f"{TEMP_DIR}/{job_id}_out.mp4"
+        file_id = str(uuid.uuid4())
+        file_path = f"{TEMP_DIR}/{file_id}_raw.mp4"
+        
+        with open(file_path, "wb") as f:
+            shutil.copyfileobj(file.file, f)
+            
+        # استخراج متن
+        segments_gen, _ = model.transcribe(file_path, language="fa", beam_size=5)
         
-        # ذخیره
-        with open(input_path, "wb") as buffer:
-            shutil.copyfileobj(file.file, buffer)
+        # تبدیل به فرمت JSON قابل ارسال
+        segments_data = []
+        for seg in segments_gen:
+            segments_data.append({
+                "start": seg.start,
+                "end": seg.end,
+                "text": seg.text
+            })
             
-        # پردازش متن
-        segments, _ = model.transcribe(input_path, language="fa", beam_size=5)
-        segments = list(segments)
+        return {"file_id": file_id, "segments": segments_data}
+        
+    except Exception as e:
+        return JSONResponse(status_code=500, content={"error": str(e)})
+
+@app.post("/api/burn")
+async def burn_video_api(data: BurnRequest):
+    try:
+        input_path = f"{TEMP_DIR}/{data.file_id}_raw.mp4"
+        ass_path = f"{TEMP_DIR}/{data.file_id}.ass"
+        output_path = f"{TEMP_DIR}/{data.file_id}_final.mp4"
         
-        if not segments:
-            return JSONResponse(status_code=400, content={"error": "متنی یافت نشد"})
+        if not os.path.exists(input_path):
+            return JSONResponse(status_code=404, content={"error": "فایل یافت نشد (منقضی شده)"})
 
-        # ساخت زیرنویس با استایل انتخابی
-        create_custom_ass(segments, ass_path, style, font)
+        # 1. ساخت فایل ASS از روی دیتای ویرایش شده کاربر
+        create_ass_file(data.segments, ass_path, data.style, data.font)
         
-        # چسباندن
+        # 2. چسباندن
         if burn_subtitles(input_path, ass_path, output_path):
+            # پاک کردن فایل‌های موقت
             os.remove(input_path)
             os.remove(ass_path)
-            return {"url": f"/download/{job_id}_out.mp4"}
+            return {"url": f"/download/{data.file_id}_final.mp4"}
         else:
-            return JSONResponse(status_code=500, content={"error": "Burn Error"})
+            return JSONResponse(status_code=500, content={"error": "خطا در پردازش ویدیو"})
             
     except Exception as e:
         return JSONResponse(status_code=500, content={"error": str(e)})
 
 @app.get("/download/{filename}")
-async def download_file(filename: str):
+async def download(filename: str):
     path = f"{TEMP_DIR}/{filename}"
     if os.path.exists(path): return FileResponse(path)
     return JSONResponse(status_code=404, content={"error": "Not Found"})