Update app.py

app.py

@@ -141,14 +141,14 @@ def get_frame_category_mapping(text):
 
 def format_frame_categories_table(mapping):
     """
-    Returns a markdown-formatted table that displays each frame along with four columns:
+    Returns a markdown-formatted table displaying each frame with columns:
     Major Focus, Significant Focus, Minor Mention, and Not Applicable.
-    A tick (✓) is shown only in the column corresponding to the assigned category.
+    A tick (✓) marks the assigned category.
     """
     header = "| Frame | Major Focus | Significant Focus | Minor Mention | Not Applicable |\n"
     header += "| --- | --- | --- | --- | --- |\n"
-    rows = ""
     tick = "✓"
+    rows = ""
     for frame, category in mapping.items():
         major = tick if category == "Major Focus" else ""
         significant = tick if category == "Significant Focus" else ""
@@ -161,7 +161,6 @@ def format_frame_categories_table(mapping):
 # Existing functions for file processing
 # -------------------------------------------------------------------
 
-# Extract captions from DOCX
 def extract_captions_from_docx(docx_file):
     doc = Document(docx_file)
     captions = {}
@@ -175,7 +174,6 @@ def extract_captions_from_docx(docx_file):
             captions[current_post].append(text)
     return {post: " ".join(lines) for post, lines in captions.items() if lines}
 
-# Extract metadata from Excel file
 def extract_metadata_from_excel(excel_file):
     try:
         df = pd.read_excel(excel_file)
@@ -185,7 +183,6 @@ def extract_metadata_from_excel(excel_file):
         logging.error(f"Error processing Excel file: {e}")
         return []
 
-# Merge metadata with generated analysis
 def merge_metadata_with_generated_data(generated_data, excel_metadata):
     for post_data in excel_metadata:
         post_number = f"Post {post_data.get('Post Number', len(generated_data) + 1)}"
@@ -195,55 +192,6 @@ def merge_metadata_with_generated_data(generated_data, excel_metadata):
             generated_data[post_number] = post_data  
     return generated_data
 
-
-
-def add_frames_table(doc, mapping):
-    """
-    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"
-    
-    # Set header cells with bold, centered text.
-    headers = ["Frame", "Major Focus", "Significant Focus", "Minor Mention", "Not Applicable"]
-    hdr_cells = table.rows[0].cells
-    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 a row for each frame.
-    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 ""
-        # Center-align cells (except the first one, which is left-aligned).
-        for idx, cell in enumerate(row_cells):
-            for paragraph in cell.paragraphs:
-                if idx == 0:
-                    paragraph.alignment = WD_ALIGN_PARAGRAPH.LEFT
-                else:
-                    paragraph.alignment = WD_ALIGN_PARAGRAPH.CENTER
-                for run in paragraph.runs:
-                    run.font.size = Pt(11)
-    
-    # Optionally, set fixed column widths.
-    col_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 = col_widths[idx]
-    return table
-
 def create_docx_from_data(extracted_data):
     doc = Document()
     for post_number, data in extracted_data.items():
@@ -257,17 +205,33 @@ def create_docx_from_data(extracted_data):
             value = data.get(key, "N/A")
             if key in ["Tone", "Hashtags"]:
                 value = ", ".join(value) if isinstance(value, list) else value
-            para = doc.add_paragraph(f"{key}: {value}")
-            para.style.font.size = Pt(11)
-        
-        # If a FramesMapping exists, add a proper table.
+            para = doc.add_paragraph()
+            run = para.add_run(f"**{key}:** {value}")
+            run.font.size = Pt(11)
+        # Add a proper table for Frames if a mapping is available.
         if "FramesMapping" in data:
             doc.add_paragraph("Frames:")
-            add_frames_table(doc, data["FramesMapping"])
+            mapping = data["FramesMapping"]
+            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 ""
         else:
-            # Fallback: add plain text.
-            doc.add_paragraph(f"Frames: {data.get('Frames', 'N/A')}")
-        doc.add_paragraph("")
+            value = data.get("Frames", "N/A")
+            doc.add_paragraph(f"**Frames:** {value}")
+        doc.add_paragraph("\n")
     return doc