Update app.py

app.py

@@ -3,6 +3,7 @@ from PIL import Image
 from transformers import pipeline
 import pandas as pd
 import matplotlib.pyplot as plt
+import seaborn as sns
 
 # Initialize session state for results, image names, and image sizes if not already present
 if 'results' not in st.session_state:
@@ -94,6 +95,14 @@ if st.button("Generate DataFrame"):
         # Display the DataFrame
         st.write(df_emotions)
 
+        # Plotting the heatmap for the first seven columns
+        plt.figure(figsize=(10, 10))
+        sns.heatmap(df_emotions.iloc[:, :7], annot=True, fmt=".1f", cmap='viridis')
+        plt.title('Heatmap of Emotion Scores')
+        plt.xlabel('Emotion Categories')
+        plt.ylabel('Data Points')
+        st.pyplot(plt)
+
         # Optional: Save the DataFrame to a CSV file
         df_emotions.to_csv('emotion_scores.csv', index=False)
         st.success('DataFrame generated and saved as emotion_scores.csv')