Update app.py

app.py

@@ -141,11 +141,8 @@ def generate_report(image_items, job_details, compress_images, quality, progress
 
     # Header
     header = section.header
-    # --- FIX START 1 ---
-    # Original code: header_table = header.add_table(rows=1, cols=2, width=Inches(CONTENT_WIDTH_IN))
-    header_table = header.add_table(rows=1, cols=2)
-    header_table.width = Inches(CONTENT_WIDTH_IN)
-    # --- FIX END 1 ---
+    # --- FIX 1: The header's add_table method REQUIRES the width argument directly ---
+    header_table = header.add_table(rows=1, cols=2, width=Inches(CONTENT_WIDTH_IN))
     header_table.autofit = False
     set_table_fixed_layout(header_table)
     header_table.columns[0].width = Inches(2.5)
@@ -180,20 +177,18 @@ def generate_report(image_items, job_details, compress_images, quality, progress
 
         page_items = image_items[start:start + 2]
 
-        # 2x2 table with borders
-        # --- FIX START 2 ---
-        # Original code: page_table = doc.add_table(rows=2, cols=2, width=Inches(CONTENT_WIDTH_IN))
+        # --- FIX 2: The main document's add_table method does NOT accept the 'width' keyword ---
+        # So we create the table first, then set its width property.
         page_table = doc.add_table(rows=2, cols=2)
         page_table.width = Inches(CONTENT_WIDTH_IN)
-        # --- FIX END 2 ---
         page_table.autofit = False
         set_table_fixed_layout(page_table)
-        page_table.style = 'Table Grid'  # borders ON
+        page_table.style = 'Table Grid'
         page_table.alignment = WD_TABLE_ALIGNMENT.CENTER
 
         # Lock column widths and row heights
-        page_table.columns[0].width = Inches(DESC_COL_IN)   # 1 part
-        page_table.columns[1].width = Inches(PHOTO_COL_IN)  # 4 parts
+        page_table.columns[0].width = Inches(DESC_COL_IN)
+        page_table.columns[1].width = Inches(PHOTO_COL_IN)
         for r in page_table.rows:
             r.height = Inches(ROW_HEIGHT_IN)
             r.height_rule = WD_ROW_HEIGHT_RULE.EXACTLY
@@ -258,7 +253,6 @@ def generate_report(image_items, job_details, compress_images, quality, progress
                 if num_images:
                     progress_bar.progress(processed / num_images)
             else:
-                # Empty row for last page single-item case (grid remains consistent)
                 pass
 
     # Save DOCX
@@ -290,21 +284,18 @@ def generate_report(image_items, job_details, compress_images, quality, progress
 # --- STREAMLIT UI ---
 st.set_page_config(layout="centered", page_title="HKC Report Generator")
 
-# Session state
 if 'image_dict' not in st.session_state:
     st.session_state.image_dict = {}
 
 def reset_app():
     st.session_state.clear()
 
-# Header/logo
 if os.path.exists("logo.png"):
     st.image("logo.png", width=200)
 
 st.title("Photo Report Generator")
 st.markdown("---")
 
-# Sidebar
 with st.sidebar:
     st.header("1. Job Information")
     job_key = st.selectbox(
@@ -326,7 +317,6 @@ with st.sidebar:
         reset_app()
         st.rerun()
 
-# Main area
 st.header("Upload, Arrange, Rotate, and Describe Images")
 uploaded_files = st.file_uploader(
     "Upload all your photos here",
@@ -335,7 +325,6 @@ uploaded_files = st.file_uploader(
     key="file_uploader"
 )
 
-# Add newly uploaded files to session
 if uploaded_files:
     for file in uploaded_files:
         fid = getattr(file, "file_id", None) or f"{file.name}-{id(file)}"
@@ -348,7 +337,6 @@ if uploaded_files:
                 'rotation': 0
             }
 
-# Display sortable/rotatable items
 if st.session_state.image_dict:
     st.info("Set the order, rotate if needed, then add descriptions. Exactly 2 photos per page (last page may have 1).")
     items = list(st.session_state.image_dict.items())
@@ -378,7 +366,6 @@ if st.session_state.image_dict:
                 "Description", value=item_data.get('description', ''), key=f"desc_{fid}", height=120
             )
 
-# Generate
 if generate_button:
     if not job_details:
         st.error("Please select a job.")
@@ -394,7 +381,6 @@ if generate_button:
                 )
             progress_bar.success("Report generated successfully!")
 
-            # File name per dropdown + yymmdd
             date_str = datetime.now().strftime("%y%m%d")
             file_base = f"{job_key} - {date_str}"