Update app.py

app.py

@@ -3,7 +3,10 @@ import uuid
 import requests
 import threading
 import time
-from flask import Flask, request, jsonify, render_template, send_from_directory
+import sqlite3
+import tempfile
+import ffmpeg
+from flask import Flask, request, jsonify, render_template, send_from_directory, send_file, after_this_request
 from deep_translator import GoogleTranslator
 
 app = Flask(__name__)
@@ -14,12 +17,46 @@ GITHUB_REPO = os.environ.get("GITHUB_REPO", "Action")
 SPACE_HOST = os.environ.get("SPACE_HOST", "")
 SPACE_URL = f"https://{SPACE_HOST}"
 
+# --- تنظیمات دیتابیس SQLite ---
+# در هاگینگ فیس باکت‌ها معمولاً در مسیر /data مونت می‌شوند
+DB_DIR = "/data" if os.path.exists("/data") else "."
+DB_PATH = os.path.join(DB_DIR, "users.db")
+TEMP_DIR = "tmp"
+USAGE_LIMIT = 5
+ONE_WEEK_SECONDS = 7 * 24 * 60 * 60
+
+os.makedirs(TEMP_DIR, 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)''')
+    conn.commit()
+    conn.close()
+
+init_db()
+
+def get_db_connection():
+    conn = sqlite3.connect(DB_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}"
     headers = {"Accept": "application/vnd.github.v3+json", "Authorization": f"token {GITHUB_TOKEN}"}
     requests.delete(url, headers=headers)
 
+def get_user_identifier(data):
+    fingerprint = data.get('fingerprint')
+    if fingerprint: return str(fingerprint)
+    if request.headers.getlist("X-Forwarded-For"):
+        return request.headers.getlist("X-Forwarded-For")[0].split(',')[0].strip()
+    return request.remote_addr
+
+# --- مسیرهای صفحات HTML ---
 @app.route('/')
 def index():
     return render_template('fluxpro.html')
@@ -109,7 +146,7 @@ def edit_image():
     else:
         return jsonify({"status": "error", "message": r.text}), 400
 
-# --- API جدید مخصوص تبدیل عکس به ویدیو (Image to Video) ---
+# --- API مخصوص تبدیل عکس به ویدیو ---
 @app.route('/api/generate-video', methods=['POST'])
 def generate_video():
     if not GITHUB_TOKEN: return jsonify({"status": "error", "message": "Token missing"}), 500
@@ -152,12 +189,142 @@ def generate_video():
     else:
         return jsonify({"status": "error", "message": r.text}), 400
 
-# وب‌هوک برای دریافت فایل نهایی از گیت‌هاب (با پشتیبانی از فرمت‌های مختلف)
+# --- 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:
+        return jsonify({"error": "هر دو فایل ویدیویی برای میکس لازم هستند."}), 400
+
+    base_video_file = request.files['base_video']
+    new_clip_file = request.files['new_clip']
+    
+    temp_files_to_clean = []
+
+    try:
+        base_video_path = os.path.join(TEMP_DIR, f"{uuid.uuid4().hex}.mp4")
+        new_clip_path = os.path.join(TEMP_DIR, f"{uuid.uuid4().hex}.mp4")
+        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'])
+
+        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.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 Exception as e:
+                pass
+            return response
+
+        return send_file(output_path, mimetype='video/mp4', as_attachment=True, download_name='merged_video.mp4')
+
+    except ffmpeg.Error as e:
+        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
+
+# --- مدیریت کاربران و محدودیت با SQLite ---
+@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
+    })
+
+@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):
+            c.execute("UPDATE users SET count = 1, week_start = ? WHERE id = ?", (now, user_id))
+        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
+            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
+    
+    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') # پسوند پیش‌فرض webp برای عکس و mp4 برای ویدیو
+    ext = request.form.get('ext', 'webp')
     
     if 'file' not in request.files or not run_id:
         return "Invalid request", 400
@@ -168,19 +335,15 @@ def webhook_upload():
     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):
-    # چک کردن فایل ویدیویی
     if os.path.exists(f"static/images/{run_id}.mp4"):
         return jsonify({"status": "ready", "url": f"/static/images/{run_id}.mp4"})
-    # چک کردن فایل تصویری
     if os.path.exists(f"static/images/{run_id}.webp"):
         return jsonify({"status": "ready", "url": f"/static/images/{run_id}.webp"})
         
     return jsonify({"status": "processing"})
 
-# دسترسی دادن به پوشه تصاویر و ویدیوها
 @app.route('/static/images/<filename>')
 def serve_file(filename):
     return send_from_directory('static/images', filename)