Update app.py

app.py

@@ -193,3 +193,74 @@ def create_image_ep(content: Text = None):
                            headers=headers
                            )
     return JSONResponse(content=result.json())
+
+
+
+
+from fastapi import FastAPI, Request, Response
+from fastapi.responses import PlainTextResponse
+from hashlib import sha1
+from time import time
+from xml.etree.ElementTree import Element, tostring
+
+
+
+
+def chat_gpt_response(prompt):
+    # Replace with your GPT-3.5 implementation
+    return "你好呀，小哥哥"
+
+
+@app.get('/wechat')
+def verify_server_address(signature: str, timestamp: str, nonce: str, echostr: str):
+    token = 'zsj'
+    if check_signature(token, signature, timestamp, nonce):
+        return PlainTextResponse(echostr)
+
+
+@app.post('/wechat')
+def process_message(request: Request):
+    xml_data = await request.body()
+    xml_tree = ElementTree.fromstring(xml_data)
+
+    msg_type = xml_tree.find('MsgType').text
+    if msg_type == 'text':
+        content = xml_tree.find('Content').text
+        user_open_id = xml_tree.find('FromUserName').text
+        public_account_id = xml_tree.find('ToUserName').text
+
+        reply_content = chat_gpt_response(content)
+
+        reply = Element('xml')
+        to_user_name = Element('ToUserName')
+        to_user_name.text = user_open_id
+        reply.append(to_user_name)
+
+        from_user_name = Element('FromUserName')
+        from_user_name.text = public_account_id
+        reply.append(from_user_name)
+
+        create_time = Element('CreateTime')
+        create_time.text = str(int(time()))
+        reply.append(create_time)
+
+        msg_type = Element('MsgType')
+        msg_type.text = 'text'
+        reply.append(msg_type)
+
+        content = Element('Content')
+        content.text = reply_content
+        reply.append(content)
+
+        response_xml = tostring(reply, encoding='utf-8')
+        return Response(content=response_xml, media_type='application/xml')
+
+
+def check_signature(token, signature, timestamp, nonce):
+    tmp_list = [token, timestamp, nonce]
+    tmp_list.sort()
+    tmp_str = ''.join(tmp_list)
+    tmp_str = sha1(tmp_str.encode('utf-8')).hexdigest()
+
+    return tmp_str == signature
+