from http.server import BaseHTTPRequestHandler, HTTPServer
import json
import requests


class MyRequestHandler(BaseHTTPRequestHandler):

    def do_GET(self):
        # 处理GET请求
        self.send_response(200)
        self.send_header('Content-type', 'text/html')
        self.end_headers()
        self.wfile.write(b"Welcome to my web page!")

    def do_POST(self):
        # 处理POST请求
        content_len = int(self.headers['Content-Length'])
        post_data = self.rfile.read(content_len)
        request_data = json.loads(post_data.decode('utf-8'))

        ret = requests.post("https://api.openai.com/v1/chat/completions", data=json.dumps({
            "model": "gpt-3.5-turbo",
            "messages": [
                {"role": "user", "content": request_data['content']}
            ],
            "temperature": 0.9,
            "max_tokens": 1000,
            "top_p": 1,
            "frequency_penalty": 0.0,
            "presence_penalty": 0.6,
            "stop": ["\\n\\n\\n"]
        }), headers={
            "Authorization": "Bearer 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",
            "Content-Type": "application/json"
        })
        print(ret)

        # 设置响应的Content-type
        self.send_response(200)
        self.send_header('Content-type', 'application/json')
        self.end_headers()

        # 将处理后的数据返回给客户端
        # self.wfile.write(json.dumps(ret.content).encode('utf-8'))
        self.wfile.write(ret.content)

if __name__ == '__main__':
    server_address = ('', 8000)
    httpd = HTTPServer(server_address, MyRequestHandler)
    print(f"Starting the web server at {server_address[0]}:{server_address[1]}")
    httpd.serve_forever()