Screen streaming under docker environment (#674)

Co-authored-by: Shuchang Zheng <wintonzheng0325@gmail.com>

run_streaming.py

@@ -0,0 +1,38 @@
+import asyncio
+import subprocess
+
+import structlog
+import typer
+
+from skyvern.forge import app
+from skyvern.forge.sdk.settings_manager import SettingsManager
+
+INTERVAL = 1
+LOG = structlog.get_logger()
+
+
+async def run() -> None:
+    file_name = "skyvern_screenshot.png"
+    png_file_path = f"{SettingsManager.get_settings().STREAMING_FILE_BASE_PATH}/{file_name}"
+
+    while True:
+        # run subprocess to take screenshot
+        subprocess.run(
+            f"xwd -root | xwdtopnm 2>/dev/null | pnmtopng > {png_file_path}", shell=True, env={"DISPLAY": ":99"}
+        )
+
+        # upload screenshot to S3
+        try:
+            await app.STORAGE.save_streaming_file("placeholder_org", file_name)
+        except Exception:
+            LOG.info("Failed to save screenshot")
+
+        await asyncio.sleep(INTERVAL)
+
+
+def main() -> None:
+    asyncio.run(run())
+
+
+if __name__ == "__main__":
+    typer.run(main)