Update app.py

app.py

@@ -76,30 +76,19 @@ def create_print_layout(data, title):
     if data.empty:
         return None
         
-    # 设置 A5 横向尺寸（单位：英寸）
-    fig = plt.figure(figsize=(8.27, 5.83), dpi=300)
-    
-    # 减小边距，最大化利用空间
-    plt.subplots_adjust(left=0.02, right=0.98, top=0.98, bottom=0.02)
-    
-    # 创建网格，减小间距
-    rows = (len(data) + 3) // 4  # 向上取整除法
-    gs = gridspec.GridSpec(rows, 4, hspace=0.1, wspace=0.1)  # 显著减小间距
+    # 设置 A5 纵向尺寸（单位：英寸）
+    fig, axes = plt.subplots(figsize=(5.83, 8.27), dpi=300, nrows=len(data), ncols=1)
     
     # 设置字体和样式
     plt.rcParams['font.family'] = 'sans-serif'
     plt.rcParams['font.sans-serif'] = ['SimHei']  # 中文字体
     
     # 计算适合的字体大小
-    base_fontsize = min(45, 220 / rows)  # 增加基础字体大小
+    base_fontsize = 45  # 增加基础字体大小
     
     # 填充数据
     for idx, (hall, end_time) in enumerate(data.values):
-        row = idx // 4
-        col = idx % 4
-        
-        # 创建子图
-        ax = plt.subplot(gs[row, col])
+        ax = axes[idx]
         
         # 设置更深的边框颜色
         for spine in ax.spines.values():
@@ -114,14 +103,13 @@ def create_print_layout(data, title):
                 ha='center', 
                 va='center')
         
-        # 设置更小的内边距
-        ax.set_xlim(-0.02, 1.02)
-        ax.set_ylim(-0.02, 1.02)
-        
         # 移除坐标轴
         ax.set_xticks([])
         ax.set_yticks([])
     
+    # 调整图形大小，减少多余边距
+    fig.subplots_adjust(left=0.02, right=0.98, top=0.98, bottom=0.02)
+    
     # 将图表转换为 base64 字符串
     buffer = io.BytesIO()
     plt.savefig(buffer, format='png', bbox_inches='tight', pad_inches=0.05)  # 减小边距
@@ -161,4 +149,4 @@ if uploaded_file:
             if part2_image:
                 st.image(part2_image)
             else:
-                st.info("夜班部分没有数据")
+                st.info("夜班部分没有数据")