Update app.py

app.py

@@ -112,33 +112,34 @@ if st.button("Generate HeatMap & DataFrame"):
             mime='text/csv',
         )
 
-        # Save the DataFrame to an Excel file with images
-        if st.button("Save to Excel"):
-            # Create a new Excel writer object
-            writer = pd.ExcelWriter('emotion_scores.xlsx', engine='xlsxwriter')
-            df_emotions.to_excel(writer, index=False)
-
-            # Access the xlsxwriter workbook and worksheet objects
-            workbook  = writer.book
-            worksheet = writer.sheets['Sheet1']
-
-            # Iterate over the images and insert them into the 'Image View' column
-            for idx, image_path in enumerate(st.session_state['image_names']):
-                # Your logic to define the image_path
-                worksheet.insert_image(f'J{idx + 2}', image_path)  # 'J' is the 10th column
-
-            # Close the writer and save the Excel file
-            writer.save()
-            st.success('DataFrame generated and saved as emotion_scores.xlsx')
-
-            with open('emotion_scores.xlsx', 'rb') as f:
-                excel_file = f.read()
-
-            st.download_button(
-                label='Download Emotion Scores as Excel',
-                data=excel_file,
-                file_name='emotion_scores.xlsx',
-                mime='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
-            )
-    else:
-        st.error("No results to generate DataFrame. Please predict emotions first.")
+# Save the DataFrame to an Excel file with images
+if st.button("Save to Excel"):
+    # Create a new Excel writer object
+    writer = pd.ExcelWriter('emotion_scores.xlsx', engine='xlsxwriter')
+    df_emotions.to_excel(writer, index=False)
+
+    # Access the xlsxwriter workbook and worksheet objects
+    workbook  = writer.book
+    worksheet = writer.sheets['Sheet1']
+
+    # Iterate over the images and insert them into the 'Image View' column
+    for idx, image_path in enumerate(st.session_state['image_names']):
+        # Your logic to define the image_path
+        worksheet.insert_image(f'J{idx + 2}', image_path)  # 'J' is the 10th column
+
+    # Close the writer and save the Excel file
+    writer.save()
+    st.session_state['excel_saved'] = True
+    st.success('DataFrame generated and saved as emotion_scores.xlsx')
+
+# Check if the Excel file was saved and provide a download button
+if st.session_state.get('excel_saved', False):
+    with open('emotion_scores.xlsx', 'rb') as f:
+        excel_file = f.read()
+
+    st.download_button(
+        label='Download Emotion Scores as Excel',
+        data=excel_file,
+        file_name='emotion_scores.xlsx',
+        mime='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
+    )