Update app.py

app.py

@@ -138,7 +138,6 @@ def extract_captions_from_docx(docx_file):
 def extract_metadata_from_excel(excel_file):
     try:
         df = pd.read_excel(excel_file)
-        # Ensure the required columns are present
         required_columns = ["Date", "Media Type", "Number of Pictures", "Number of Videos", "Number of Audios", "Likes", "Comments", "Tagged Audience"]
         if not all(col in df.columns for col in required_columns):
             st.error("Excel file is missing required columns.")
@@ -236,23 +235,32 @@ if uploaded_excel:
 if uploaded_excel:
     output_data = merge_metadata_with_generated_data(output_data, excel_metadata)
 
+# Filtering or sorting options
+tone_filter = st.multiselect("Filter by Tone", options=["Emotional", "Harsh", "Somber", "Motivational", "Informative", "Positive", "Angry", "Fearful", "Sarcastic", "Hopeful"])
+frame_filter = st.multiselect("Filter by Frame", options=["Human Rights & Justice", "Political & State Accountability", "Gender & Patriarchy", "Religious Freedom & Persecution", "Grassroots Mobilization", "Environmental Crisis & Activism", "Anti-Extremism & Anti-Violence", "Social Inequality & Economic Disparities", "Activism & Advocacy", "Systemic Oppression", "Intersectionality", "Call to Action", "Empowerment & Resistance", "Climate Justice", "Human Rights Advocacy"])
+
+# Filtered output
+filtered_data = output_data
+
+# Apply filters
+if tone_filter:
+    filtered_data = {key: value for key, value in filtered_data.items() if any(tone in tone_filter for tone in value["Tone"])}
+if frame_filter:
+    filtered_data = {key: value for key, value in filtered_data.items() if any(frame in frame_filter for frame in value["Frames"])}
+
 # Display results in collapsible sections for better UI
-if output_data:
-    for post_number, data in output_data.items():
+if filtered_data:
+    for post_number, data in filtered_data.items():
         with st.expander(post_number):
             for key, value in data.items():
                 st.write(f"**{key}:** {value}")
 
 # Create DOCX file for download
-if output_data:
-    doc = create_docx_from_data(output_data)
+if filtered_data:
+    doc = create_docx_from_data(filtered_data)
     docx_io = io.BytesIO()
     doc.save(docx_io)
     docx_io.seek(0)
-
-    st.download_button(
-        label="Download Extracted Data as DOCX",
-        data=docx_io,
-        file_name="extracted_data.docx",
-        mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
-    )
+    st.download_button(label="Download Analysis as DOCX", data=docx_io, file_name="analysis.docx")
+else:
+    st.warning("No data available after applying filters.")