import os
from auto_media_publisher.rewriter.docx_rewriter_gemini import GeminiDocxRewriter

def main():
   
    input_docx = "input.docx"
    output_docx = "rewritten_output.docx"

    rewriter = GeminiDocxRewriter(rewrite_style="random" )

    print("开始改写文档并生成新文件...")
    summary = rewriter.rewrite(input_docx, output_docx)

    print("\n=== 改写及摘要完成 ===")
    print(f"生成文件：{output_docx}")
    print("摘要信息：")
    print(f"标题：{summary.get('title', '')}")
    print(f"简介：{summary.get('summary', '')}")
    print(f"关键词：{summary.get('keywords', [])}")

if __name__ == "__main__":
    main()