Update app.py

app.py

@@ -195,7 +195,38 @@ def merge_metadata_with_generated_data(generated_data, excel_metadata):
             generated_data[post_number] = post_data  
     return generated_data
 
-# Create DOCX file matching the uploaded format
+def add_frames_table(doc, mapping):
+    """
+    Adds a well-structured table for the frames mapping.
+    Each row corresponds to a frame from the mapping and shows which category is ticked.
+    """
+    # Create a table with 1 header row and 5 columns.
+    table = doc.add_table(rows=1, cols=5)
+    table.style = "Table Grid"  # Use a grid style for clearer borders
+    hdr_cells = table.rows[0].cells
+    hdr_cells[0].text = "Frame"
+    hdr_cells[1].text = "Major Focus"
+    hdr_cells[2].text = "Significant Focus"
+    hdr_cells[3].text = "Minor Mention"
+    hdr_cells[4].text = "Not Applicable"
+
+    tick = "✓"
+    # Add one row per frame in the mapping.
+    for frame, category in mapping.items():
+        row_cells = table.add_row().cells
+        row_cells[0].text = frame
+        row_cells[1].text = tick if category == "Major Focus" else ""
+        row_cells[2].text = tick if category == "Significant Focus" else ""
+        row_cells[3].text = tick if category == "Minor Mention" else ""
+        row_cells[4].text = tick if category == "Not Applicable" else ""
+    
+    # Optionally set a fixed column width for consistency.
+    # (Adjust the Inches values as necessary.)
+    for col in table.columns:
+        for cell in col.cells:
+            cell.width = Inches(1.5)
+    return table
+
 def create_docx_from_data(extracted_data):
     doc = Document()
     for post_number, data in extracted_data.items():
@@ -212,34 +243,19 @@ def create_docx_from_data(extracted_data):
             para = doc.add_paragraph()
             run = para.add_run(f"**{key}:** {value}")
             run.font.size = Pt(11)
-        # Instead of adding the Frames as plain text, add a proper table if FramesMapping is present.
+        
+        # Instead of a plain-text table, use the structured table for frames if available.
         if "FramesMapping" in data:
             doc.add_paragraph("Frames:")
-            mapping = data["FramesMapping"]
-            # Create a table with header row and one row per frame
-            table = doc.add_table(rows=1, cols=5)
-            table.style = "Light List Accent 1"
-            hdr_cells = table.rows[0].cells
-            hdr_cells[0].text = "Frame"
-            hdr_cells[1].text = "Major Focus"
-            hdr_cells[2].text = "Significant Focus"
-            hdr_cells[3].text = "Minor Mention"
-            hdr_cells[4].text = "Not Applicable"
-            tick = "✓"
-            for frame, category in mapping.items():
-                row_cells = table.add_row().cells
-                row_cells[0].text = frame
-                row_cells[1].text = tick if category == "Major Focus" else ""
-                row_cells[2].text = tick if category == "Significant Focus" else ""
-                row_cells[3].text = tick if category == "Minor Mention" else ""
-                row_cells[4].text = tick if category == "Not Applicable" else ""
+            add_frames_table(doc, data["FramesMapping"])
         else:
-            # Fallback: if no FramesMapping, simply add the plain text value for "Frames"
+            # Fallback: if no mapping, add the markdown text.
             value = data.get("Frames", "N/A")
             doc.add_paragraph(f"**Frames:** {value}")
         doc.add_paragraph("\n")
     return doc
 
+
 # -------------------------------------------------------------------
 # Streamlit App UI
 # -------------------------------------------------------------------