Update app.py

app.py

@@ -302,10 +302,9 @@ def DispersionPlot(textParty):
         print(f"Debug DispersionPlot: FreqDist sample: {list(fdistance.most_common(10))}") # Debug print
 
         # --- Improved word selection logic ---
-        # Get common words, excluding very short words which might be punctuation/non-informative
-        # or words that dispersion_plot might have trouble with.
+        # Get common words, handle potential IndexError if less than 5 unique words
         common_words_raw = fdistance.most_common(15) # Check a few more common words
-        # Filter: length > 2, isalpha (to avoid punctuation), not just digits
+        # Filter: length > 2, isalpha (to avoid punctuation/non-informative), not just digits
         common_words_filtered = [(word, freq) for word, freq in common_words_raw if len(word) > 2 and word.isalpha() and not word.isdigit()]
         print(f"Debug DispersionPlot: Filtered common words: {common_words_filtered}") # Debug print
 
@@ -316,7 +315,6 @@ def DispersionPlot(textParty):
             word_Lst = [common_words_filtered[x][0] for x in range(5)]
 
         # Final check: Ensure words are present in the Text object (moby)
-        # This is crucial for dispersion_plot
         final_word_list = [word for word in word_Lst if word in moby] # Check membership in the Text object
         print(f"Debug DispersionPlot: Final word list for plot: {final_word_list}") # Debug print
 
@@ -360,11 +358,8 @@ def DispersionPlot(textParty):
             buf.close() # Ensure buffer is closed on error
         traceback.print_exc()
         plt.close('all') # Aggressive close on error
-        # Optionally return a placeholder image or None
         return None # Return None on error
 
-
-
 def word_cloud_generator(parsed_text_name, text_Party):
     """Generates the word cloud image."""
     buf = None # Initialize buffer