Update app.py

app.py

@@ -4,6 +4,7 @@ import subprocess
 import os
 import uuid
 import logging
+import json
 
 # تنظیم لاگ برای دیباگ
 logging.basicConfig(level=logging.INFO)
@@ -11,6 +12,24 @@ logger = logging.getLogger(__name__)
 
 app = FastAPI()
 
+def get_video_resolution(input_path: str) -> tuple[int, int]:
+    """دریافت رزولوشن ویدیوی ورودی با استفاده از ffprobe"""
+    try:
+        command = [
+            "ffprobe", "-v", "error", "-select_streams", "v:0",
+            "-show_entries", "stream=width,height", "-of", "json",
+            input_path
+        ]
+        result = subprocess.run(command, capture_output=True, text=True, check=True)
+        data = json.loads(result.stdout)
+        width = data["streams"][0]["width"]
+        height = data["streams"][0]["height"]
+        logger.info(f"Input resolution: {width}x{height}")
+        return width, height
+    except subprocess.CalledProcessError as e:
+        logger.error(f"ffprobe error: {e.stderr}")
+        raise ValueError("خطا در دریافت رزولوشن ویدیو")
+
 def compress_video(input_path: str, crf: int, resolution: str) -> str:
     logger.info(f"Starting video compression for {input_path}")
     try:
@@ -21,17 +40,30 @@ def compress_video(input_path: str, crf: int, resolution: str) -> str:
         # تولید نام یکتا برای فایل خروجی
         output_path = f"compressed_{uuid.uuid4().hex}.mp4"
 
-        # تنظیم رزولوشن
+        # دریافت رزولوشن ورودی
+        input_width, input_height = get_video_resolution(input_path)
+
+        # تنظیم رزولوشن خروجی
         scale_filter = None
-        if resolution != "original":
-            scale_filter = {
-                "720p": "1280:720",
-                "480p": "854:480",
-                "360p": "640:360",
-                "240p": "426:240"
-            }.get(resolution)
-            if not scale_filter:
-                raise ValueError("رزولوشن نامعتبر!")
+        target_resolutions = {
+            "720p": (1280, 720),
+            "480p": (854, 480),
+            "360p": (640, 360),
+            "240p": (426, 240),
+            "original": (input_width, input_height)
+        }
+
+        if resolution not in target_resolutions:
+            raise ValueError("رزولوشن نامعتبر!")
+
+        target_width, target_height = target_resolutions[resolution]
+
+        # محدود کردن رزولوشن خروجی به حداکثر رزولوشن ورودی
+        if target_width > input_width or target_height > input_height:
+            logger.info(f"Target resolution {target_width}x{target_height} is larger than input {input_width}x{input_height}, using input resolution")
+            target_width, target_height = input_width, input_height
+
+        scale_filter = f"scale={target_width}:{target_height}:force_original_aspect_ratio=decrease"
 
         # ساخت دستور FFmpeg
         command = [
@@ -41,7 +73,7 @@ def compress_video(input_path: str, crf: int, resolution: str) -> str:
         ]
 
         if scale_filter:
-            command.extend(["-vf", f"scale={scale_filter}"])
+            command.extend(["-vf", scale_filter])
 
         command.append(output_path)
 
@@ -100,7 +132,6 @@ async def compress_video_api(
         # فقط فایل ورودی رو حذف کن
         if input_path and os.path.exists(input_path):
             os.remove(input_path)
-        # فایل خروجی رو بعد از ارسال نگه دار (یا بعداً با یه مکانیزم پاک‌سازی حذف کن)
 
 if __name__ == "__main__":
     import uvicorn