Update app.py

app.py

@@ -136,7 +136,6 @@ def match_distribution(schedule_df, conference_team_df):
 
 # Inter-Conference Match Analysis
 def inter_conference_analysis(schedule_df, conference_team_df):
-    # Check if the DataFrame is None
     if schedule_df is None:
         plt.figure(figsize=(10, 6))
         plt.text(0.5, 0.5, 'Please generate the schedule first before viewing analytics.', 
@@ -146,29 +145,29 @@ def inter_conference_analysis(schedule_df, conference_team_df):
         plt.tight_layout()
         plt.show()
         return
-    """Generate a heatmap showing inter-conference match frequencies."""
-    # Map team names to their conferences
-    team_to_conference = dict(zip(conference_team_df['Team'], conference_team_df['Conference']))
+
+    # Mapping teams to their conferences from the conference_team_df
+    team_to_conference = conference_team_df.set_index('Team')['Conference'].to_dict()
     schedule_df['Conference 1'] = schedule_df['Team 1'].map(team_to_conference)
     schedule_df['Conference 2'] = schedule_df['Team 2'].map(team_to_conference)
+
+    # Filtering out the intra-conference matches
+    inter_conference_df = schedule_df[schedule_df['Conference 1'] != schedule_df['Conference 2']]
     
-    # Filter out intra-conference matches
-    inter_conference_df = st.session_state.schedule_df[st.session_state.schedule_df['Conference 1'] != st.session_state.schedule_df['Conference 2']]
-    
-    # Create a crosstab for the heatmap
+    # Creating a crosstab for the heatmap
     heatmap_data = pd.crosstab(inter_conference_df['Conference 1'], inter_conference_df['Conference 2'])
-    
-    # Ensure every conference combination has a value
-    all_conferences = st.session_state.schedule_df['Conference 1'].unique()
+
+    # Ensuring every conference combination has a value
+    all_conferences = set(conference_team_df['Conference'])
     for conf in all_conferences:
         if conf not in heatmap_data.columns:
             heatmap_data[conf] = 0
         if conf not in heatmap_data.index:
             heatmap_data.loc[conf] = 0
-    
-    heatmap_data = heatmap_data.sort_index().sort_index(axis=1)
-    
-    # Plot
+
+    heatmap_data = heatmap_data.loc[sorted(all_conferences), sorted(all_conferences)]
+
+    # Plotting the heatmap
     plt.figure(figsize=(8, 6))
     sns.heatmap(heatmap_data, annot=True, cmap='Oranges', linewidths=.5, cbar_kws={'label': 'Number of Matches'})
     plt.title('Inter-Conference Match Analysis')
@@ -176,6 +175,7 @@ def inter_conference_analysis(schedule_df, conference_team_df):
     plt.xlabel('Conference 2')
     plt.show()
 
+
 # Commissioner Analytics
 def commissioner_analytics(schedule_df, conference_team_df, commissioners):
     # Check if the DataFrame is None