添加自定义端口和健康检测

main.py

@@ -4,6 +4,7 @@ from flask import Flask, request, jsonify
 import threading
 import queue
 import time
+import argparse
 from ffmpeg_utils import process_ffmpeg
 from sanjuuni_utils import process_sanjuuni
 from file_cleanup import start_cleanup_thread
@@ -122,7 +123,22 @@ def download_file_endpoint(file_id, filename):
         logging.error(f"下载文件时出错: {e}")
         return jsonify({'status': 'error', 'error': str(e)}), 500
 
+
+
+
+
+@app.route('/health', methods=['GET'])
+def health_check():
+    """健康检查接口，直接返回 'ok'"""
+    return "OK"
+
+
 if __name__ == '__main__':
-    logging.basicConfig(level=logging.INFO)
+    # 配置命令行参数解析
+    parser = argparse.ArgumentParser(description='启动FFmpeg处理服务')
+    parser.add_argument('--port', type=int, default=5000, help='服务监听的端口号（默认: 5000）')
+    args = parser.parse_args()
+
+    logging.info(f"启动应用程序，端口: {args.port}...")
     start_cleanup_thread()
-    app.run(debug=True)
+    app.run(host='0.0.0.0', port=args.port, threaded=True)