Update modules/semantic_analysis.py

modules/semantic_analysis.py

@@ -139,10 +139,25 @@ def visualize_context_graph(doc, lang):
 
     return plt
 
+def create_semantic_graph(doc, lang):
+    G = nx.Graph()
+    pos_counts = count_pos(doc)
+
+    for token in doc:
+        if token.pos_ != 'PUNCT':
+            G.add_node(token.text, 
+                       pos=token.pos_,
+                       color=POS_COLORS.get(token.pos_, '#FFFFFF'),
+                       size=pos_counts[token.pos_] * 100)
+
+    for token in doc:
+        if token.dep_ != "ROOT":
+            G.add_edge(token.head.text, token.text, label=token.dep_)
+
+    return G
+
 def visualize_semantic_relations(doc, lang):
-    # Esta función puede mantener la lógica que ya tienes en visualize_syntax_graph
-    # con algunas modificaciones para enfocarse en relaciones semánticas
-    G, word_colors = create_syntax_graph(doc, lang)
+    G = create_semantic_graph(doc, lang)
 
     plt.figure(figsize=(24, 18))
     pos = nx.spring_layout(G, k=0.9, iterations=50)
@@ -150,11 +165,8 @@ def visualize_semantic_relations(doc, lang):
     node_colors = [data['color'] for _, data in G.nodes(data=True)]
     node_sizes = [data['size'] for _, data in G.nodes(data=True)]
 
-    nx.draw(G, pos, with_labels=False, node_color=node_colors, node_size=node_sizes, arrows=True, 
-            arrowsize=20, width=2, edge_color='gray')
-
-    nx.draw_networkx_labels(G, pos, {node: data['label'] for node, data in G.nodes(data=True)}, 
-                            font_size=10, font_weight='bold')
+    nx.draw(G, pos, node_color=node_colors, node_size=node_sizes, with_labels=True, 
+            font_size=8, font_weight='bold', arrows=True, arrowsize=20, width=2, edge_color='gray')
 
     edge_labels = nx.get_edge_attributes(G, 'label')
     nx.draw_networkx_edge_labels(G, pos, edge_labels=edge_labels, font_size=8)
@@ -164,8 +176,8 @@ def visualize_semantic_relations(doc, lang):
     plt.axis('off')
 
     legend_elements = [plt.Rectangle((0,0),1,1, facecolor=color, edgecolor='none', 
-                       label=f"{POS_TRANSLATIONS[lang][pos]} ({count_pos(doc)[pos]})")
-                       for pos, color in POS_COLORS.items() if pos in set(nx.get_node_attributes(G, 'pos').values())]
+                       label=f"{POS_TRANSLATIONS[lang][pos]} ({pos_counts[pos]})")
+                       for pos, color in POS_COLORS.items() if pos in pos_counts]
     plt.legend(handles=legend_elements, loc='center left', bbox_to_anchor=(1, 0.5), fontsize=12)
 
     return plt