Use lock

processing.py

@@ -13,6 +13,7 @@ import requests
 from pathlib import Path
 from screenshot import get_screen_shot
 import boto3
+from threading import Lock
 
 
 class SpaceData(TypedDict):
@@ -39,6 +40,7 @@ s3_client = boto3.client(
     aws_access_key_id=os.getenv("AWS_ACCESS_KEY"),
     aws_secret_access_key=os.getenv("AWS_SECRET_ACCESS_KEY"),
 )
+lock = Lock()
 
 
 api = huggingface_hub.HfApi(token=os.getenv("HF_TOKEN"))
@@ -66,12 +68,13 @@ def get_info(space_name: SpaceInfo) -> SpaceData | None:
     light_file = str(screen_shot_dir / Path(img_id + "_light.jpg"))
     dark_file = str(screen_shot_dir / Path(img_id + "_dark.jpg"))
     if status == "RUNNING":
-        get_screen_shot(
-            f"https://{space_info.subdomain}.hf.space?__theme=light", 10, light_file
-        )
-        get_screen_shot(
-            f"https://{space_info.subdomain}.hf.space?__theme=dark", 10, dark_file
-        )
+        with lock:
+            get_screen_shot(
+                f"https://{space_info.subdomain}.hf.space?__theme=light", 10, light_file
+            )
+            get_screen_shot(
+                f"https://{space_info.subdomain}.hf.space?__theme=dark", 10, dark_file
+            )
         s3_client.upload_file(
             light_file,
             "gradio-theme-screenshots",