ip detect

app.py

@@ -5,6 +5,7 @@ import random as rd
 import os
 import json
 import time
+import schedule
 import openai
 import requests
 
@@ -101,7 +102,19 @@ def reset_user_input():
     return gr.update(value='')
 
 
-def reset_state():
+def rate_cnt():
+    global host_cnt
+    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
+    else:
+        host_cnt[host] += 1
+    if host_cnt[host] >= 10:
+        return chatbot, messages, answer_output, answerBtn, answer, idx, question, show_ans
     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"))
@@ -121,6 +134,8 @@ def show_hide_answer(answer, show_ans):
         return gr.update(value=answer), gr.update(value="Hide Answer"), show_ans
 
 
+host_cnt = {}
+
 with gr.Blocks() as demo:
     gr.HTML("""<h1 align="center">海龟汤</h1>""")
 
@@ -128,6 +143,7 @@ with gr.Blocks() as demo:
         rule = gr.Textbox(label='规则', value='海龟汤是一个推理类游戏，游戏开始时会给出一段隐去关键信息的叙述，即汤面，玩家根据汤面提出问题，问题需要能够通过“是”或“否”来回答。玩家通过提问不同可能性，缩小真相的范围，直到最终猜到真相（即汤底）的关键信息。玩家可以点击"New Game"按钮开始新的游戏，点击"Show Answer"可查看汤底。', 
                               lines=1, max_lines=3).style(container=False)
     data = {'refresh': ''}
+    schedule.every(1).minutes.do(rate_cnt)
     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"))
     chatbot = gr.Chatbot(data['chatbot'])
@@ -158,7 +174,7 @@ with gr.Blocks() as demo:
                     show_progress=True)
     submitBtn.click(reset_user_input, [], [user_input])
 
-    emptyBtn.click(reset_state, outputs=[chatbot, messages, answer_output, answerBtn, answer, idx, question, show_ans], show_progress=True)
+    emptyBtn.click(reset_state, [chatbot, messages, answer_output, answerBtn, answer, idx, question, show_ans], outputs=[chatbot, messages, answer_output, answerBtn, answer, idx, question, show_ans], show_progress=True)
 
     answerBtn.click(show_hide_answer, [answer, show_ans], outputs=[answer_output, answerBtn, show_ans])