Update app.py

app.py

@@ -12,7 +12,7 @@ from matplotlib.backends.backend_pdf import PdfPages
 import matplotlib.gridspec as gridspec
 import math
 
-# --- Constants for the second script ---
+# --- 常量定义 ---
 SPLIT_TIME = "17:30"
 BUSINESS_START = "09:30"
 BUSINESS_END = "01:30"
@@ -22,20 +22,20 @@ A5_WIDTH_IN = 5.83
 A5_HEIGHT_IN = 8.27
 NUM_COLS = 3
 
-# --- Functions from the first script ---
+# --- 字体加载与文本处理函数 ---
 def get_font(size=14):
-    """Loads a specific TrueType font, defaulting to a common Chinese font."""
+    """加载中文字体文件"""
     font_path = "simHei.ttc"
     if not os.path.exists(font_path):
         font_path = "SimHei.ttf"
     if os.path.exists(font_path):
         return font_manager.FontProperties(fname=font_path, size=size)
     else:
-        st.warning("SimHei font not found. Display may not be correct. Please add simHei.ttc or SimHei.ttf.")
+        st.warning("警告：未找到SimHei字体文件，显示可能不正确。请将 simHei.ttc 或 SimHei.ttf 文件放入项目文件夹。")
         return font_manager.FontProperties(family='sans-serif', size=size)
 
 def get_pinyin_abbr(text):
-    """Gets the first letter of the Pinyin for the first two Chinese characters of a text."""
+    """获取中文文本前两个字的拼音首字母"""
     if not text:
         return ""
     chars = [c for c in text if '\u4e00' <= c <= '\u9fff'][:2]
@@ -45,7 +45,7 @@ def get_pinyin_abbr(text):
     return ''.join(pinyin_list).upper()
 
 def format_seq(n):
-    """Converts an integer to a circled number string (e.g., 1 -> ①)."""
+    """将数字转换为带圈序号 (①, ②, ③...)"""
     if not isinstance(n, int) or n <= 0:
         return str(n)
     circled_chars = "①②③④⑤⑥⑦⑧⑨⑩⑪⑫⑬⑭⑮⑯⑰⑱⑲⑳" \
@@ -55,8 +55,9 @@ def format_seq(n):
         return circled_chars[n - 1]
     return f'({n})'
 
+# --- '放映时间核对表' 处理函数 ---
 def process_schedule_led(file):
-    """Processes '放映时间核对表.xls'."""
+    """处理 '放映时间核对表.xls' 文件"""
     try:
         date_df = pd.read_excel(file, header=None, skiprows=7, nrows=1, usecols=[3])
         date_str = pd.to_datetime(date_df.iloc[0, 0]).strftime('%Y-%m-%d')
@@ -72,11 +73,9 @@ def process_schedule_led(file):
         df.dropna(subset=['StartTime', 'EndTime', 'Movie'], inplace=True)
         df['Hall'] = df['Hall'].astype(str).str.extract(r'(\d+号)')
         df['StartTime_dt'] = pd.to_datetime(df['StartTime'], format='%H:%M', errors='coerce').apply(
-            lambda t: t.replace(year=base_date.year, month=base_date.month, day=base_date.day) if pd.notnull(t) else t
-        )
+            lambda t: t.replace(year=base_date.year, month=base_date.month, day=base_date.day) if pd.notnull(t) else t)
         df['EndTime_dt'] = pd.to_datetime(df['EndTime'], format='%H:%M', errors='coerce').apply(
-            lambda t: t.replace(year=base_date.year, month=base_date.month, day=base_date.day) if pd.notnull(t) else t
-        )
+            lambda t: t.replace(year=base_date.year, month=base_date.month, day=base_date.day) if pd.notnull(t) else t)
         df.loc[df['EndTime_dt'] < df['StartTime_dt'], 'EndTime_dt'] += timedelta(days=1)
         df = df.sort_values(['Hall', 'StartTime_dt'])
         merged_rows = []
@@ -100,11 +99,11 @@ def process_schedule_led(file):
         merged_df['EndTime_str'] = merged_df['EndTime_dt'].dt.strftime('%H:%M')
         return merged_df[['Hall', 'Seq', 'Movie', 'StartTime_str', 'EndTime_str']], date_str
     except Exception as e:
-        st.error(f"Error processing schedule data: {e}. Please check the file format.")
+        st.error(f"处理数据出错: {e}。请检查文件格式是否正确。")
         return None, date_str
 
 def create_print_layout_led(data, date_str):
-    """Generates print layouts for '放映时间核对表.xls'."""
+    """为 '放映时间核对表' 生成打印布局"""
     if data is None or data.empty:
         return None
     A4_width_in, A4_height_in = 8.27, 11.69
@@ -123,8 +122,7 @@ def create_print_layout_led(data, date_str):
     seq_font_size_pt = (row_height * 0.5) * 72
 
     def get_col_width_in(series, font_size_pt, is_math=False):
-        if series.empty:
-            return 0
+        if series.empty: return 0
         font_prop = get_font(font_size_pt)
         longest_str_idx = series.astype(str).str.len().idxmax()
         max_content = str(series.loc[longest_str_idx])
@@ -152,8 +150,7 @@ def create_print_layout_led(data, date_str):
         for col_name in ['hall', 'seq', 'movie', 'pinyin']:
             x_line = col_x_starts[col_name] + col_widths[col_name]
             line_top_y, line_bottom_y = A4_height_in - row_height, row_height
-            ax.add_line(
-                Line2D([x_line, x_line], [line_bottom_y, line_top_y], color='gray', linestyle=':', linewidth=0.5))
+            ax.add_line(Line2D([x_line, x_line], [line_bottom_y, line_top_y], color='gray', linestyle=':', linewidth=0.5))
         last_hall_drawn = None
         for i, row in data.iterrows():
             y_bottom = A4_height_in - (i + 2) * row_height
@@ -176,17 +173,14 @@ def create_print_layout_led(data, date_str):
             if text_w_in > max_width_in:
                 movie_font_size *= (max_width_in / text_w_in)
                 movie_font_prop = get_font(movie_font_size)
-            ax.text(col_x_starts['movie'] + 0.05, y_center, row['Movie'],
-                    fontproperties=movie_font_prop, ha='left', va='center')
+            ax.text(col_x_starts['movie'] + 0.05, y_center, row['Movie'], fontproperties=movie_font_prop, ha='left', va='center')
             is_last_in_hall = (i == len(data) - 1) or (row['Hall'] != data.loc[i + 1, 'Hall'])
             line_start_x = margin_col_width
             line_end_x = A4_width_in - margin_col_width
             if is_last_in_hall:
-                ax.add_line(Line2D([line_start_x, line_end_x], [y_bottom, y_bottom], color='black', linestyle='-',
-                                   linewidth=0.8))
+                ax.add_line(Line2D([line_start_x, line_end_x], [y_bottom, y_bottom], color='black', linestyle='-', linewidth=0.8))
             else:
-                ax.add_line(Line2D([line_start_x, line_end_x], [y_bottom, y_bottom], color='gray', linestyle=':',
-                                   linewidth=0.5))
+                ax.add_line(Line2D([line_start_x, line_end_x], [y_bottom, y_bottom], color='gray', linestyle=':', linewidth=0.5))
 
     outputs = {}
     for format_type in ['png', 'pdf']:
@@ -207,16 +201,15 @@ def create_print_layout_led(data, date_str):
         plt.close(fig)
     return outputs
 
-# --- Functions from the second script ---
+# --- '放映场次核对表' 处理函数 ---
 def process_schedule_times(file):
-    """Processes '放映场次核对表.xls'."""
+    """处理 '放映场次核对表.xls' 文件"""
     try:
         df = pd.read_excel(file, skiprows=8)
         df = df.iloc[:, [6, 7, 9]]
         df.columns = ['Hall', 'StartTime', 'EndTime']
         df = df.dropna(subset=['Hall', 'StartTime', 'EndTime'])
         df['Hall'] = df['Hall'].str.extract(r'(\d+)号').astype(str) + ' '
-        df['original_end'] = df['EndTime']
         base_date = datetime.today().date()
         df['StartTime'] = pd.to_datetime(df['StartTime'])
         df['EndTime'] = pd.to_datetime(df['EndTime'])
@@ -230,31 +223,19 @@ def process_schedule_times(file):
                 df.at[idx, 'EndTime'] = end_time + timedelta(days=1)
             if row['StartTime'].hour >= 21 and end_time.hour < 9:
                 df.at[idx, 'EndTime'] = end_time + timedelta(days=1)
-        df['time_for_comparison'] = df['EndTime'].apply(
-            lambda x: datetime.combine(base_date, x.time())
-        )
+        df['time_for_comparison'] = df['EndTime'].apply(lambda x: datetime.combine(base_date, x.time()))
         df.loc[df['time_for_comparison'].dt.hour < 9, 'time_for_comparison'] += timedelta(days=1)
-        valid_times = (
-                ((df['time_for_comparison'] >= datetime.combine(base_date, business_start.time())) &
-                 (df['time_for_comparison'] <= datetime.combine(base_date + timedelta(days=1), business_end.time())))
-        )
+        valid_times = (((df['time_for_comparison'] >= datetime.combine(base_date, business_start.time())) &
+                        (df['time_for_comparison'] <= datetime.combine(base_date + timedelta(days=1), business_end.time()))))
         df = df[valid_times]
         df = df.sort_values('EndTime')
         split_time = datetime.strptime(f"{base_date} {SPLIT_TIME}", "%Y-%m-%d %H:%M")
-        split_time_for_comparison = df['time_for_comparison'].apply(
-            lambda x: datetime.combine(base_date, split_time.time())
-        )
+        split_time_for_comparison = df['time_for_comparison'].apply(lambda x: datetime.combine(base_date, split_time.time()))
         part1 = df[df['time_for_comparison'] <= split_time_for_comparison].copy()
         part2 = df[df['time_for_comparison'] > split_time_for_comparison].copy()
         for part in [part1, part2]:
             part['EndTime'] = part['EndTime'].dt.strftime('%-H:%M')
-        date_df = pd.read_excel(
-            file,
-            skiprows=5,
-            nrows=1,
-            usecols=[2],
-            header=None
-        )
+        date_df = pd.read_excel(file, skiprows=5, nrows=1, usecols=[2], header=None)
         date_cell = date_df.iloc[0, 0]
         try:
             if isinstance(date_cell, str):
@@ -265,11 +246,11 @@ def process_schedule_times(file):
             date_str = datetime.today().strftime('%Y-%m-%d')
         return part1[['Hall', 'EndTime']], part2[['Hall', 'EndTime']], date_str
     except Exception as e:
-        st.error(f"处理文件时出错: {str(e)}")
+        st.error(f"处理文件出错: {str(e)}")
         return None, None, None
 
 def create_print_layout_times(data, title, date_str):
-    """Creates print layouts for '放映场次核对表.xls'."""
+    """为 '放映场次核对表' 生成打印布局"""
     if data.empty:
         return None
 
@@ -289,13 +270,12 @@ def create_print_layout_times(data, title, date_str):
         fig = plt.figure(figsize=(A5_WIDTH_IN, A5_HEIGHT_IN), dpi=300)
         fig.subplots_adjust(left=0, right=1, top=1, bottom=0)
         plt.rcParams['font.family'] = 'sans-serif'
-        plt.rcParams['font.sans-serif'] = ['Arial Unicode MS']
+        plt.rcParams['font.sans-serif'] = ['SimHei'] 
         gs = gridspec.GridSpec(
             num_rows + 1, NUM_COLS,
             hspace=0, wspace=0,
             height_ratios=[date_header_height_in] + [cell_height_in] * num_rows,
-            figure=fig
-        )
+            figure=fig)
         data_values = data.values.tolist()
         while len(data_values) % NUM_COLS != 0:
             data_values.append(['', ''])
@@ -320,7 +300,7 @@ def create_print_layout_times(data, title, date_str):
                     spine.set_linewidth(0.75)
                 display_text = f"{hall}{end_time}"
                 ax.text(0.5, 0.5, display_text,
-                        fontsize=base_fontsize,
+                        fontproperties=get_font(base_fontsize),
                         fontweight='bold',
                         ha='center', va='center',
                         transform=ax.transAxes)
@@ -329,7 +309,7 @@ def create_print_layout_times(data, title, date_str):
                 ax.set_facecolor('none')
         ax_date = fig.add_subplot(gs[0, :])
         ax_date.text(0.01, 0.5, f"{date_str} {title}",
-                     fontsize=base_fontsize * 0.5,
+                     fontproperties=get_font(base_fontsize * 0.5),
                      color=DATE_COLOR, fontweight='bold',
                      ha='left', va='center',
                      transform=ax_date.transAxes)
@@ -348,61 +328,58 @@ def create_print_layout_times(data, title, date_str):
     pdf_buffer.seek(0)
     pdf_base64 = base64.b64encode(pdf_buffer.getvalue()).decode()
     plt.close(fig_for_output)
-    return {
-        'png': f'data:image/png;base64,{png_base64}',
-        'pdf': f'data:application/pdf;base64,{pdf_base64}'
-    }
+    return {'png': f'data:image/png;base64,{png_base64}', 'pdf': f'data:application/pdf;base64,{pdf_base64}'}
 
 def display_pdf(base64_pdf):
-    """Generates the HTML to embed a PDF in Streamlit."""
+    """在Streamlit中嵌入显示PDF"""
     return f'<iframe src="{base64_pdf}" width="100%" height="800" type="application/pdf"></iframe>'
 
-# --- Streamlit App ---
-st.set_page_config(page_title="Schedule Printer", layout="wide")
-st.title("Schedule Printer")
+# --- Streamlit 主程序 ---
+st.set_page_config(page_title="影院排期打印工具", layout="wide")
+st.title("影院排期打印工具")
 
-uploaded_file = st.file_uploader("Select '放映时间核对表.xls' or '放映场次核对表.xls'",
+uploaded_file = st.file_uploader("请上传 '放映时间核对表.xls' 或 '放映场次核对表.xls'",
                                  type=["xls", "xlsx"])
 
 if uploaded_file:
+    # 根据文件名中的关键字判断使用哪个处理流程
     if "时间" in uploaded_file.name:
-        st.header("LED Screen Schedule")
-        with st.spinner("Processing file, please wait..."):
+        st.header("LED屏排片表")
+        with st.spinner("正在处理文件，请稍候..."):
             schedule, date_str = process_schedule_led(uploaded_file)
             if schedule is not None and not schedule.empty:
                 output = create_print_layout_led(schedule, date_str)
-                tab1, tab2 = st.tabs(["PDF Preview", "PNG Preview"])
+                tab1, tab2 = st.tabs(["PDF 预览", "图片预览 (PNG)"])
                 with tab1:
                     st.markdown(display_pdf(output['pdf']), unsafe_allow_html=True)
                 with tab2:
                     st.image(output['png'], use_container_width=True)
             else:
-                st.error("Could not process the file. Please check if the file format and content are correct.")
+                st.error("无法处理文件。请检查文件内容和格式是否正确。")
     elif "场次" in uploaded_file.name:
-        st.header("Screening Times Quick Print")
+        st.header("散场时间快捷打印")
         part1, part2, date_str = process_schedule_times(uploaded_file)
         if part1 is not None and part2 is not None:
             part1_output = create_print_layout_times(part1, "A", date_str)
             part2_output = create_print_layout_times(part2, "C", date_str)
             col1, col2 = st.columns(2)
             with col1:
-                st.subheader("Day Shift (End time <= 17:30)")
+                st.subheader("白班 (散场时间 ≤ 17:30)")
                 if part1_output:
-                    tab1_1, tab1_2 = st.tabs(["PDF Preview", "PNG Preview"])
+                    tab1_1, tab1_2 = st.tabs(["PDF 预览 ", "图片预览 (PNG) "]) # 添加空格以区分
                     with tab1_1:
                         st.markdown(display_pdf(part1_output['pdf']), unsafe_allow_html=True)
                     with tab1_2:
                         st.image(part1_output['png'])
                 else:
-                    st.info("No data for the day shift.")
+                    st.info("白班没有排期数据。")
             with col2:
-                st.subheader("Night Shift (End time > 17:30)")
+                st.subheader("晚班 (散场时间 > 17:30)")
                 if part2_output:
-                    tab2_1, tab2_2 = st.tabs(["PDF Preview", "PNG Preview"])
+                    tab2_1, tab2_2 = st.tabs(["PDF 预览  ", "图片预览 (PNG)  "]) # 添加空格以区分
                     with tab2_1:
                         st.markdown(display_pdf(part2_output['pdf']), unsafe_allow_html=True)
                     with tab2_2:
                         st.image(part2_output['png'])
                 else:
-                    st.info("No data for the night shift.")
-                    
+                    st.info("晚班没有排期数据。")