Update app.py

app.py

@@ -50,6 +50,9 @@ if st.button("Predict Emotions") and selected_images:
     # Predict emotion for each selected image using the pipeline
     st.session_state['results'] = [pipe(image) for image in selected_images]
 
+# Initialize an empty DataFrame outside of the button press condition
+df_emotions = pd.DataFrame()
+
 # Generate DataFrame from results
 if st.button("Generate HeatMap & DataFrame"):
     # Access the results, image names, and sizes from the session state
@@ -113,7 +116,7 @@ if st.button("Generate HeatMap & DataFrame"):
         )
 
 # Save the DataFrame to an Excel file with images
-if st.button("Save to Excel"):
+if st.button("Save to Excel") and not df_emotions.empty:
     # Create a new Excel writer object
     writer = pd.ExcelWriter('emotion_scores.xlsx', engine='xlsxwriter')
     df_emotions.to_excel(writer, index=False)