add more

app.py

@@ -85,6 +85,8 @@ def infer(prompt: str, negative: str, width: int, height: int, sampler: str,
 
     count += 1
     if count > 5:
+        print(client_ip)
+        print(count)
         return None, "Too many requests, please try again later more."
     else:
         count_client_map.put(client_ip, count, 24 * 60 * 60)  # 添加一个过期时间为 24 小时的项

set.py

@@ -1,3 +1,4 @@
+import time
 from datetime import datetime, timedelta
 from collections import defaultdict
 
@@ -21,3 +22,12 @@ class ExpiringMap:
     def remove(self, key):
         if key in self.data:
             del self.data[key]
+
+
+if __name__ == "__main__":
+    time_client_map = ExpiringMap()
+    time_client_map.put('a', 'b', 10)
+    time.sleep(5)
+    print(time_client_map.get('a'))
+    time.sleep(6)
+    print(time_client_map.get('a'))