Spaces:
Runtime error
Runtime error
File size: 3,224 Bytes
822c3cb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
from docxtpl import DocxTemplate
import pandas as pd
import gradio as gr
import os
from dotenv import load_dotenv
# Load environment variables at the start of your script
load_dotenv()
def process_excel_and_generate_docs(excel_file, term, start_year, end_year, start_month, start_day, end_month, end_day):
# Read the Excel file
df = pd.read_excel(excel_file.name)
# Initialize the news dictionary
news_dict = {'news': {}}
# Group by area and convert to the desired format
for area, group in df.groupby('area'):
news_dict['news'][area.lower()] = [
{
'title': row['translated_title'],
'date': row['date'],
'content': str(row['translated_content']).replace('\r\n', '\n').replace('\r', '\n').replace('\n', '\r\n'), # 统一转换为 Word 兼容的换行符
'comment': row['comment'],
'country': row['country']
}
for _, row in group.iterrows()
]
# Create the context dictionary with all required fields
context = {
'term': term,
'start_year': start_year,
'end_year': end_year,
"start_month": start_month,
"start_day": start_day,
"end_month": end_month,
"end_day": end_day,
**news_dict
}
# Render PDF template
pdf_output_path = "pdf.docx"
pdf_tpl = DocxTemplate("v1.1 周报模板.docx")
pdf_tpl.render(context)
pdf_tpl.save(pdf_output_path)
# Render Email template
email_output_path = "email.docx"
email_tpl = DocxTemplate("v1.1 周报邮件格式调整.docx")
email_tpl.render(context)
email_tpl.save(email_output_path)
return [pdf_output_path, email_output_path]
def create_gradio_interface():
# Define authentication credentials
with gr.Blocks() as app:
gr.Markdown("# Weekly Report Generator")
with gr.Row():
excel_file = gr.File(label="Upload Excel File")
with gr.Row():
term = gr.Textbox(label="期数", value="201")
start_year = gr.Textbox(label="起始年", value="2024")
end_year = gr.Textbox(label="结束年", value="2024")
with gr.Row():
start_month = gr.Textbox(label="起始月份", value="9")
start_day = gr.Textbox(label="起始日", value="1")
end_month = gr.Textbox(label="结束月份", value="9")
end_day = gr.Textbox(label="结束日", value="15")
generate_btn = gr.Button("Generate Reports")
with gr.Row():
pdf_output = gr.File(label="Download PDF Report")
email_output = gr.File(label="Download Email Template")
generate_btn.click(
fn=process_excel_and_generate_docs,
inputs=[excel_file, term, start_year, end_year, start_month, start_day, end_month, end_day],
outputs=[pdf_output, email_output]
)
return app
if __name__ == "__main__":
app = create_gradio_interface()
app.launch(
share=True,
auth=auth,
max_threads=3, # Limit concurrent processing
show_error=True
)
|