Update app.py

app.py

@@ -210,15 +210,18 @@ def fDistancePlot(text2Party):
         plt.tight_layout()
     return safe_plot(plot_func)
 
-
 def DispersionPlot(textParty):
     """Generates the word dispersion plot."""
+    buf = None # Initialize buffer
     try:
         word_tokens_party = word_tokenize(textParty)
         if not word_tokens_party:
+            print("Warning: No tokens found for dispersion plot.")
             return None
+
         moby = Text(word_tokens_party)
         fdistance = FreqDist(word_tokens_party)
+        # Get top 5 words, handle potential IndexError if less than 5 unique words
         common_words = fdistance.most_common(6)
         if len(common_words) < 5:
             word_Lst = [word for word, _ in common_words]
@@ -226,33 +229,40 @@ def DispersionPlot(textParty):
             word_Lst = [common_words[x][0] for x in range(5)]
 
         if not word_Lst:
+            print("Warning: No common words found for dispersion plot.")
             return None
 
-        fig, ax = plt.subplots(figsize=(10, 5)) # Explicitly create figure and axes
-        ax.set_title('Dispersion Plot')
-        moby.dispersion_plot(word_Lst, ax=ax) # Pass the axes object
-        fig.tight_layout() # Use fig.tight_layout()
+        # --- Key Fix: Manage figure explicitly without passing 'ax' ---
+        fig = plt.figure(figsize=(10, 5)) # Create figure explicitly
+        plt.title('Dispersion Plot')
+        # Call dispersion_plot without 'ax' argument
+        moby.dispersion_plot(word_Lst)
+        plt.tight_layout()
+
         buf = BytesIO()
-        # Handle potential apply_aspect error for dispersion plot too
+        # Handle potential apply_aspect error for dispersion plot
         try:
             fig.savefig(buf, format='png', bbox_inches='tight')
         except AttributeError as ae:
             if "apply_aspect" in str(ae):
                  print(f"Warning: bbox_inches='tight' failed for Dispersion Plot ({ae}), saving without it.")
                  buf.seek(0)
-                 buf = BytesIO()
+                 buf = BytesIO() # Get a fresh buffer
                  fig.savefig(buf, format='png')
             else:
-                 raise
+                 raise # Re-raise if it's a different AttributeError
         buf.seek(0)
         img = Image.open(buf)
-        plt.close(fig) # Close the specific figure
+        plt.close(fig) # Close the specific figure created
         return img
+
     except Exception as e:
         print(f"Dispersion plot error: {e}")
+        if buf:
+            buf.close() # Ensure buffer is closed on error
         traceback.print_exc()
         plt.close('all') # Aggressive close on error
-        return None
+        return None # Return None on error
 
 
 def word_cloud_generator(parsed_text_name, text_Party):