Update app.py

app.py

@@ -197,21 +197,27 @@ def merge_metadata_with_generated_data(generated_data, excel_metadata):
 
 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.
+    Adds a well-formatted table for the frames mapping into the given Document.
+    The table has 5 columns with headers: Frame, Major Focus, Significant Focus,
+    Minor Mention, and Not Applicable.
     """
     # 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
+    table.style = "Table Grid"
+    
+    # Set header cells with bold, centered text.
     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"
-
+    headers = ["Frame", "Major Focus", "Significant Focus", "Minor Mention", "Not Applicable"]
+    for idx, header in enumerate(headers):
+        hdr_cells[idx].text = header
+        for paragraph in hdr_cells[idx].paragraphs:
+            paragraph.alignment = WD_ALIGN_PARAGRAPH.CENTER
+            for run in paragraph.runs:
+                run.font.bold = True
+                run.font.size = Pt(11)
+    
     tick = "✓"
-    # Add one row per frame in the mapping.
+    # Add one row per frame.
     for frame, category in mapping.items():
         row_cells = table.add_row().cells
         row_cells[0].text = frame
@@ -219,12 +225,18 @@ def add_frames_table(doc, mapping):
         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 ""
+        # Format cells: first cell left-aligned, rest centered.
+        for idx, cell in enumerate(row_cells):
+            for paragraph in cell.paragraphs:
+                paragraph.alignment = WD_ALIGN_PARAGRAPH.CENTER if idx > 0 else WD_ALIGN_PARAGRAPH.LEFT
+                for run in paragraph.runs:
+                    run.font.size = Pt(11)
     
-    # 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)
+    # Optionally, set fixed column widths.
+    widths = [Inches(2), Inches(1), Inches(1), Inches(1), Inches(1)]
+    for row in table.rows:
+        for idx, cell in enumerate(row.cells):
+            cell.width = widths[idx]
     return table
 
 def create_docx_from_data(extracted_data):
@@ -241,17 +253,14 @@ def create_docx_from_data(extracted_data):
             if key in ["Tone", "Hashtags"]:
                 value = ", ".join(value) if isinstance(value, list) else value
             para = doc.add_paragraph()
-            run = para.add_run(f"**{key}:** {value}")
+            run = para.add_run(f"{key}: {value}")
             run.font.size = Pt(11)
-        
-        # Instead of a plain-text table, use the structured table for frames if available.
+        # Use a proper table if a FramesMapping is available.
         if "FramesMapping" in data:
             doc.add_paragraph("Frames:")
             add_frames_table(doc, data["FramesMapping"])
         else:
-            # Fallback: if no mapping, add the markdown text.
-            value = data.get("Frames", "N/A")
-            doc.add_paragraph(f"**Frames:** {value}")
+            doc.add_paragraph(f"Frames: {data.get('Frames', 'N/A')}")
         doc.add_paragraph("\n")
     return doc