能否将上面这段代码分客户端与服务端来优化？

当然可以。将代码分成客户端和服务端可以使代码更加模块化和易于维护，并且可以使不同的开发者分别负责不同的部分，提高开发效率。

下面是一个简单的客户端-服务端架构的示例：

客户端代码：

import socket

HOST = '127.0.0.1'  # 服务端的 IP 地址
PORT = 8888        # 服务端的端口号

with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
    s.connect((HOST, PORT))
    while True:
        user_input = input("请输入你的问题：")
        s.sendall(user_input.encode())
        data = s.recv(1024)
        print(f"AI 回答：{data.decode()}")

服务端代码：

import socket

HOST = '127.0.0.1'  # 服务端的 IP 地址
PORT = 8888        # 服务端的端口号

def generate_answer(question):
    # 这里写生成 AI 回答的代码
    return "这是 AI 的回答"

with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
    s.bind((HOST, PORT))
    s.listen()
    conn, addr = s.accept()
    with conn:
        print('连接来自：', addr)
        while True:
            data = conn.recv(1024)
            if not data:
                break
            question = data.decode()
            answer = generate_answer(question)
            conn.sendall(answer.encode())

这里的客户端代码主要负责与用户交互，将用户输入的问题发送给服务端，然后等待服务端的回复并输出到控制台。

服务端代码则主要负责接收客户端发送的数据，生成 AI 的回答，并将回答发送给客户端。可以看到，服务端代码中的 `generate_answer` 函数是用来生成 AI 回答的，可以根据具体情况进行修改。