Create app.py

app.py

@@ -0,0 +1,48 @@
+import hashlib
+
+import gradio as gr
+import random
+import time
+
+def generate_user_id(request_headers):
+        # 提取所需的信息
+        ip_address = request_headers['host']
+        user_agent = request_headers['user-agent']
+        accept_language = request_headers['accept-language']
+        if 'cookie' in request_headers:
+            cookie = request_headers['cookie']
+        else:
+            cookie = 'empty'
+            print(f'cookie not in request_headers: {request_headers}')
+
+        # 将信息组合成一个字符串
+        combined_info = f"{ip_address}-{user_agent}-{accept_language}-{cookie}"
+        print('combined_info:', combined_info)
+
+        # 使用哈希函数生成唯一标识符
+        user_id = hashlib.sha256(combined_info.encode()).hexdigest()
+        print('user_id:', user_id)
+
+        return f"{combined_info}\n user_id:{user_id}"
+
+def page_load(request: gr.Request):
+    uuid = generate_user_id(request.headers)
+    print('[page_load] uuid:', uuid)
+
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox()
+    clear = gr.ClearButton([msg, chatbot])
+
+    def respond(message, chat_history, request: gr.Request):
+        uuid = generate_user_id(request_headers=request.headers)
+        bot_message = random.choice(["How are you?", "I love you", "I'm very hungry"])
+        chat_history.append((message, bot_message))
+        time.sleep(2)
+        return "", chat_history
+
+    msg.submit(respond, [msg, chatbot], [msg, chatbot])
+    demo.load(page_load, None, None)
+
+if __name__ == "__main__":
+    demo.launch()