Update app.py

app.py

@@ -4,6 +4,7 @@ import nltk
 from nltk.tokenize import word_tokenize, sent_tokenize
 from nltk.corpus import stopwords
 from nltk.stem import WordNetLemmatizer
+from nltk.text import Text
 from nltk.probability import FreqDist
 from cleantext import clean
 import textract
@@ -31,8 +32,9 @@ import contractions
 load_dotenv()
 
 # Download NLTK resources
-nltk.download(['punkt', 'punkt_tab', 'stopwords', 'wordnet', 'words'])
-  
+nltk.download(['stopwords', 'wordnet', 'words'])
+nltk.download('punkt')
+nltk.download('punkt_tab')
 # Initialize Groq client
 groq_api_key = os.getenv("GROQ_API_KEY")
 groq_client = groq.Groq(api_key=groq_api_key) if groq_api_key else None
@@ -138,10 +140,24 @@ def fDistancePlot(text2Party):
     return safe_plot(lambda: FreqDist(word_tokenize(text2Party)).plot(15, title='Frequency Distribution'))
 
 def DispersionPlot(textParty):
-    tokens = word_tokenize(textParty)
-    moby = Text(tokens)
-    common_words = [item[0] for item in FreqDist(tokens).most_common(5)]
-    return safe_plot(lambda: moby.dispersion_plot(common_words, title='Dispersion Plot'))
+    try:
+        word_tokens_party = word_tokenize(textParty)
+        moby = Text(word_tokens_party)  # Ensure Text is imported
+        fdistance = FreqDist(word_tokens_party)
+        word_Lst = [fdistance.most_common(6)[x][0] for x in range(5)]
+        plt.figure(figsize=(4, 3))
+        plt.title('Dispersion Plot')
+        moby.dispersion_plot(word_Lst)
+        plt.tight_layout()
+        buf = BytesIO()
+        plt.savefig(buf, format='png')
+        buf.seek(0)
+        img = Image.open(buf)
+        plt.clf()
+        return img
+    except Exception as e:
+        print(f"Dispersion plot error: {e}")
+        return None
 
 def word_cloud_generator(parsed_text_name, text_Party):
     try: