Update app.py

app.py

@@ -78,6 +78,7 @@ from diffusers import FluxPipeline
 # import urllib.parse
 import subprocess
 
+os.system("pip list")
 IS_ZERO_GPU = bool(os.getenv("SPACES_ZERO_GPU"))
 HIDE_API = bool(os.getenv("HIDE_API"))
 if IS_ZERO_GPU:
@@ -180,6 +181,10 @@ class GuiSD:
         self.active_downloads = set()
         self.download_lock = threading.Lock()
 
+        # Anti-abuse: Track new model requests
+        self.used_models = []
+        self.new_model_history = []
+
     def update_storage_models(self, storage_floor_gb=30, required_inventory_for_purge=3):
         while get_used_storage_gb() > storage_floor_gb:
             if len(self.inventory) < required_inventory_for_purge:
@@ -205,6 +210,33 @@ class GuiSD:
 
     def load_new_model(self, model_name, vae_model, task, controlnet_model, progress=gr.Progress(track_tqdm=True)):
 
+        # --- Anti-Abuse Check Start ---
+        if model_name in self.used_models:
+            # Move to the end to mark as the most recently used (prevents it from being forgotten)
+            self.used_models.remove(model_name)
+            self.used_models.append(model_name)
+        else:
+            current_time = datetime.now()
+            # Retain history of new model requests from the last 10 minutes (600 seconds)
+            self.new_model_history = [
+                t for t in self.new_model_history 
+                if (current_time - t).total_seconds() < 600
+            ]
+            
+            # Allow a maximum of 5 NEW model requests per 10 minutes
+            if len(self.new_model_history) >= 5:
+                yield "Rate limit exceeded: Too many new models requested."
+                raise gr.Error("Too many new models requested. Please reuse your previously loaded models or wait a few minutes before trying new ones.")
+            
+            # Log the new model attempt
+            self.new_model_history.append(current_time)
+            self.used_models.append(model_name)
+            
+            # Prevent the bypass list from growing infinitely (cap at 5 most recent models)
+            if len(self.used_models) > 5:
+                self.used_models.pop(0)
+        # --- Anti-Abuse Check End ---
+
         lock_key = model_name
 
         while True: