Update app.py

app.py

@@ -72,6 +72,64 @@ frame_categories = {
     "Human Rights Advocacy": ["human rights", "violations", "honor killing", "workplace discrimination", "law reform"]
 }
 
+def suggest_themes(keywords):
+    """
+    Suggest themes based on extracted keywords using a simple mapping.
+    You can adjust the mapping dictionary as needed.
+    """
+    theme_mapping = {
+        "violence": "Conflict",
+        "crisis": "Conflict",
+        "repression": "Oppression",
+        "oppression": "Oppression",
+        "freedom": "Empowerment",
+        "hope": "Optimism",
+        "unity": "Solidarity",
+        "progress": "Advancement",
+        "justice": "Social Justice",
+        "rights": "Social Justice",
+        "equality": "Equality",
+        "exploitation": "Exploitation",
+        "mobilize": "Mobilization",
+        "protest": "Activism",
+        "environment": "Environmental",
+        "climate": "Environmental"
+    }
+    suggested = set()
+    for kw in keywords:
+        lower_kw = kw.lower()
+        for key, theme in theme_mapping.items():
+            if key in lower_kw:
+                suggested.add(theme)
+    return list(suggested)
+
+def suggest_frames(themes):
+    """
+    Suggest frames based on the suggested themes.
+    Adjust this mapping to reflect the relationship between themes and your framing categories.
+    """
+    frame_mapping = {
+        "Conflict": "Anti-Extremism & Anti-Violence",
+        "Oppression": "Systemic Oppression",
+        "Empowerment": "Empowerment & Resistance",
+        "Optimism": "Hopeful",
+        "Solidarity": "Positive",
+        "Advancement": "Informative",
+        "Social Justice": "Human Rights & Justice",
+        "Equality": "Gender & Patriarchy",
+        "Exploitation": "Political & State Accountability",
+        "Mobilization": "Grassroots Mobilization",
+        "Activism": "Activism & Advocacy",
+        "Environmental": "Environmental Crisis & Activism"
+    }
+    suggested_frames = set()
+    for theme in themes:
+        for key, frame in frame_mapping.items():
+            if key.lower() in theme.lower():
+                suggested_frames.add(frame)
+    return list(suggested_frames)
+
+
 def extract_keywords(text):
     # Initialize RAKE with default NLTK stopwords
     r = Rake()
@@ -223,7 +281,8 @@ def create_docx_from_data(extracted_data):
             para = doc.add_paragraph()
             run = para.add_run(f"**{key}:** {value}")
             run.font.size = Pt(11)
-        # (rest of your code for Frames table remains unchanged)
+        
+        # Existing code to add the Frames table (if present)
         if "FramesMapping" in data:
             doc.add_paragraph("Frames:")
             mapping = data["FramesMapping"]
@@ -246,10 +305,33 @@ def create_docx_from_data(extracted_data):
         else:
             value = data.get("Frames", "N/A")
             doc.add_paragraph(f"**Frames:** {value}")
+
+        # --- New: Table for Keywords, Themes, and Frames ---
+        # Assume that 'Keywords' is already extracted and stored in data.
+        keywords = data.get("Keywords", [])
+        # Generate suggested themes and frames from keywords
+        themes = suggest_themes(keywords) if keywords else []
+        frames_from_themes = suggest_frames(themes) if themes else []
+        
+        # Create a new table with 3 columns: Keywords, Themes, Frames
+        doc.add_paragraph("Summary Table:")
+        summary_table = doc.add_table(rows=1, cols=3)
+        summary_table.style = "Light List Accent 1"
+        hdr_cells = summary_table.rows[0].cells
+        hdr_cells[0].text = "Keywords"
+        hdr_cells[1].text = "Themes"
+        hdr_cells[2].text = "Frames"
+        
+        row_cells = summary_table.add_row().cells
+        row_cells[0].text = ", ".join(keywords) if keywords else "N/A"
+        row_cells[1].text = ", ".join(themes) if themes else "N/A"
+        row_cells[2].text = ", ".join(frames_from_themes) if frames_from_themes else "N/A"
+        
         doc.add_paragraph("\n")
     return doc
 
 
+
 # -------------------------------------------------------------------
 # Streamlit App UI
 # -------------------------------------------------------------------