Update Fsg_pp.py

Fsg_pp.py

@@ -23,8 +23,30 @@ class ImageGallery:
         self.selected = 0
         self.search_counter = 0
 
+
     def return_images(self, image_locs):
         self.imgz = image_locs
+        total_images = len([name for name in os.listdir('./Images') if os.path.isfile(os.path.join('./Images', name))])
+        if total_images >= 20:
+            # Move images in self.imgz to a temporary directory
+            os.makedirs("./TempImages", exist_ok=True)
+            for img in self.imgz:
+                os.rename(img, os.path.join("./TempImages", os.path.basename(img)))
+    
+            # Delete and recreate the Images directory
+            os.system("rm -r ./Images")
+            os.makedirs("./Images")
+    
+            # Move images from the temporary directory back to the Images directory
+            for img in os.listdir('./TempImages'):
+                os.rename(os.path.join("./TempImages", img), os.path.join("./Images", img))
+    
+            # Delete the temporary directory
+            os.system("rm -r ./TempImages")
+    
+        # Log the return value
+        logging.info(self.imgz)
+        
         return self.imgz if self.imgz else []
 
 
@@ -78,15 +100,6 @@ def create_gallery_tab(tab_name, search_fn, search_inputs, gallery_instance, fn_
     with gr.Column():
         gallery=gr.Gallery(label="Image Preview", preview=True, object_fit="cover", container=True, columns=5)
 
-        total_images = len(glob.glob('./Images/*'))
-        if total_images >= 20:
-            os.system("rm -r ./Images")
-            os.makedirs("./Images")
-
-        # Log the return value
-        logging.info(gallery_instance.imgz)
-        
-
         with gr.Row():
             crop_btn = gr.Button(value="Crop Selected Image",variant='secondary')
             crop_btn.click(fn=fn_on_click, outputs=[image,tabs])