Update modules/semantic_analysis.py

modules/semantic_analysis.py

@@ -85,6 +85,9 @@ POS_TRANSLATIONS = {
 }
 ########################################################################################################################################
 
+def count_pos(doc):
+    return Counter(token.pos_ for token in doc if token.pos_ != 'PUNCT')
+
 def extract_entities(doc):
     entities = {
         "Personas": [],
@@ -154,10 +157,10 @@ def create_semantic_graph(doc, lang):
         if token.dep_ != "ROOT":
             G.add_edge(token.head.text, token.text, label=token.dep_)
 
-    return G
+    return G, pos_counts
 
 def visualize_semantic_relations(doc, lang):
-    G = create_semantic_graph(doc, lang)
+    G, pos_counts = create_semantic_graph(doc, lang)
 
     plt.figure(figsize=(24, 18))
     pos = nx.spring_layout(G, k=0.9, iterations=50)