lzy/sci-gui-agent-benchmark
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix error in long video recording in server/main.py

Browse Source
This commit is contained in:
rhythmcao
2024-02-02 07:24:26 +08:00
parent e5dbf3b31f
commit 3184f091a8
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
desktop_env/server/main.py
View File

@@ -2,7 +2,7 @@ import ctypes
import os import os
import platform import platform
import shlex import shlex
import subprocess import subprocess, signal
from pathlib import Path from pathlib import Path
from typing import Any, Optional from typing import Any, Optional
from typing import List, Dict, Tuple from typing import List, Dict, Tuple
@@ -997,7 +997,7 @@ def start_recording():
start_command = f"ffmpeg -y -f x11grab -draw_mouse 1 -s {screen_width}x{screen_height} -i :0.0 -c:v libx264 -r 30 {recording_path}" start_command = f"ffmpeg -y -f x11grab -draw_mouse 1 -s {screen_width}x{screen_height} -i :0.0 -c:v libx264 -r 30 {recording_path}"
recording_process = subprocess.Popen(shlex.split(start_command), stdout=subprocess.PIPE, stderr=subprocess.PIPE) recording_process = subprocess.Popen(shlex.split(start_command), stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
return jsonify({'status': 'success', 'message': 'Started recording.'}) return jsonify({'status': 'success', 'message': 'Started recording.'})
@@ -1009,10 +1009,8 @@ def end_recording():
if not recording_process: if not recording_process:
return jsonify({'status': 'error', 'message': 'No recording in progress to stop.'}), 400 return jsonify({'status': 'error', 'message': 'No recording in progress to stop.'}), 400
recording_process.terminate() recording_process.send_signal(signal.SIGINT)
recording_process.wait() recording_process.wait()
# return_code = recording_process.returncode
output, error = recording_process.communicate()
recording_process = None recording_process = None
# return recording video file # return recording video file