Update app.py

app.py

@@ -4,47 +4,41 @@ import subprocess
 import io
 import os
 from typing import Optional
+from urllib.parse import urlparse
 
-app = FastAPI(title="FFmpeg API for n8n")
-
-@app.get("/")
-async def health_check():
-    return {"status": "online", "service": "FFmpeg Image Processor"}
+app = FastAPI()
 
 @app.post("/process")
 async def process_image(
     file: Optional[UploadFile] = File(None),
     url: Optional[str] = Form(None),
-    options: Optional[str] = Form("-vf scale=800:-1") # 默认缩小到800宽
+    options: Optional[str] = Form("-q:v 2"),
+    filename: Optional[str] = Form(None) # 新增：支持自定义文件名
 ):
-    """
-    支持两种输入方式：
-    1. file: 上传二进制图片文件
-    2. url: 远程图片地址
-    """
     input_source = ""
     input_data = None
 
+    # 1. 确定输入源
     if file:
         input_data = await file.read()
         input_source = "pipe:0"
+        # 如果没传 filename，则使用上传文件的原始名，但后缀改为 .jpg
+        if not filename:
+            filename = os.path.splitext(file.filename)[0] + ".jpg"
     elif url:
         input_source = url
+        # 如果没传 filename，从 URL 中解析文件名，后缀改为 .jpg
+        if not filename:
+            path = urlparse(url).path
+            base_name = os.path.basename(path) or "downloaded_image"
+            filename = os.path.splitext(base_name)[0] + ".jpg"
     else:
-        raise HTTPException(status_code=400, detail="需提供 file 或 url 参数")
+        raise HTTPException(status_code=400, detail="Missing source")
 
-    # 构建命令
-    # 默认输出为 .jpg 格式，你可以根据需要调整
-    command = [
-        'ffmpeg',
-        '-i', input_source,
-    ]
-    
-    # 解析 options 参数并加入命令
+    # 2. 执行 FFmpeg 转换 (PNG -> JPG)
+    command = ['ffmpeg', '-i', input_source]
     if options:
         command.extend(options.split())
-    
-    # 强制输出为图像流
     command.extend(['-f', 'image2', 'pipe:1'])
 
     try:
@@ -54,13 +48,19 @@ async def process_image(
             stdout=subprocess.PIPE,
             stderr=subprocess.PIPE
         )
-        
         out, err = process.communicate(input=input_data)
 
-        if process.returncode != 0:
-            return {"error": err.decode()}
-
-        return StreamingResponse(io.BytesIO(out), media_type="image/jpeg")
+        # 3. 构建响应，添加 Content-Disposition Header
+        # 这会让 n8n 和浏览器识别出文件名
+        headers = {
+            'Content-Disposition': f'attachment; filename="{filename}"'
+        }
+        
+        return StreamingResponse(
+            io.BytesIO(out), 
+            media_type="image/jpeg", 
+            headers=headers
+        )
 
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))