Limit Storage of Pics

main.py

@@ -65,22 +65,26 @@ class Core(BaseModel):
 async def core():
     if not os.path.exists(pictures_directory):
         os.makedirs(pictures_directory)
+    files = sorted([f for f in os.listdir(pictures_directory) if f.endswith('.json')])
+    first_100_files = files[:100]
+    for file in files[100:]:
+        os.remove(os.path.join(pictures_directory, file))
+    
     async def generator():
         # Start JSON array
         yield '['
         first = True
-        for filename in os.listdir(pictures_directory):
-            if filename.endswith('.json'):
-                file_path = os.path.join(pictures_directory, filename)
-                with open(file_path, 'r') as file:
-                    data = json.load(file)
-                    
-                    # For JSON formatting, ensure only the first item doesn't have a preceding comma
-                    if first:
-                        first = False
-                    else:
-                        yield ','
-                    yield json.dumps({"base64": data["base64image"], "prompt": data["returnedPrompt"]})
+        for filename in first_100_files:
+            file_path = os.path.join(pictures_directory, filename)
+            with open(file_path, 'r') as file:
+                data = json.load(file)
+                
+                # For JSON formatting, ensure only the first item doesn't have a preceding comma
+                if first:
+                    first = False
+                else:
+                    yield ','
+                yield json.dumps({"base64": data["base64image"], "prompt": data["returnedPrompt"]})
         # End JSON array
         yield ']'