Update modules/semantic_analysis.py

modules/semantic_analysis.py

@@ -150,11 +150,11 @@ def create_semantic_graph(doc, lang):
         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)
+                       color=POS_COLORS.get(token.pos_, '#CCCCCC'),  # Color gris por defecto
+                       size=pos_counts.get(token.pos_, 1) * 100)  # Tamaño mínimo si no hay conteo
 
     for token in doc:
-        if token.dep_ != "ROOT":
+        if token.dep_ != "ROOT" and token.head.text in G.nodes and token.text in G.nodes:
             G.add_edge(token.head.text, token.text, label=token.dep_)
 
     return G, pos_counts
@@ -165,8 +165,8 @@ def visualize_semantic_relations(doc, lang):
     plt.figure(figsize=(24, 18))
     pos = nx.spring_layout(G, k=0.9, iterations=50)
 
-    node_colors = [data['color'] for _, data in G.nodes(data=True)]
-    node_sizes = [data['size'] for _, data in G.nodes(data=True)]
+    node_colors = [G.nodes[node].get('color', '#CCCCCC') for node in G.nodes()]
+    node_sizes = [G.nodes[node].get('size', 100) for node in G.nodes()]
 
     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')
@@ -179,7 +179,7 @@ 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]} ({pos_counts[pos]})")
+                       label=f"{POS_TRANSLATIONS[lang].get(pos, pos)} ({pos_counts.get(pos, 0)})")
                        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)