Update app.py

app.py

@@ -6,7 +6,8 @@ import time
 import sqlite3
 import tempfile
 import ffmpeg
-from flask import Flask, request, jsonify, render_template, send_from_directory, send_file, after_this_request
+import json
+from flask import Flask, request, jsonify, render_template, send_from_directory, send_file, after_this_request, Response
 from deep_translator import GoogleTranslator
 
 app = Flask(__name__)
@@ -17,10 +18,8 @@ GITHUB_REPO = os.environ.get("GITHUB_REPO", "Action")
 SPACE_HOST = os.environ.get("SPACE_HOST", "")
 SPACE_URL = f"https://{SPACE_HOST}"
 
-# --- کلمات ممنوعه ---
 FORBIDDEN_KEYWORDS = {"sex", "sexy", "porn", "nude", "erotic", "سکس", "سکسی", "پورن", "شهوانی", "برهنه", "لخت"}
 
-# --- تنظیمات دیتابیس SQLite ---
 DB_DIR = "/data" if os.path.exists("/data") else "."
 DB_PATH = os.path.join(DB_DIR, "users.db")
 TEMP_DIR = "tmp"
@@ -28,44 +27,49 @@ USAGE_LIMIT = 5
 ONE_WEEK_SECONDS = 7 * 24 * 60 * 60
 ONE_DAY_SECONDS = 24 * 60 * 60  
 
-# متغیرهای مربوط به دیتابیس جدید تصاویر
 DB_IMAGE_PATH = os.path.join(DB_DIR, "users_image.db")
 IMAGE_USAGE_LIMIT = 5
 
-# متغیرهای مربوط به دیتابیس جدید ویرایش تصاویر
 DB_IMAGE_EDIT_PATH = os.path.join(DB_DIR, "users_image_edit.db")
 EDIT_USAGE_LIMIT = 5
 
+DB_CHAT_QUEUE_PATH = os.path.join(DB_DIR, "chat_queue.db")
+
 os.makedirs(TEMP_DIR, exist_ok=True)
 os.makedirs("static/images", exist_ok=True)
 
 def init_db():
     conn = sqlite3.connect(DB_PATH)
     c = conn.cursor()
-    c.execute('''CREATE TABLE IF NOT EXISTS users
-                 (id TEXT PRIMARY KEY, count INTEGER, week_start REAL)''')
+    c.execute('''CREATE TABLE IF NOT EXISTS users (id TEXT PRIMARY KEY, count INTEGER, week_start REAL)''')
     conn.commit()
     conn.close()
 
 def init_image_db():
     conn = sqlite3.connect(DB_IMAGE_PATH)
     c = conn.cursor()
-    c.execute('''CREATE TABLE IF NOT EXISTS users
-                 (id TEXT PRIMARY KEY, count INTEGER, week_start REAL)''')
+    c.execute('''CREATE TABLE IF NOT EXISTS users (id TEXT PRIMARY KEY, count INTEGER, week_start REAL)''')
     conn.commit()
     conn.close()
 
 def init_image_edit_db():
     conn = sqlite3.connect(DB_IMAGE_EDIT_PATH)
     c = conn.cursor()
-    c.execute('''CREATE TABLE IF NOT EXISTS users
-                 (id TEXT PRIMARY KEY, count INTEGER, week_start REAL)''')
+    c.execute('''CREATE TABLE IF NOT EXISTS users (id TEXT PRIMARY KEY, count INTEGER, week_start REAL)''')
+    conn.commit()
+    conn.close()
+
+def init_chat_queue_db():
+    conn = sqlite3.connect(DB_CHAT_QUEUE_PATH)
+    c = conn.cursor()
+    c.execute('''CREATE TABLE IF NOT EXISTS queue (run_id TEXT PRIMARY KEY, prompt TEXT, file_url TEXT, file_mime TEXT, status TEXT, timestamp REAL)''')
     conn.commit()
     conn.close()
 
 init_db()
 init_image_db()
 init_image_edit_db()
+init_chat_queue_db()
 
 def get_db_connection():
     conn = sqlite3.connect(DB_PATH)
@@ -82,7 +86,11 @@ def get_image_edit_db_connection():
     conn.row_factory = sqlite3.Row
     return conn
 
-# --- توابع کمکی ---
+def get_chat_queue_db_connection():
+    conn = sqlite3.connect(DB_CHAT_QUEUE_PATH)
+    conn.row_factory = sqlite3.Row
+    return conn
+
 def delete_github_run(gh_run_id):
     time.sleep(20)
     url = f"https://api.github.com/repos/{GITHUB_USER}/{GITHUB_REPO}/actions/runs/{gh_run_id}"
@@ -100,36 +108,25 @@ def has_forbidden_words(text):
     text_lower = text.lower()
     return any(keyword in text_lower for keyword in FORBIDDEN_KEYWORDS)
 
-# --- مسیرهای صفحات HTML ---
 @app.route('/')
 def index():
     return render_template('fluxpro.html')
 
 @app.route('/<page_name>')
 def serve_html(page_name):
-    if not page_name.endswith('.html'):
-        page_name += '.html'
-    try:
-        return render_template(page_name)
-    except:
-        return "Page not found", 404
+    if not page_name.endswith('.html'): page_name += '.html'
+    try: return render_template(page_name)
+    except: return "Page not found", 404
 
-# --- API برای ساخت عکس (Flux و Cartoon) ---
 @app.route('/api/generate', methods=['POST'])
 def generate():
-    if not GITHUB_TOKEN: 
-        return jsonify({"status": "error", "message": "Secret GITHUB_TOKEN is missing!"}), 500
-        
+    if not GITHUB_TOKEN: return jsonify({"status": "error", "message": "Secret GITHUB_TOKEN is missing!"}), 500
     data = request.json
     persian_prompt = data.get('prompt', '')
 
-    if has_forbidden_words(persian_prompt):
-        return jsonify({"status": "error", "message": "متن ورودی شامل کلمات نامناسب است."}), 400
-    
-    try: 
-        english_prompt = GoogleTranslator(source='fa', target='en').translate(persian_prompt)
-    except: 
-        english_prompt = persian_prompt
+    if has_forbidden_words(persian_prompt): return jsonify({"status": "error", "message": "متن ورودی شامل کلمات نامناسب است."}), 400
+    try: english_prompt = GoogleTranslator(source='fa', target='en').translate(persian_prompt)
+    except: english_prompt = persian_prompt
 
     run_id = str(uuid.uuid4())
     url = f"https://api.github.com/repos/{GITHUB_USER}/{GITHUB_REPO}/dispatches"
@@ -137,66 +134,37 @@ def generate():
     
     payload = {
         "event_type": data.get('action_name', 'generate-flux'),
-        "client_payload": {
-            "prompt": english_prompt, 
-            "width": data.get('width', 1024), 
-            "height": data.get('height', 1024), 
-            "run_id": run_id, 
-            "space_url": SPACE_URL
-        }
+        "client_payload": {"prompt": english_prompt, "width": data.get('width', 1024), "height": data.get('height', 1024), "run_id": run_id, "space_url": SPACE_URL}
     }
     r = requests.post(url, headers=headers, json=payload)
-    
     if r.status_code == 204:
-        return jsonify({
-            "status": "success", 
-            "run_id": run_id, 
-            "translated_prompt": english_prompt, 
-            "translated": english_prompt
-        })
-    else:
-        return jsonify({"status": "error", "message": r.text}), 400
+        return jsonify({"status": "success", "run_id": run_id, "translated_prompt": english_prompt, "translated": english_prompt})
+    else: return jsonify({"status": "error", "message": r.text}), 400
 
-# --- API مخصوص ویرایش عکس (AI Photoshop) ---
 @app.route('/api/edit', methods=['POST'])
 def edit_image():
     if not GITHUB_TOKEN: return jsonify({"status": "error", "message": "Token missing"}), 500
     if 'image' not in request.files: return jsonify({"status": "error", "message": "Image required"}), 400
-    
     file = request.files['image']
     persian_prompt = request.form.get('prompt', '')
 
-    if has_forbidden_words(persian_prompt):
-        return jsonify({"status": "error", "message": "متن ورودی شامل کلمات نامناسب است."}), 400
-    
+    if has_forbidden_words(persian_prompt): return jsonify({"status": "error", "message": "متن نامناسب"}), 400
     try: english_prompt = GoogleTranslator(source='fa', target='en').translate(persian_prompt)
     except: english_prompt = persian_prompt
 
     run_id = str(uuid.uuid4())
-    
     input_filename = f"{run_id}_input.png"
     file.save(f"static/images/{input_filename}")
     image_public_url = f"{SPACE_URL}/static/images/{input_filename}"
 
     url = f"https://api.github.com/repos/{GITHUB_USER}/{GITHUB_REPO}/dispatches"
     headers = {"Accept": "application/vnd.github.v3+json", "Authorization": f"token {GITHUB_TOKEN}"}
-    payload = {
-        "event_type": "generate-editor",
-        "client_payload": {
-            "prompt": english_prompt,
-            "image_url": image_public_url,
-            "run_id": run_id,
-            "space_url": SPACE_URL
-        }
-    }
+    payload = {"event_type": "generate-editor", "client_payload": {"prompt": english_prompt, "image_url": image_public_url, "run_id": run_id, "space_url": SPACE_URL}}
     r = requests.post(url, headers=headers, json=payload)
     
-    if r.status_code == 204:
-        return jsonify({"status": "success", "run_id": run_id, "translated": english_prompt})
-    else:
-        return jsonify({"status": "error", "message": r.text}), 400
+    if r.status_code == 204: return jsonify({"status": "success", "run_id": run_id, "translated": english_prompt})
+    else: return jsonify({"status": "error", "message": r.text}), 400
 
-# --- API مخصوص تبدیل عکس به ویدیو ---
 @app.route('/api/generate-video', methods=['POST'])
 def generate_video():
     if not GITHUB_TOKEN: return jsonify({"status": "error", "message": "Token missing"}), 500
@@ -206,116 +174,62 @@ def generate_video():
     user_prompt = request.form.get('prompt', '').strip()
     duration = request.form.get('duration', 5.0)
 
-    if has_forbidden_words(user_prompt):
-        return jsonify({"status": "error", "message": "متن ورودی شامل کلمات نامناسب است."}), 400
+    if has_forbidden_words(user_prompt): return jsonify({"status": "error", "message": "کلمات نامناسب"}), 400
     
     run_id = str(uuid.uuid4())
-    
     input_filename = f"{run_id}_video_input.png"
-    local_image_path = os.path.join("static/images", input_filename)
-    file.save(local_image_path)
+    file.save(os.path.join("static/images", input_filename))
     image_public_url = f"{SPACE_URL}/static/images/{input_filename}"
 
-    master_prompt = """You are an expert AI Animation Planner. Your absolute highest priority is to faithfully and creatively execute the user's specific request based on the provided image.
-
-    1. If the user prompt is empty or generic (like "animate this"), add subtle, high-quality, believable cinematic motion (e.g., slow zoom, water flowing, wind in hair).
-    2. If the user gives specific directions, focus ENTIRELY on executing that command perfectly. If the action is not visible in-frame, use cinematic camera movements to reveal it.
-    3. You must output ONLY a highly detailed, descriptive animation prompt in ENGLISH. Do not translate literally; ENHANCE the prompt for a text-to-video AI model.
-    4. MUST Include keywords at the end: cinematic, photorealistic, high detail, smooth motion, 8k.
-
-    CRITICAL RULE: DO NOT say "Here is the prompt" or give any conversational explanations. DO NOT output JSON. Output ONLY the raw English animation prompt text and NOTHING ELSE."""
-
-    english_prompt = ""
-    gemma_success = False
-    
-    prompt_for_gemma = user_prompt if user_prompt else "لطفاً این تصویر را به یک ویدیوی سینمایی بسیار جذاب و واقع‌گرایانه متحرک کن."
-    combined_prompt = f"{master_prompt}\n\nUser request: {prompt_for_gemma}"
+    master_prompt = "You are an expert AI Animation Planner. Your absolute highest priority is to faithfully and creatively execute the user's specific request based on the provided image. MUST Include keywords at the end: cinematic, photorealistic, high detail, smooth motion, 8k. Output ONLY the raw English animation prompt text."
+    combined_prompt = f"{master_prompt}\n\nUser request: {user_prompt if user_prompt else 'animate this image cinematic'}"
     
     url = f"https://api.github.com/repos/{GITHUB_USER}/{GITHUB_REPO}/dispatches"
     headers = {"Accept": "application/vnd.github.v3+json", "Authorization": f"token {GITHUB_TOKEN}"}
+    gemma_success = False
+    english_prompt = ""
 
     for attempt in range(2):
         try:
             gemma_run_id = f"gemma_{run_id}_{attempt}"
             gemma_txt_path = os.path.join("static/images", f"{gemma_run_id}.txt")
+            payload_gemma = {"event_type": "chat-gemma", "client_payload": {"prompt": combined_prompt, "file_url": image_public_url, "file_mime": "image/png", "run_id": gemma_run_id, "space_url": SPACE_URL}}
             
-            payload_gemma = {
-                "event_type": "chat-gemma",
-                "client_payload": {
-                    "prompt": combined_prompt,
-                    "file_url": image_public_url,
-                    "file_mime": "image/png",
-                    "run_id": gemma_run_id,
-                    "space_url": SPACE_URL
-                }
-            }
-            
-            r_gemma = requests.post(url, headers=headers, json=payload_gemma)
-            
-            if r_gemma.status_code == 204:
+            if requests.post(url, headers=headers, json=payload_gemma).status_code == 204:
                 waited = 0
                 while waited < 90:
                     if os.path.exists(gemma_txt_path):
-                        with open(gemma_txt_path, "r", encoding="utf-8") as f:
-                            english_prompt = f.read().strip()
+                        with open(gemma_txt_path, "r", encoding="utf-8") as f: english_prompt = f.read().strip()
                         gemma_success = True
                         break
-                    time.sleep(3)
-                    waited += 3
+                    time.sleep(3); waited += 3
             
             if gemma_success:
-                if "```" in english_prompt:
-                    english_prompt = english_prompt.replace("```text", "").replace("```", "").replace("```json", "").strip()
-
-                if "پردازش متوقف شد" in english_prompt or "سهمیه سرور موقتاً پر شده" in english_prompt:
-                    print(f"Gemma returned quota error on attempt {attempt + 1}. Retrying...")
-                    gemma_success = False
-                else:
-                    break 
-                
-        except Exception as e:
-            print(f"Gemma Action Error on attempt {attempt + 1}: {e}")
-        
-        if not gemma_success:
-            time.sleep(4) 
+                english_prompt = english_prompt.replace("```text", "").replace("```", "").replace("```json", "").strip()
+                if "پردازش متوقف شد" in english_prompt: gemma_success = False
+                else: break 
+        except: pass
+        if not gemma_success: time.sleep(4) 
 
     if not gemma_success:
-        print("Gemma action failed or returned errors. Using fallback translation.")
-        try: 
-            if user_prompt:
-                english_prompt = GoogleTranslator(source='auto', target='en').translate(user_prompt)
-                english_prompt += ", cinematic motion, photorealistic, high detail, smooth animation, 8k"
-            else:
-                english_prompt = "cinematic motion, photorealistic, high detail, smooth animation, 8k"
-        except: 
-            english_prompt = "cinematic motion, photorealistic, high detail, smooth animation, 8k"
-
-    payload_video = {
-        "event_type": "generate-video",
-        "client_payload": {
-            "prompt": english_prompt,
-            "duration": duration,
-            "image_url": image_public_url,
-            "run_id": run_id,
-            "space_url": SPACE_URL
-        }
-    }
+        try: english_prompt = GoogleTranslator(source='auto', target='en').translate(user_prompt) + ", cinematic motion, 8k" if user_prompt else "cinematic motion, photorealistic, high detail, smooth animation, 8k"
+        except: english_prompt = "cinematic motion, photorealistic, high detail, smooth animation, 8k"
+
+    payload_video = {"event_type": "generate-video", "client_payload": {"prompt": english_prompt, "duration": duration, "image_url": image_public_url, "run_id": run_id, "space_url": SPACE_URL}}
     r_vid = requests.post(url, headers=headers, json=payload_video)
     
-    if r_vid.status_code == 204:
-        return jsonify({"status": "success", "run_id": run_id, "translated": english_prompt})
-    else:
-        return jsonify({"status": "error", "message": r_vid.text}), 400
+    if r_vid.status_code == 204: return jsonify({"status": "success", "run_id": run_id, "translated": english_prompt})
+    else: return jsonify({"status": "error", "message": r_vid.text}), 400
+
+# ==========================================
+# سیستم چت بات همراه با استریم و دیباگر هوشمند
+# ==========================================
 
-# --- API مخصوص چت با Gemma-4 ---
 @app.route('/api/chat', methods=['POST'])
 def chat_api():
-    if not GITHUB_TOKEN: return jsonify({"status": "error", "message": "Token missing"}), 500
-
+    if not GITHUB_TOKEN: return jsonify({"status": "error", "message": "⚠️ توکن گیت‌هاب در اسپیس ست نشده است!"}), 500
     text_prompt = request.form.get('prompt', '')
-    
-    if has_forbidden_words(text_prompt):
-        return jsonify({"status": "error", "message": "متن ورودی شامل کلمات نامناسب است."}), 400
+    if has_forbidden_words(text_prompt): return jsonify({"status": "error", "message": "متن ورودی شامل کلمات نامناسب است."}), 400
 
     file_obj = request.files.get('file')
     run_id = str(uuid.uuid4())
@@ -329,27 +243,131 @@ def chat_api():
         file_obj.save(f"static/images/{filename}")
         file_url = f"{SPACE_URL}/static/images/{filename}"
 
-    url = f"https://api.github.com/repos/{GITHUB_USER}/{GITHUB_REPO}/dispatches"
-    headers = {"Accept": "application/vnd.github.v3+json", "Authorization": f"token {GITHUB_TOKEN}"}
-    
-    payload = {
-        "event_type": "chat-gemma",
-        "client_payload": {
-            "prompt": text_prompt,
-            "file_url": file_url,
-            "file_mime": mime_type,
-            "run_id": run_id,
-            "space_url": SPACE_URL
-        }
-    }
-    r = requests.post(url, headers=headers, json=payload)
-    
-    if r.status_code == 204:
+    try:
+        conn = get_chat_queue_db_connection()
+        c = conn.cursor()
+        c.execute("INSERT INTO queue (run_id, prompt, file_url, file_mime, status, timestamp) VALUES (?, ?, ?, ?, 'pending', ?)",
+                  (run_id, text_prompt, file_url, mime_type, time.time()))
+        conn.commit()
+        conn.close()
+        
+        headers = {"Accept": "application/vnd.github.v3+json", "Authorization": f"token {GITHUB_TOKEN}"}
+        payload = {"event_type": "chat-worker-pool", "client_payload": {"space_url": SPACE_URL}}
+        requests.post(f"https://api.github.com/repos/{GITHUB_USER}/{GITHUB_REPO}/dispatches", headers=headers, json=payload)
+
         return jsonify({"status": "success", "run_id": run_id})
-    else:
-        return jsonify({"status": "error", "message": r.text}), 400
+    except Exception as e:
+        print(f"Chat DB Error: {e}")
+        return jsonify({"status": "error", "message": f"⚠️ خطای دیتابیس در ثبت پیام: {str(e)}"}), 500
+
+@app.route('/api/worker/poll', methods=['POST'])
+def worker_poll():
+    start_time = time.time()
+    while time.time() - start_time < 25:
+        try:
+            conn = get_chat_queue_db_connection()
+            conn.row_factory = sqlite3.Row
+            c = conn.cursor()
+            
+            c.execute("SELECT run_id FROM queue WHERE status='pending' ORDER BY timestamp ASC LIMIT 1")
+            row = c.fetchone()
+            
+            if row:
+                r_id = row['run_id']
+                c.execute("UPDATE queue SET status='processing' WHERE run_id=? AND status='pending'", (r_id,))
+                if c.rowcount > 0:
+                    c.execute("SELECT * FROM queue WHERE run_id=?", (r_id,))
+                    job = c.fetchone()
+                    conn.commit()
+                    conn.close()
+                    return jsonify({"status": "job", "job": dict(job)})
+            conn.commit()
+            conn.close()
+        except Exception as e:
+            print(f"Poll DB error: {e}") # پرینت در لاگ اسپیس برای دیباگ
+            
+        time.sleep(2)
+    return jsonify({"status": "empty"})
+
+@app.route('/api/webhook/stream', methods=['POST'])
+def webhook_stream():
+    run_id = request.form.get('run_id')
+    text = request.form.get('text', '')
+    if run_id:
+        try:
+            with open(f"static/images/{run_id}.stream.txt", "w", encoding="utf-8") as f:
+                f.write(text)
+        except Exception as e: 
+            print(f"Webhook write error: {e}")
+    return "OK", 200
+
+@app.route('/api/stream_chat/<run_id>')
+def stream_chat(run_id):
+    def generate():
+        stream_file = f"static/images/{run_id}.stream.txt"
+        final_file = f"static/images/{run_id}.txt"
+        last_text = ""
+        
+        # 60 ثانیه انتظار - اگر اکشن گیت هاب کار نکند این ارور در صفحه چت نوشته میشود
+        for _ in range(300):
+            if os.path.exists(stream_file) or os.path.exists(final_file):
+                break
+            time.sleep(0.2)
+        else:
+            error_msg = "⚠️ **خطای سیستم:** اکشن گیت‌هاب روشن نشد یا نتوانست در ۶۰ ثانیه پیام را پردازش کند. لطفاً بخش لاگ‌های Actions در گیت‌هاب را بررسی کنید."
+            yield f"data: {json.dumps({'text': error_msg})}\n\n"
+            yield f"data: DONE\n\n"
+            return
+            
+        while True:
+            if os.path.exists(stream_file):
+                try:
+                    with open(stream_file, "r", encoding="utf-8") as f: current_text = f.read()
+                    if current_text and current_text != last_text:
+                        yield f"data: {json.dumps({'text': current_text})}\n\n"
+                        last_text = current_text
+                except: pass
+                
+            if os.path.exists(final_file):
+                try:
+                    with open(final_file, "r", encoding="utf-8") as f: final_text = f.read()
+                    if final_text and final_text != last_text:
+                        yield f"data: {json.dumps({'text': final_text})}\n\n"
+                except: pass
+                yield f"data: DONE\n\n"
+                break
+            time.sleep(0.15)
+
+    return Response(generate(), mimetype='text/event-stream', headers={'Cache-Control': 'no-cache', 'X-Accel-Buffering': 'no', 'Connection': 'keep-alive'})
+
+def maintain_worker_pool():
+    while True:
+        time.sleep(40) 
+        if not GITHUB_TOKEN or not SPACE_URL: continue
+        try:
+            url = f"https://api.github.com/repos/{GITHUB_USER}/{GITHUB_REPO}/actions/runs?event=repository_dispatch"
+            headers = {"Accept": "application/vnd.github.v3+json", "Authorization": f"token {GITHUB_TOKEN}"}
+            r = requests.get(url, headers=headers)
+            
+            if r.status_code != 200:
+                payload = {"event_type": "chat-worker-pool", "client_payload": {"space_url": SPACE_URL}}
+                requests.post(f"https://api.github.com/repos/{GITHUB_USER}/{GITHUB_REPO}/dispatches", headers=headers, json=payload)
+                time.sleep(120)
+                continue
+                
+            runs = r.json().get('workflow_runs', [])
+            active_count = sum(1 for run in runs if run.get('name') == 'Gemma 4 Chatbot Worker' and run.get('status') in ['in_progress', 'queued'])
+            
+            if active_count < 3: 
+                dispatch_url = f"https://api.github.com/repos/{GITHUB_USER}/{GITHUB_REPO}/dispatches"
+                payload = {"event_type": "chat-worker-pool", "client_payload": {"space_url": SPACE_URL}}
+                for _ in range(3 - active_count):
+                    requests.post(dispatch_url, headers=headers, json=payload)
+                    time.sleep(1)
+        except: pass
+
+# ==========================================
 
-# --- API میکس ویدیوها سمت سرور ---
 @app.route('/api/merge-videos', methods=['POST'])
 def merge_videos():
     if 'base_video' not in request.files or 'new_clip' not in request.files:
@@ -357,7 +375,6 @@ def merge_videos():
 
     base_video_file = request.files['base_video']
     new_clip_file = request.files['new_clip']
-    
     temp_files_to_clean = []
 
     try:
@@ -366,96 +383,66 @@ def merge_videos():
         output_path = os.path.join(TEMP_DIR, f"{uuid.uuid4().hex}.mp4")
 
         temp_files_to_clean.extend([base_video_path, new_clip_path, output_path])
-
         base_video_file.save(base_video_path)
         new_clip_file.save(new_clip_path)
 
         probe = ffmpeg.probe(base_video_path)
         video_stream = next((stream for stream in probe['streams'] if stream['codec_type'] == 'video'), None)
-        width = int(video_stream['width'])
-        height = int(video_stream['height'])
+        width, height = int(video_stream['width']), int(video_stream['height'])
 
         input1 = ffmpeg.input(base_video_path)
         input2 = ffmpeg.input(new_clip_path).filter('scale', width, height).filter('setsar', '1')
-
-        merged_video = ffmpeg.concat(input1, input2, v=1, a=0).output(
-            output_path,
-            crf=18,
-            preset='slow',
-            pix_fmt='yuv420p'
-        )
-        
+        merged_video = ffmpeg.concat(input1, input2, v=1, a=0).output(output_path, crf=18, preset='slow', pix_fmt='yuv420p')
         merged_video.run(overwrite_output=True, quiet=True)
         
         @after_this_request
         def cleanup(response):
             try:
                 for f_path in temp_files_to_clean:
-                    if os.path.exists(f_path):
-                        os.remove(f_path)
-            except:
-                pass
+                    if os.path.exists(f_path): os.remove(f_path)
+            except: pass
             return response
-
         return send_file(output_path, mimetype='video/mp4', as_attachment=True, download_name='merged_video.mp4')
 
-    except ffmpeg.Error as e:
+    except Exception:
         for f_path in temp_files_to_clean:
-            if os.path.exists(f_path):
-                os.remove(f_path)
-        return jsonify({"error": "خطا در پردازش و میکس ویدیو در سرور."}), 500
-    except Exception as e:
-        for f_path in temp_files_to_clean:
-            if os.path.exists(f_path):
-                os.remove(f_path)
-        return jsonify({"error": "خطای پیش‌بینی نشده در سرور."}), 500
+            if os.path.exists(f_path): os.remove(f_path)
+        return jsonify({"error": "خطا در پردازش ویدیو"}), 500
 
-# --- مدیریت اعتبار برای ویدیو ---
 @app.route('/api/check-credit', methods=['POST'])
 def check_credit():
     data = request.get_json()
     if not data: return jsonify({"error": "Invalid request"}), 400
     user_id = get_user_identifier(data)
-    
     conn = get_db_connection()
     c = conn.cursor()
     c.execute("SELECT * FROM users WHERE id = ?", (user_id,))
     user_record = c.fetchone()
-    
     now = time.time()
     credits_remaining = USAGE_LIMIT
     limit_reached = False
     reset_timestamp = 0
-    
     if user_record:
         if user_record['week_start'] < (now - ONE_WEEK_SECONDS):
             c.execute("UPDATE users SET count = 0, week_start = ? WHERE id = ?", (now, user_id))
             conn.commit()
-            credits_remaining = USAGE_LIMIT
         else:
             credits_remaining = max(0, USAGE_LIMIT - user_record['count'])
             if credits_remaining == 0:
                 limit_reached = True
                 reset_timestamp = user_record['week_start'] + ONE_WEEK_SECONDS
-                
     conn.close()
-    return jsonify({
-        "credits_remaining": credits_remaining, 
-        "limit_reached": limit_reached, 
-        "reset_timestamp": reset_timestamp
-    })
+    return jsonify({"credits_remaining": credits_remaining, "limit_reached": limit_reached, "reset_timestamp": reset_timestamp})
 
 @app.route('/api/use-credit', methods=['POST'])
 def use_credit():
     data = request.get_json()
     if not data: return jsonify({"error": "Invalid request"}), 400
     user_id = get_user_identifier(data)
-    
     conn = get_db_connection()
     c = conn.cursor()
     c.execute("SELECT * FROM users WHERE id = ?", (user_id,))
     user_record = c.fetchone()
-    
     now = time.time()
     if user_record:
         if user_record['week_start'] < (now - ONE_WEEK_SECONDS):
@@ -463,67 +450,50 @@ def use_credit():
         else:
             if user_record['count'] >= USAGE_LIMIT:
                 conn.close()
-                reset_timestamp = user_record['week_start'] + ONE_WEEK_SECONDS
-                return jsonify({"status": "limit_reached", "credits_remaining": 0, "reset_timestamp": reset_timestamp}), 429
+                return jsonify({"status": "limit_reached", "credits_remaining": 0, "reset_timestamp": user_record['week_start'] + ONE_WEEK_SECONDS}), 429
             c.execute("UPDATE users SET count = count + 1 WHERE id = ?", (user_id,))
     else:
         c.execute("INSERT INTO users (id, count, week_start) VALUES (?, 1, ?)", (user_id, now))
-        
     conn.commit()
     c.execute("SELECT count FROM users WHERE id = ?", (user_id,))
-    new_count = c.fetchone()['count']
-    credits_remaining = USAGE_LIMIT - new_count
-    
+    credits_remaining = USAGE_LIMIT - c.fetchone()['count']
     conn.close()
     return jsonify({"status": "success", "credits_remaining": credits_remaining})
 
-
-# --- مدیریت اعتبار اختصاصی روزانه برای صفحه جدید ساخت تصویر ---
 @app.route('/api/check-image-credit', methods=['POST'])
 def check_image_credit():
     data = request.get_json()
     if not data: return jsonify({"error": "Invalid request"}), 400
     user_id = get_user_identifier(data)
-    
     conn = get_image_db_connection()
     c = conn.cursor()
     c.execute("SELECT * FROM users WHERE id = ?", (user_id,))
     user_record = c.fetchone()
-    
     now = time.time()
     credits_remaining = IMAGE_USAGE_LIMIT
     limit_reached = False
     reset_timestamp = 0
-    
     if user_record:
         if user_record['week_start'] < (now - ONE_DAY_SECONDS):
             c.execute("UPDATE users SET count = 0, week_start = ? WHERE id = ?", (now, user_id))
             conn.commit()
-            credits_remaining = IMAGE_USAGE_LIMIT
         else:
             credits_remaining = max(0, IMAGE_USAGE_LIMIT - user_record['count'])
             if credits_remaining == 0:
                 limit_reached = True
                 reset_timestamp = user_record['week_start'] + ONE_DAY_SECONDS
-                
     conn.close()
-    return jsonify({
-        "credits_remaining": credits_remaining, 
-        "limit_reached": limit_reached, 
-        "reset_timestamp": reset_timestamp
-    })
+    return jsonify({"credits_remaining": credits_remaining, "limit_reached": limit_reached, "reset_timestamp": reset_timestamp})
 
 @app.route('/api/use-image-credit', methods=['POST'])
 def use_image_credit():
     data = request.get_json()
     if not data: return jsonify({"error": "Invalid request"}), 400
     user_id = get_user_identifier(data)
-    
     conn = get_image_db_connection()
     c = conn.cursor()
     c.execute("SELECT * FROM users WHERE id = ?", (user_id,))
     user_record = c.fetchone()
-    
     now = time.time()
     if user_record:
         if user_record['week_start'] < (now - ONE_DAY_SECONDS):
@@ -531,67 +501,50 @@ def use_image_credit():
         else:
             if user_record['count'] >= IMAGE_USAGE_LIMIT:
                 conn.close()
-                reset_timestamp = user_record['week_start'] + ONE_DAY_SECONDS
-                return jsonify({"status": "limit_reached", "credits_remaining": 0, "reset_timestamp": reset_timestamp}), 429
+                return jsonify({"status": "limit_reached", "credits_remaining": 0, "reset_timestamp": user_record['week_start'] + ONE_DAY_SECONDS}), 429
             c.execute("UPDATE users SET count = count + 1 WHERE id = ?", (user_id,))
     else:
         c.execute("INSERT INTO users (id, count, week_start) VALUES (?, 1, ?)", (user_id, now))
-        
     conn.commit()
     c.execute("SELECT count FROM users WHERE id = ?", (user_id,))
-    new_count = c.fetchone()['count']
-    credits_remaining = IMAGE_USAGE_LIMIT - new_count
-    
+    credits_remaining = IMAGE_USAGE_LIMIT - c.fetchone()['count']
     conn.close()
     return jsonify({"status": "success", "credits_remaining": credits_remaining})
 
-
-# --- مدیریت اعتبار اختصاصی هفتگی برای ویرایش تصویر ---
 @app.route('/api/check-edit-credit', methods=['POST'])
 def check_edit_credit():
     data = request.get_json()
     if not data: return jsonify({"error": "Invalid request"}), 400
     user_id = get_user_identifier(data)
-    
     conn = get_image_edit_db_connection()
     c = conn.cursor()
     c.execute("SELECT * FROM users WHERE id = ?", (user_id,))
     user_record = c.fetchone()
-    
     now = time.time()
     credits_remaining = EDIT_USAGE_LIMIT
     limit_reached = False
     reset_timestamp = 0
-    
     if user_record:
         if user_record['week_start'] < (now - ONE_WEEK_SECONDS):
             c.execute("UPDATE users SET count = 0, week_start = ? WHERE id = ?", (now, user_id))
             conn.commit()
-            credits_remaining = EDIT_USAGE_LIMIT
         else:
             credits_remaining = max(0, EDIT_USAGE_LIMIT - user_record['count'])
             if credits_remaining == 0:
                 limit_reached = True
                 reset_timestamp = user_record['week_start'] + ONE_WEEK_SECONDS
-                
     conn.close()
-    return jsonify({
-        "credits_remaining": credits_remaining, 
-        "limit_reached": limit_reached, 
-        "reset_timestamp": reset_timestamp
-    })
+    return jsonify({"credits_remaining": credits_remaining, "limit_reached": limit_reached, "reset_timestamp": reset_timestamp})
 
 @app.route('/api/use-edit-credit', methods=['POST'])
 def use_edit_credit():
     data = request.get_json()
     if not data: return jsonify({"error": "Invalid request"}), 400
     user_id = get_user_identifier(data)
-    
     conn = get_image_edit_db_connection()
     c = conn.cursor()
     c.execute("SELECT * FROM users WHERE id = ?", (user_id,))
     user_record = c.fetchone()
-    
     now = time.time()
     if user_record:
         if user_record['week_start'] < (now - ONE_WEEK_SECONDS):
@@ -599,57 +552,36 @@ def use_edit_credit():
         else:
             if user_record['count'] >= EDIT_USAGE_LIMIT:
                 conn.close()
-                reset_timestamp = user_record['week_start'] + ONE_WEEK_SECONDS
-                return jsonify({"status": "limit_reached", "credits_remaining": 0, "reset_timestamp": reset_timestamp}), 429
+                return jsonify({"status": "limit_reached", "credits_remaining": 0, "reset_timestamp": user_record['week_start'] + ONE_WEEK_SECONDS}), 429
             c.execute("UPDATE users SET count = count + 1 WHERE id = ?", (user_id,))
     else:
         c.execute("INSERT INTO users (id, count, week_start) VALUES (?, 1, ?)", (user_id, now))
-        
     conn.commit()
     c.execute("SELECT count FROM users WHERE id = ?", (user_id,))
-    new_count = c.fetchone()['count']
-    credits_remaining = EDIT_USAGE_LIMIT - new_count
-    
+    credits_remaining = EDIT_USAGE_LIMIT - c.fetchone()['count']
     conn.close()
     return jsonify({"status": "success", "credits_remaining": credits_remaining})
 
-
-# --- دریافت و ارسال فایل ---
 @app.route('/api/webhook/upload', methods=['POST'])
 def webhook_upload():
     run_id = request.form.get('run_id')
     gh_run_id = request.form.get('github_run_id')
     ext = request.form.get('ext', 'webp')
-    
-    if 'file' not in request.files or not run_id:
-        return "Invalid request", 400
-        
+    if 'file' not in request.files or not run_id: return "Invalid request", 400
     file = request.files['file']
     file.save(f"static/images/{run_id}.{ext}")
-    
     if gh_run_id: threading.Thread(target=delete_github_run, args=(gh_run_id,)).start()
     return "OK", 200
 
 @app.route('/api/status/<run_id>')
 def check_status(run_id):
-    # لیست تمام فرمت‌های مجاز و رایج که پایتون باید بررسی کند
-    valid_extensions = [
-        'png', 'jpg', 'jpeg', 'webp', 'gif',  # فرمت‌های عکس
-        'mp4', 'webm', 'mov', 'mkv', 'avi',   # فرمت‌های ویدیو
-        'mp3', 'wav', 'ogg', 'm4a',           # فرمت‌های صدا
-        'txt', 'json'                         # فرمت‌های متن
-    ]
-    
-    # بررسی می‌کند که آیا فایلی با این run_id و هر کدام از پسوندهای بالا وجود دارد یا خیر
-    for ext in valid_extensions:
-        if os.path.exists(f"static/images/{run_id}.{ext}"):
-            return jsonify({"status": "ready", "url": f"/static/images/{run_id}.{ext}"})
-            
+    for ext in ['png', 'jpg', 'jpeg', 'webp', 'gif', 'mp4', 'webm', 'mov', 'mkv', 'avi', 'mp3', 'wav', 'ogg', 'm4a', 'txt', 'json']:
+        if os.path.exists(f"static/images/{run_id}.{ext}"): return jsonify({"status": "ready", "url": f"/static/images/{run_id}.{ext}"})
     return jsonify({"status": "processing"})
 
 @app.route('/static/images/<filename>')
-def serve_file(filename):
-    return send_from_directory('static/images', filename)
+def serve_file(filename): return send_from_directory('static/images', filename)
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860)
+    threading.Thread(target=maintain_worker_pool, daemon=True).start()
+    app.run(host='0.0.0.0', port=7860)