Changes DV - network analysis-4

functions/visualizations.py

@@ -451,6 +451,9 @@ def generate_network_analysis(df):
     st.header("Network Analysis")
     tab1, tab2 = st.tabs(["Artist Collaborations", "Genre Crossover"])
     
+    # Ensure column names are stripped of spaces
+    df.columns = df.columns.str.strip()
+    
     with tab1:
         st.markdown("**Top Collaborating Artists:** Shows which artists collaborate the most.")
         if 'Artists' in df.columns:
@@ -469,7 +472,7 @@ def generate_network_analysis(df):
             
             st.pyplot(fig)
         else:
-            st.error("Cannot plot: 'Artists' column missing.")
+            st.error("Cannot plot: 'Artists' column missing. Available columns: " + ", ".join(df.columns))
     
     with tab2:
         st.markdown("**Genre Crossover:** Shows relationships between genres using a stacked bar chart.")
@@ -501,4 +504,5 @@ def generate_network_analysis(df):
             
             st.pyplot(fig)
         else:
-            st.error("Cannot plot: 'Genres' column missing.")
+            st.error("Cannot plot: 'Genres' column missing. Available columns: " + ", ".join(df.columns))
+