Update app.py

app.py

@@ -112,9 +112,17 @@ def generate_charts(ner_output_ext: dict) -> Tuple[go.Figure, np.ndarray]:
     return fig1, wordcloud_image
 
 def generate_wordcloud(entities: List[Dict], color_map: Dict[str, str], file_path: str) -> np.ndarray:
-
-    image_path = os.path.join(os.path.dirname(__file__), file_path)
+    # Construct the absolute path
+    base_path = os.path.dirname(os.path.abspath(__file__))
+    image_path = os.path.join(base_path, file_path)
     
+    # Debugging statement to print the image path
+    print(f"Image path: {image_path}")
+
+    # Check if the file exists
+    if not os.path.exists(image_path):
+        raise FileNotFoundError(f"Mask image file not found: {image_path}")
+
     mask_image = np.array(Image.open(image_path))
     
     token_texts = []
@@ -123,14 +131,12 @@ def generate_wordcloud(entities: List[Dict], color_map: Dict[str, str], file_pat
 
     for entity in entities:
         for token in entity['tokens']:
-            # Remove any leading non-alphanumeric characters
             cleaned_token = re.sub(r'^\W+', '', token)
             token_texts.append(cleaned_token)
             token_scores.append(entity['score'])
             token_types.append(entity['entity'])
             print(f"{cleaned_token} ({entity['entity']}): {entity['score']}")
 
-    # Create a dictionary for word cloud
     word_freq = {text: score for text, score in zip(token_texts, token_scores)}
 
     def color_func(word, font_size, position, orientation, random_state=None, **kwargs):
@@ -139,13 +145,11 @@ def generate_wordcloud(entities: List[Dict], color_map: Dict[str, str], file_pat
 
     wordcloud = WordCloud(width=800, height=400, background_color='#121212', mask=mask_image, color_func=color_func).generate_from_frequencies(word_freq)
 
-    # Convert to image array
     plt.figure(figsize=(10, 5))
     plt.imshow(wordcloud, interpolation='bilinear')
     plt.axis('off')
     plt.tight_layout(pad=0)
 
-    # Convert plt to numpy array
     plt_image = plt.gcf()
     plt_image.canvas.draw()
     image_array = np.frombuffer(plt_image.canvas.tostring_rgb(), dtype=np.uint8)