Update modules/semantic_analysis.py

modules/semantic_analysis.py

@@ -137,45 +137,45 @@ def extract_entities(doc, lang):
 
 #####################################################################################################################
 
-def visualize_context_graph(doc, lang):
-    G = nx.Graph()
-    entities = extract_entities(doc, lang)
-    color_map = ENTITY_LABELS[lang]
+#def visualize_context_graph(doc, lang):
+#    G = nx.Graph()
+#    entities = extract_entities(doc, lang)
+#    color_map = ENTITY_LABELS[lang]
 
     # Add nodes
-    for category, items in entities.items():
-        for item in items:
-            G.add_node(item, category=category)
+#    for category, items in entities.items():
+#        for item in items:
+#            G.add_node(item, category=category)
 
     # Add edges
-    for sent in doc.sents:
-        sent_entities = [ent for ent in sent.ents if ent.text in G.nodes()]
-        for i in range(len(sent_entities)):
-            for j in range(i+1, len(sent_entities)):
-                G.add_edge(sent_entities[i].text, sent_entities[j].text)
+#    for sent in doc.sents:
+#        sent_entities = [ent for ent in sent.ents if ent.text in G.nodes()]
+#        for i in range(len(sent_entities)):
+#            for j in range(i+1, len(sent_entities)):
+#                G.add_edge(sent_entities[i].text, sent_entities[j].text)
 
     # Visualize
-    plt.figure(figsize=(30, 22))  # Increased figure size
-    pos = nx.spring_layout(G, k=0.7, iterations=50)  # Adjusted layout
+#    plt.figure(figsize=(30, 22))  # Increased figure size
+#    pos = nx.spring_layout(G, k=0.7, iterations=50)  # Adjusted layout
 
-    node_colors = [color_map[G.nodes[node]['category']] for node in G.nodes()]
+#    node_colors = [color_map[G.nodes[node]['category']] for node in G.nodes()]
 
-    nx.draw(G, pos, node_color=node_colors, with_labels=True, 
-            node_size=10000,  # Increased node size
-            font_size=18,  # Increased font size
-            font_weight='bold',
-            width=2,  # Increased edge width
-            arrowsize=30)  # Increased arrow size
+#    nx.draw(G, pos, node_color=node_colors, with_labels=True, 
+#            node_size=10000,  # Increased node size
+#            font_size=18,  # Increased font size
+#            font_weight='bold',
+#            width=2,  # Increased edge width
+#            arrowsize=30)  # Increased arrow size
 
     # Add a legend
-    legend_elements = [plt.Rectangle((0,0),1,1,fc=color, edgecolor='none', label=category) 
+#    legend_elements = [plt.Rectangle((0,0),1,1,fc=color, edgecolor='none', label=category) 
                        for category, color in color_map.items()]
-    plt.legend(handles=legend_elements, loc='upper left', bbox_to_anchor=(1, 1), fontsize=16)  # Increased legend font size
+#    plt.legend(handles=legend_elements, loc='upper left', bbox_to_anchor=(1, 1), fontsize=16)  # Increased legend font size
 
-    plt.title("Análisis del Contexto" if lang == 'es' else "Context Analysis" if lang == 'en' else "Analyse du Contexte", fontsize=24)  # Increased title font size
-    plt.axis('off')
+#    plt.title("Análisis del Contexto" if lang == 'es' else "Context Analysis" if lang == 'en' else "Analyse du Contexte", fontsize=24)  # Increased title font size
+#    plt.axis('off')
 
-    return plt
+#    return plt
 
 ############################################################################################################################################