import io import docx import configparser import pandas as pd import asyncio from docx import Document from docxtpl import DocxTemplate from docx.shared import Pt from docx.opc.constants import RELATIONSHIP_TYPE as RT from docx.enum.dml import MSO_THEME_COLOR_INDEX from docx.enum.style import WD_STYLE_TYPE from docx.shared import Cm, Inches from docx.oxml.shared import OxmlElement from docx.enum.text import WD_ALIGN_PARAGRAPH from docx.enum.text import WD_PARAGRAPH_ALIGNMENT from docx.oxml.ns import qn from docx.shared import RGBColor from docx.enum.text import WD_COLOR_INDEX from requests_toolbelt import MultipartEncoder from datetime import datetime, timedelta def count_values(df, col_name): value_counts = df[col_name].value_counts() result_df = pd.DataFrame(value_counts) result_df.columns = ['count'] result_df.reset_index(inplace=True) result_df.rename(columns={'index': col_name}, inplace=True) return result_df def add_hyperlink(paragraph, url, text): """ A function that places a hyperlink within a paragraph object. :param paragraph: The paragraph we are adding the hyperlink to. :param url: A string containing the required url :param text: The text displayed for the url :return: A Run object containing the hyperlink """ # This gets access to the document.xml.rels file and gets a new relation id value part = paragraph.part r_id = part.relate_to(url, RT.HYPERLINK, is_external=True) # Create the w:hyperlink tag and add needed values hyperlink = OxmlElement('w:hyperlink') hyperlink.set(qn('r:id'), r_id, ) hyperlink.set(qn('w:history'), '1') # Create a w:r element new_run = OxmlElement('w:r') # Create a new w:rPr element rPr = OxmlElement('w:rPr') # Create a w:rStyle element, note this currently does not add the hyperlink style as its not in # the default template, I have left it here in case someone uses one that has the style in it rStyle = OxmlElement('w:rStyle') rStyle.set(qn('w:val'), 'Hyperlink') # Join all the xml elements together add add the required text to the w:r element rPr.append(rStyle) new_run.append(rPr) new_run.text = text hyperlink.append(new_run) # Create a new Run object and add the hyperlink into it r = paragraph.add_run() r._r.append(hyperlink) # A workaround for the lack of a hyperlink style (doesn't go purple after using the link) # Delete this if using a template that has the hyperlink style in it r.font.color.theme_color = MSO_THEME_COLOR_INDEX.HYPERLINK r.font.underline = True return r def create_table(document,count_df1): table = document.add_table(rows=2, cols=2) # 设置表格宽度 table.columns[0].width = docx.shared.Inches(3.7) table.columns[1].width = docx.shared.Inches(3.7) # 设置表格边框 table.style = 'Table Grid' # 设置表格第一行内容 table.rows[0].height = docx.shared.Pt(9) first_row_cells = table.rows[0].cells first_row_cells[0].text = "技术进展" first_row_cells[0].paragraphs[0].alignment = docx.enum.text.WD_PARAGRAPH_ALIGNMENT.CENTER first_row_cells[1].text = "业内动态" first_row_cells[1].paragraphs[0].alignment = docx.enum.text.WD_PARAGRAPH_ALIGNMENT.CENTER # 设置第一行字体 font = first_row_cells[0].paragraphs[0].runs[0].font font.name = "思源黑体 Regular" first_row_cells[0].paragraphs[0].runs[0]._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Regular") font.size = docx.shared.Pt(8) font.bold = True font = first_row_cells[1].paragraphs[0].runs[0].font font.name = "思源黑体 Regular" first_row_cells[1].paragraphs[0].runs[0]._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Regular") font.size = docx.shared.Pt(8) font.bold = True # 设置表格第二行内容 second_row_cells = table.rows[1].cells second_row_cells[0].text = '''\t图像理解与生成 \t{0}项\n\t计算光学 \t{1}项\n\t图像处理 \t{2}项\n\t机器学习前沿 \t{3}项\n\t自然语言交互 \t{4}项\n\t量子计算 \t{5}项\n\t计算机视觉前沿 \t{6}项'''.format(count_df1[0],count_df1[1],count_df1[2], count_df1[3],count_df1[4],count_df1[5],count_df1[6]) second_row_cells[0].paragraphs[0].alignment = docx.enum.text.WD_PARAGRAPH_ALIGNMENT.LEFT second_row_cells[1].text = "\t大厂动态 \t{0}项\n".format(count_df1[7]) second_row_cells[1].paragraphs[0].alignment = docx.enum.text.WD_PARAGRAPH_ALIGNMENT.LEFT # 设置第二行字体 font = second_row_cells[0].paragraphs[0].runs[0].font font.name = "思源黑体 Regular" second_row_cells[0].paragraphs[0].runs[0]._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Regular") font.size = docx.shared.Pt(8) font = second_row_cells[1].paragraphs[0].runs[0].font font.name = "思源黑体 Regular" second_row_cells[1].paragraphs[0].runs[0]._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Regular") font.size = docx.shared.Pt(8) # 设置行高 table.rows[0].height = docx.shared.Pt(9) def 荣耀周报排版(xlsx,template): document = Document(template) df = pd.read_excel(xlsx) res = df.sort_values(by='领域', ascending=True) count_df = count_values(df, '领域') count_df1 = count_df.sort_values(by='领域', ascending=True)["count"] count_df1 = list(count_df1) sections = ["图像理解与生成", "计算光学", "图像处理", "机器学习前沿", "自然语言交互", "计算机视觉前沿","量子计算", "定向追踪"] # 开头标注时间 思源黑体 Regular 四号 try: date_style = document.styles['date_range'] date_style.font.name = "思源黑体 Regular" date_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Regular") date_style.font.bold = True date_style.font.size = Pt(14) except: date_style = document.styles.add_style('date_range', 1) date_style.font.name = "思源黑体 Regular" date_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Regular") date_style.font.bold = True date_style.font.size = Pt(14) # 设置标题样式 思源黑体 Bold 三号 try: title_style = document.styles['title2'] title_style.font.name = "思源黑体 Bold" title_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Bold") title_style.font.bold = True title_style.font.size = Pt(16) except: title_style = document.styles.add_style('title2',1) title_style.font.name = "思源黑体 Bold" title_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Bold") title_style.font.bold = True title_style.font.size = Pt(16) try: title_style = document.styles['title'] title_style.base_style = document.styles['Heading 1'] title_style.font.name = "思源黑体 Bold" title_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Bold") title_style.font.color.rgb=RGBColor(0,0,0) title_style.font.bold = True title_style.font.size = Pt(16) except: title_style = document.styles.add_style('title',1) title_style.base_style = document.styles['Heading 1'] title_style.font.name = "思源黑体 Bold" title_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Bold") title_style.font.color.rgb=RGBColor(0,0,0) title_style.font.bold = True title_style.font.size = Pt(16) # 热点速览技术进展小标题 思源黑体 小五 下划线 try: tech_style = document.styles['tech_progress'] tech_style.font.name = "思源黑体 Regular" tech_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Regular") tech_style.font.bold = False tech_style.font.size = Pt(9) tech_style.font.underline = True except: tech_style = document.styles.add_style('tech_progress', 1) tech_style.font.name = "思源黑体 Regular" tech_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Regular") tech_style.font.bold = False tech_style.font.size = Pt(9) tech_style.font.underline = True # 热点速览注释与详情 思源黑体 小五 try: cont_style = document.styles['content'] cont_style.font.name = "思源黑体 Regular" cont_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Regular") cont_style.font.bold = False cont_style.font.size = Pt(9) cont_style.font.color.rgb=RGBColor(89,89,89) except: cont_style = document.styles.add_style('content', 1) cont_style.font.name = "思源黑体 Regular" cont_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Regular") cont_style.font.bold = False cont_style.font.size = Pt(9) cont_style.font.color.rgb=RGBColor(89,89,89) # 思源黑体 小四 --部分正文--段落 try: part1_style = document.styles['weekly_summary'] part1_style.font.name = "思源黑体 Regular" part1_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Regular") part1_style.font.bold = False part1_style.font.size = Pt(12) except: part1_style = document.styles.add_style('weekly_summary', 1) part1_style.font.name = "思源黑体 Regular" part1_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Regular") part1_style.font.bold = False part1_style.font.size = Pt(12) # 思源黑体 小四 --部分正文--字符 try: part2_style = document.styles['inside_para'] part2_style.font.name = "思源黑体 Regular" part2_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Regular") part2_style.font.bold = False part2_style.font.size = Pt(12) except: part2_style = document.styles.add_style('inside_para', 2) part2_style.font.name = "思源黑体 Regular" part2_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Regular") part2_style.font.bold = False part2_style.font.size = Pt(12) # 思源黑体 Regular 11号字--热点正文--段落 try: part3_style = document.styles['part3_style'] part3_style.font.name = "思源黑体 Regular" part3_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Regular") part3_style.font.bold = False part3_style.font.size = Pt(11) except: part3_style = document.styles.add_style('part3_style', 1) part3_style.font.name = "思源黑体 Regular" part3_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Regular") part3_style.font.bold = False part3_style.font.size = Pt(11) # 定向追踪-技术进展 思源黑体 Bold 四号 try: tech1_style = document.styles['tech'] tech1_style.font.name = "思源黑体 Bold" tech1_style.base_style = document.styles['Heading 1'] tech1_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Bold") tech1_style.font.color.rgb=RGBColor(0,0,0) tech1_style.font.bold = True tech1_style.font.size = Pt(14) except: tech1_style = document.styles.add_style('tech',1) tech1_style.font.name = "思源黑体 Bold" tech1_style.base_style = document.styles['Heading 1'] tech1_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Bold") tech1_style.font.color.rgb=RGBColor(0,0,0) tech1_style.font.bold = True tech1_style.font.size = Pt(14) try: tech1_style = document.styles['tech2'] tech1_style.font.name = "思源黑体 Bold" tech1_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Bold") tech1_style.font.bold = True tech1_style.font.size = Pt(14) except: tech1_style = document.styles.add_style('tech2',1) tech1_style.font.name = "思源黑体 Bold" tech1_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Bold") tech1_style.font.bold = True tech1_style.font.size = Pt(14) # 定向追踪-技术进展 思源黑体 Regular 小四号，背景灰色-25% try: tech1_style = document.styles['tech1'] tech1_style.font.name = "思源黑体 Regular" tech1_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Regular") tech1_style.font.bold = True tech1_style.font.size = Pt(12) tech1_style.font.highlight_color=WD_COLOR_INDEX.GRAY_25 except: tech1_style = document.styles.add_style('tech1',1) tech1_style.font.name = "思源黑体 Regular" tech1_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Regular") tech1_style.font.bold = True tech1_style.font.size = Pt(12) tech1_style.font.highlight_color=WD_COLOR_INDEX.GRAY_25 # 思源黑体 Bold 小四 --定向追踪标题时间--段落 try: part4_style = document.styles['title_date'] part4_style.font.name = "思源黑体 Bold" part4_style._element.rPr.rFonts.set(qn('w:eastAsia'),"思源黑体 Bold") part4_style.font.bold = False part4_style.font.size = Pt(12) except: part4_style = document.styles.add_style('title_date', 1) part4_style.font.name = "思源黑体 Bold" part4_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Bold") part4_style.font.bold = False part4_style.font.size = Pt(12) # 思源黑体 Light 10 --定向追踪技术--段落 try: part4_style = document.styles['tech_detail'] part4_style.font.name = "思源黑体 Light" part4_style._element.rPr.rFonts.set(qn('w:eastAsia'),"思源黑体 Light") part4_style.font.bold = False part4_style.font.size = Pt(10) except: part4_style = document.styles.add_style('tech_detail', 1) part4_style.font.name = "思源黑体 Light" part4_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Light") part4_style.font.bold = False part4_style.font.size = Pt(10) # 定向追踪-专家点评 思源黑体 Regular 小四号 try: tech1_style = document.styles['expert'] tech1_style.font.name = "思源黑体 Regular" tech1_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Regular") tech1_style.font.bold = True tech1_style.font.size = Pt(12) except: tech1_style = document.styles.add_style('expert',1) tech1_style.font.name = "思源黑体 Regular" tech1_style._element.rPr.rFonts.set(qn('w:eastAsia'), "思源黑体 Regular") tech1_style.font.bold = True tech1_style.font.size = Pt(12) # 设置标题 titles = ['一、本期目录', '二、热点速览', '三、定向追踪'] t1 = "2023 年 x 月 x 日 —— 2023 年 x 月 x 日" t2 = "【本期荣耀周报内容概览】" para1 = document.add_paragraph(t1) para1.style = document.styles["date_range"] run = para1.add_run(" ") run.style = document.styles["inside_para"] run = para1.add_run(t2) run.style = document.styles["inside_para"] run.font.bold = False document.add_paragraph("",style = "weekly_summary") #一、本期目录 document.add_paragraph(titles[0], style='title') document.add_paragraph("", style='title2') # 二、热点速览 document.add_paragraph(titles[1], style='title') document.add_paragraph("", style='weekly_summary') document.add_paragraph("【本周期热点总结】", style='weekly_summary') document.add_paragraph("", style='weekly_summary') document.add_paragraph("以下为本周期热点速览，以事件发生时间排序。", style='weekly_summary') # 添加段落 document.add_paragraph("", style='tech_progress') document.add_paragraph("技术进展 · 【领域】 · 【涉及技术】", style='tech_progress') document.add_paragraph("【技术进展正文】", style='part3_style') document.add_paragraph("热点注释：", style='content') document.add_paragraph("查看详情：", style='content') document.add_paragraph("", style='content') # 三、定向追踪 document.add_paragraph(titles[2], style='title') document.add_paragraph("", style='title2') for section in sections: section2 = section if section != "定向追踪": text1 = "技术进展 · "+section2 document.add_paragraph(text1, style='tech') document.add_paragraph("", style='tech2') document.add_paragraph("进展聚焦", style='tech1') document.add_paragraph("", style='tech1') document.add_paragraph("【占位】", style='weekly_summary') document.add_paragraph("", style='tech1') document.add_paragraph("进展详情", style='tech1') # 表格创建 num = int(count_df[count_df["领域"].str.contains(section)]["count"]) table = document.add_table(rows=num, cols=1) # table.style = 'Table Grid' # 表格填充 res1 = res[res["领域"].str.contains(section)].sort_values(by = "时间",ascending=False) for i, row in enumerate(table.rows): for cell in row.cells: cell.text = "" cell.paragraphs[0].style = "title_date" old_format = '%Y.%m.%d' new_format = '%Y-%m-%d' # 将日期字符串转换为 datetime 对象 date_str = res1.iloc[i]["时间"] date_obj = datetime.strptime(date_str, old_format) # 将 datetime 对象转换为新的日期格式字符串 new_date_str = datetime.strftime(date_obj, new_format) text1 = str(new_date_str )+ " | "+ str(res1.iloc[i]["标题"]) cell.add_paragraph(text1) cell.paragraphs[1].style = "title_date" text2 = "· "+ str(res1.iloc[i]["涉及技术"]) cell.add_paragraph(text2) cell.paragraphs[2].style="tech_detail" text3 = str(res1.iloc[i]["简述(摘要)"]) cell.add_paragraph(text3) cell.paragraphs[3].style = "part3_style" add_hyperlink(cell.paragraphs[3], res1.iloc[i]["源链接"], "原文链接") text4 = "" cell.add_paragraph(text4) cell.paragraphs[4].style = "part3_style" if res1.iloc[i]["是否点评"] == "是": text5 = "专家点评" cell.add_paragraph(text5) cell.paragraphs[5].style = "expert" text6 = "" cell.add_paragraph(text6) cell.paragraphs[5].style = "expert" document.add_page_break() elif section == "定向追踪": text1 = "业内动态 · "+"产品发布" document.add_paragraph(text1, style='tech') document.add_paragraph("", style='tech2') document.add_paragraph("进展聚焦", style='tech1') document.add_paragraph("", style='tech1') document.add_paragraph("【占位】", style='weekly_summary') document.add_paragraph("", style='tech1') document.add_paragraph("进展详情", style='tech1') # 表格创建 table = document.add_table(rows=1, cols=1) # table.style = 'Table Grid' # 表格填充 for row in table.rows: for cell in row.cells: cell.text = "" cell.paragraphs[0].style = "title_date" # 将日期字符串转换为 datetime 对象 text1 = "yyyy-mm-dd"+ " | "+ "【标题占位】" cell.add_paragraph(text1) cell.paragraphs[1].style = "title_date" text2 = "· "+ "【涉及技术】" cell.add_paragraph(text2) cell.paragraphs[2].style="tech_detail" text3 = "【简述(摘要)占位】" cell.add_paragraph(text3) cell.paragraphs[3].style = "part3_style" add_hyperlink(cell.paragraphs[3], "【】", "【原文链接占位】") text4 = "" cell.add_paragraph(text4) cell.paragraphs[4].style = "part3_style" text5 = "【专家点评占位】" cell.add_paragraph(text5) cell.paragraphs[5].style = "expert" text6 = "" cell.add_paragraph(text6) cell.paragraphs[5].style = "expert" document.add_page_break() text1 = "业内动态 · "+"大厂动态" document.add_paragraph(text1, style='tech') document.add_paragraph("", style='tech2') document.add_paragraph("进展聚焦", style='tech1') document.add_paragraph("", style='tech1') document.add_paragraph("【占位】", style='weekly_summary') document.add_paragraph("", style='tech1') document.add_paragraph("进展详情", style='tech1') # 表格创建 table = document.add_table(rows=1, cols=1) # table.style = 'Table Grid' # 表格填充 for row in table.rows: for cell in row.cells: cell.text = "" cell.paragraphs[0].style = "title_date" # 将日期字符串转换为 datetime 对象 text1 = "yyyy-mm-dd"+ " | "+ "【标题占位】" cell.add_paragraph(text1) cell.paragraphs[1].style = "title_date" text2 = "· "+ "【涉及技术】" cell.add_paragraph(text2) cell.paragraphs[2].style="tech_detail" text3 = "【简述(摘要)占位】" cell.add_paragraph(text3) cell.paragraphs[3].style = "part3_style" add_hyperlink(cell.paragraphs[3], "【】", "【原文链接占位】") text4 = "" cell.add_paragraph(text4) cell.paragraphs[4].style = "part3_style" text5 = "【专家点评占位】" cell.add_paragraph(text5) cell.paragraphs[5].style = "expert" text6 = "" cell.add_paragraph(text6) cell.paragraphs[5].style = "expert" document.add_page_break() text1 = "业内动态 · "+"项目开源" document.add_paragraph(text1, style='tech') document.add_paragraph("", style='tech2') document.add_paragraph("进展聚焦", style='tech1') document.add_paragraph("", style='tech1') document.add_paragraph("【占位】", style='weekly_summary') document.add_paragraph("", style='tech1') document.add_paragraph("进展详情", style='tech1') # 表格创建 table = document.add_table(rows=1, cols=1) # table.style = 'Table Grid' # 表格填充 for row in table.rows: for cell in row.cells: cell.text = "" cell.paragraphs[0].style = "title_date" # 将日期字符串转换为 datetime 对象 text1 = "yyyy-mm-dd"+ " | "+ "【标题占位】" cell.add_paragraph(text1) cell.paragraphs[1].style = "title_date" text2 = "· "+ "【涉及技术】" cell.add_paragraph(text2) cell.paragraphs[2].style="tech_detail" text3 = "【简述(摘要)占位】" cell.add_paragraph(text3) cell.paragraphs[3].style = "part3_style" add_hyperlink(cell.paragraphs[3], "【】", "【原文链接占位】") text4 = "" cell.add_paragraph(text4) cell.paragraphs[4].style = "part3_style" text5 = "【专家点评占位】" cell.add_paragraph(text5) cell.paragraphs[5].style = "expert" text6 = "" cell.add_paragraph(text6) cell.paragraphs[5].style = "expert" return document import pandas as pd import docx # Gradio 部分 import gradio as gr import streamlit as st from io import BytesIO # def excel_to_docx(xlsx): # # 处理 Excel 文件并生成 docx 文件 # document,name = 荣耀周报排版(xlsx) # return document.getvalue() # 定义 Gradio 的输入和输出界面 # inputs = gr.inputs.File(label="Excel 文件", type=["file"]) # outputs = gr.outputs.File(label="docx 文件") st.title('Translator App') st.markdown("Translate from Docx file") st.subheader("File Upload") datas=st.file_uploader("Original File") template=st.file_uploader("template File") name=st.text_input('Enter New File Name: ') stream = BytesIO() if st.button(label='生成'): st.spinner('Waiting...') document= 荣耀周报排版(datas,template) out = document.save(stream) st.success("Translated") st.download_button(label='Download Translated File',file_name=(f"{name}.docx"), data=stream.getvalue())