rate limit

app.py

@@ -102,21 +102,17 @@ def reset_user_input():
     return gr.update(value='')
 
 
-def rate_cnt():
-    global host_cnt
-    print('Resetting count')
-    host_cnt = {}
-
 def reset_state(chatbot, messages, answer_output, answerBtn, answer, idx, question, show_ans, request: gr.Request):
     global host_cnt
     host = request.client.host
+    
     if not host in host_cnt:
-        host_cnt[host] = 1
+        host_cnt[host] = time.time()
     else:
-        host_cnt[host] += 1
-    print(host_cnt[host])
-    if host_cnt[host] >= 10:
-        return chatbot, messages, answer_output, answerBtn, answer, idx, question, show_ans
+        elapse = time.time()-host_cnt[host]
+        if elapse < 10:
+            time.sleep(10-elapse)
+        host_cnt[host] = time.time()
     data = {'refresh': ''}
     data=requests.post(os.environ.get("URL"), data=json.dumps(data, ensure_ascii=False).encode('utf-8')).content
     data = json.loads(str(data, encoding="utf-8"))
@@ -137,7 +133,6 @@ def show_hide_answer(answer, show_ans):
 
 
 host_cnt = {}
-schedule.every(1).minutes.do(rate_cnt)
 
 with gr.Blocks() as demo:
     gr.HTML("""<h1 align="center">海龟汤</h1>""")