Update app.py

app.py

@@ -87,13 +87,6 @@ def categorize_frames(text):
 
     return frames
 
-# Function to generate and download the output as docx
-import os
-
-# Function to generate and download the output as docx
-import os
-from docx import Document
-
 # Function to generate and download the output as docx
 def generate_docx(output):
     try:
@@ -110,16 +103,11 @@ def generate_docx(output):
         for frame in output['Frames']:
             doc.add_paragraph(frame)
 
-        # Save the document in the temporary directory accessible in Streamlit
-        file_path = os.path.join("/tmp", "activism_message_analysis.docx")
+        # Save the document
+        file_path = "/mnt/data/activism_message_analysis.docx"
         doc.save(file_path)
-
-        # Check if the file exists and is valid
-        if os.path.exists(file_path):
-            return file_path
-        else:
-            raise Exception("File saving failed")
-
+        
+        return file_path
     except Exception as e:
         st.write(f"Error generating DOCX file: {e}")
         return None
@@ -173,7 +161,7 @@ if input_text:
         # Add download button for the generated docx
         download_button = st.download_button(
             label="Download Analysis as DOCX",
-            data=open(generate_docx(output), 'rb').read(),  # Open file in binary mode
+            data=generate_docx(output),
             file_name="activism_message_analysis.docx",
             mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
         )