Update app.py

app.py

@@ -3,7 +3,7 @@ import shutil
 import subprocess
 import uuid
 from datetime import timedelta
-from fastapi import FastAPI, UploadFile, File, BackgroundTasks
+from fastapi import FastAPI, UploadFile, File, Form
 from fastapi.responses import HTMLResponse, FileResponse, JSONResponse
 from faster_whisper import WhisperModel
 
@@ -13,15 +13,44 @@ app = FastAPI()
 TEMP_DIR = "temp"
 os.makedirs(TEMP_DIR, exist_ok=True)
 
-# لود کردن مدل هوش مصنوعی (فقط یک بار اجرا می‌شود)
+# لود مدل
 print("Loading Whisper AI...")
 model = WhisperModel("small", device="cpu", compute_type="int8")
 print("Model Loaded!")
 
-# --- توابع کمکی ---
+# --- تعریف استایل‌ها و فونت‌ها ---
+
+STYLES_CONFIG = {
+    "hormozi": {
+        "name": "هرمزی (زرد)",
+        # زرد جیغ پشت متن، متن مشکی
+        "ass_style": "BackColour=&H0000FFFF,PrimaryColour=&H00000000,BorderStyle=3,Outline=0,Shadow=0"
+    },
+    "cinema": {
+        "name": "سینمایی (مشکی)",
+        # مشکی نیمه شفاف پشت متن، متن سفید
+        "ass_style": "BackColour=&H80000000,PrimaryColour=&H00FFFFFF,BorderStyle=3,Outline=0,Shadow=0"
+    },
+    "neon": {
+        "name": "نئون (آبی)",
+        # متن سفید با حاشیه ضخیم آبی، بدون کادر
+        "ass_style": "BackColour=&H00000000,PrimaryColour=&H00FFFFFF,OutlineColour=&H00FF0000,BorderStyle=1,Outline=3,Shadow=1"
+    },
+    "clean": {
+        "name": "ساده (سفید)",
+        # متن سفید با حاشیه مشکی باریک
+        "ass_style": "BackColour=&H00000000,PrimaryColour=&H00FFFFFF,OutlineColour=&H00000000,BorderStyle=1,Outline=2,Shadow=1"
+    }
+}
+
+FONTS_CONFIG = {
+    "vazir": "Vazirmatn",
+    "lalezar": "Lalezar"
+}
+
+# --- توابع ---
 
 def format_timestamp(seconds: float):
-    """تبدیل زمان به فرمت ASS"""
     td = timedelta(seconds=seconds)
     total_seconds = int(td.total_seconds())
     hours = total_seconds // 3600
@@ -30,10 +59,58 @@ def format_timestamp(seconds: float):
     centisecs = int(td.microseconds / 10000)
     return f"{hours:01d}:{minutes:02d}:{secs:02d}.{centisecs:02d}"
 
-def create_styled_ass(segments, output_file):
-    """ساخت زیرنویس با استایل زرد و مشکی جذاب (هرمزی)"""
-    # تنظیمات استایل: فونت Noto Sans Arabic، سایز بزرگ، کادر زرد
-    ass_content = """[Script Info]
+def create_custom_ass(segments, output_file, style_key, font_key):
+    """ساخت فایل زیرنویس با تنظیمات کاربر"""
+    
+    # دریافت تنظیمات انتخاب شده
+    style_settings = STYLES_CONFIG.get(style_key, STYLES_CONFIG["hormozi"])["ass_style"]
+    font_name = FONTS_CONFIG.get(font_key, "Vazirmatn")
+    
+    # هدر فایل ASS
+    ass_content = f"""[Script Info]
+ScriptType: v4.00+
+PlayResX: 1080
+PlayResY: 1920
+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
+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]
 ScriptType: v4.00+
 PlayResX: 1080
 PlayResY: 1920
@@ -41,11 +118,12 @@ 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
-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}
 
 [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:
@@ -55,13 +133,12 @@ Format: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text
             f.write(f"Dialogue: 0,{start},{end},Default,,0,0,0,,{text}\n")
 
 def burn_subtitles(video_path, ass_path, output_path):
-    """چسباندن زیرنویس به ویدیو"""
     try:
         subprocess.run([
             "ffmpeg", "-y",
             "-i", video_path,
             "-vf", f"ass={ass_path}",
-            "-c:v", "libx264", "-preset", "ultrafast", "-crf", "28", # تنظیم سرعت بالا
+            "-c:v", "libx264", "-preset", "ultrafast", "-crf", "28",
             "-c:a", "copy",
             output_path
         ], check=True)
@@ -80,59 +157,80 @@ 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: #FFD700; --dark: #1a1a1a; }
-        body { font-family: 'Vazirmatn', sans-serif; background: #f0f2f5; display: flex; justify-content: center; align-items: center; min-height: 100vh; margin: 0; }
-        .container { background: white; width: 90%; max-width: 450px; padding: 2rem; border-radius: 24px; box-shadow: 0 10px 30px rgba(0,0,0,0.1); text-align: center; }
-        h1 { color: var(--dark); font-weight: 900; margin-bottom: 5px; }
-        p { color: #666; font-size: 0.9rem; margin-bottom: 2rem; }
+        :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); }
+        
+        h1 { text-align: center; color: #2B3674; margin-bottom: 30px; }
         
-        .upload-box { border: 2px dashed #ddd; padding: 2rem; border-radius: 16px; cursor: pointer; transition: 0.3s; background: #fafafa; position: relative; overflow: hidden; }
-        .upload-box:hover { border-color: var(--primary); background: #fffdf0; }
-        .upload-icon { font-size: 3rem; color: #ccc; margin-bottom: 10px; }
+        .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; }
         
-        #videoPreview { width: 100%; border-radius: 12px; display: none; margin-top: 1rem; box-shadow: 0 5px 15px rgba(0,0,0,0.1); }
+        .options-grid { display: grid; grid-template-columns: 1fr 1fr; gap: 15px; margin-bottom: 20px; }
         
-        .btn { background: var(--dark); color: var(--primary); border: none; padding: 1rem 2rem; font-size: 1.1rem; font-weight: bold; border-radius: 12px; cursor: pointer; width: 100%; margin-top: 1.5rem; transition: 0.3s; display: none; }
-        .btn:hover { transform: translateY(-3px); box-shadow: 0 5px 15px rgba(0,0,0,0.2); }
+        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; }
         
-        .loader { display: none; margin-top: 20px; }
-        .spinner { width: 40px; height: 40px; border: 4px solid #f3f3f3; border-top: 4px solid var(--primary); border-radius: 50%; animation: spin 1s linear infinite; margin: 0 auto 10px; }
+        .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); }
+        
+        .loader { display: none; text-align: center; margin-top: 20px; }
+        .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); } }
-        .status-text { font-size: 0.9rem; color: #555; font-weight: bold; }
         
-        .result-area { display: none; margin-top: 2rem; animation: fadeIn 0.5s; }
-        .download-btn { text-decoration: none; background: #25D366; color: white; padding: 10px 20px; border-radius: 10px; display: inline-block; font-weight: bold; margin-top: 10px; }
+        .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)} }
     </style>
 </head>
 <body>
     <div class="container">
-        <h1>زیرنویس‌ساز جادویی ✨</h1>
-        <p>ویدیو بده، با زیرنویس هرمزی تحویل بگیر!</p>
+        <h1>✨ زیرنویس‌ساز جادویی</h1>
         
-        <div class="upload-box" onclick="document.getElementById('fileInput').click()">
-            <div class="upload-icon">📂</div>
+        <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>
 
-        <button id="startBtn" class="btn" onclick="uploadAndProcess()">شروع ساخت زیرنویس ⚡</button>
-        
+        <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>
+
+        <button id="processBtn" class="btn" onclick="startProcess()">شروع پردازش ⚡</button>
+
         <div id="loader" class="loader">
             <div class="spinner"></div>
-            <div class="status-text" id="statusText">در حال گوش دادن به ویدیو...</div>
-            <small style="color:#999; display:block; margin-top:5px;">(ممکن است ۱ تا ۳ دقیقه طول بکشد)</small>
+            <span id="statusText">در حال گوش دادن و نوشتن...</span>
         </div>
 
-        <div id="resultArea" class="result-area">
-            <h3>✅ ویدیو آماده شد!</h3>
-            <video id="finalVideo" controls style="width:100%; border-radius:12px;"></video>
-            <a id="downloadLink" class="download-btn" download>دانلود ویدیو</a>
-            <button onclick="location.reload()" style="background:none; border:none; color:#666; margin-top:15px; cursor:pointer; text-decoration:underline;">ساخت ویدیو جدید</button>
+        <div id="result" class="result-box">
+            <video id="finalVideo" controls></video>
+            <br>
+            <a id="dlLink" class="dl-btn" download>دانلود ویدیو نهایی</a>
+            <br><br>
+            <a href="javascript:location.reload()" style="color: #A3AED0; text-decoration: none; font-size: 0.9rem;">شروع دوباره</a>
         </div>
     </div>
 
@@ -140,50 +238,40 @@ async def home():
         let selectedFile = null;
 
         function handleFile(input) {
-            if (input.files && input.files[0]) {
+            if (input.files[0]) {
                 selectedFile = input.files[0];
                 document.getElementById('fileName').innerText = selectedFile.name;
-                document.getElementById('startBtn').style.display = 'block';
+                document.getElementById('processBtn').style.display = 'block';
             }
         }
 
-        async function uploadAndProcess() {
+        async function startProcess() {
             if (!selectedFile) return;
 
-            // UI Updates
-            document.querySelector('.upload-box').style.display = 'none';
-            document.getElementById('startBtn').style.display = 'none';
+            // 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';
 
             const formData = new FormData();
             formData.append("file", selectedFile);
+            formData.append("style", document.getElementById('styleSelect').value);
+            formData.append("font", document.getElementById('fontSelect').value);
 
             try {
-                // Step 1: Upload & Process
-                document.getElementById('statusText').innerText = "هوش مصنوعی در حال نوشتن زیرنویس...";
+                const res = await fetch("/api/process", { method: "POST", body: formData });
+                if (!res.ok) throw new Error("خطا در سرور");
                 
-                const response = await fetch("/api/process", {
-                    method: "POST",
-                    body: formData
-                });
-
-                if (!response.ok) throw new Error("Server Error");
-
-                const data = await response.json();
+                const data = await res.json();
                 
-                // Step 2: Show Result
                 document.getElementById('loader').style.display = 'none';
-                const resultArea = document.getElementById('resultArea');
-                resultArea.style.display = 'block';
-                
-                const videoPlayer = document.getElementById('finalVideo');
-                videoPlayer.src = data.url;
-                
-                const downloadLink = document.getElementById('downloadLink');
-                downloadLink.href = data.url;
+                document.getElementById('result').style.display = 'block';
+                document.getElementById('finalVideo').src = data.url;
+                document.getElementById('dlLink').href = data.url;
 
-            } catch (error) {
-                alert("خطایی رخ داد. لطفا دوباره تلاش کنید.");
+            } catch (e) {
+                alert("مشکلی پیش آمد: " + e.message);
                 location.reload();
             }
         }
@@ -193,42 +281,44 @@ async def home():
     """
 
 @app.post("/api/process")
-async def process_video(file: UploadFile = File(...)):
+async def process_video(
+    file: UploadFile = File(...),
+    style: str = Form("hormozi"),
+    font: str = Form("vazir")
+):
     try:
         job_id = str(uuid.uuid4())
-        input_path = f"{TEMP_DIR}/{job_id}_input.mp4"
+        input_path = f"{TEMP_DIR}/{job_id}_in.mp4"
         ass_path = f"{TEMP_DIR}/{job_id}.ass"
-        output_path = f"{TEMP_DIR}/{job_id}_final.mp4"
+        output_path = f"{TEMP_DIR}/{job_id}_out.mp4"
         
-        # 1. ذخیره فایل ورودی
+        # ذخیره
         with open(input_path, "wb") as buffer:
             shutil.copyfileobj(file.file, buffer)
             
-        # 2. تبدیل صدا به متن (فارسی)
+        # پردازش متن
         segments, _ = model.transcribe(input_path, language="fa", beam_size=5)
-        segments = list(segments) # اجرا
+        segments = list(segments)
         
         if not segments:
-            return JSONResponse(status_code=400, content={"error": "No speech found"})
+            return JSONResponse(status_code=400, content={"error": "متنی یافت نشد"})
 
-        # 3. ساخت فایل زیرنویس گرافیکی
-        create_styled_ass(segments, ass_path)
+        # ساخت زیرنویس با استایل انتخابی
+        create_custom_ass(segments, ass_path, style, font)
         
-        # 4. چسباندن زیرنویس
+        # چسباندن
         if burn_subtitles(input_path, ass_path, output_path):
-            # پاک کردن فایل‌های اضافی
             os.remove(input_path)
             os.remove(ass_path)
-            return {"url": f"/download/{job_id}_final.mp4"}
+            return {"url": f"/download/{job_id}_out.mp4"}
         else:
-            return JSONResponse(status_code=500, content={"error": "FFmpeg failed"})
+            return JSONResponse(status_code=500, content={"error": "Burn Error"})
             
     except Exception as e:
         return JSONResponse(status_code=500, content={"error": str(e)})
 
 @app.get("/download/{filename}")
 async def download_file(filename: str):
-    file_path = f"{TEMP_DIR}/{filename}"
-    if os.path.exists(file_path):
-        return FileResponse(file_path)
-    return JSONResponse(status_code=404, content={"error": "File not found"})
+    path = f"{TEMP_DIR}/{filename}"
+    if os.path.exists(path): return FileResponse(path)
+    return JSONResponse(status_code=404, content={"error": "Not Found"})