如何修改我的代码，以便将Python终端上执行的输出提取到fasta文件中？

import sys

def write_to_fasta(output, filename):
    with open(filename, 'w') as file:
        file.write(output)

output = "这里是你的输出内容"

write_to_fasta(output, 'output.fasta')

import sys

def write_to_fasta(output, filename):
    with open(filename, 'w') as file:
        file.write(output)

# 修改你的代码，将输出保存到一个变量中
output = "这里是你的输出内容"

# 调用函数将输出写入fasta文件
write_to_fasta(output, 'output.fasta')