Update modules/semantic_analysis.py

modules/semantic_analysis.py

@@ -135,6 +135,7 @@ def extract_entities(doc, lang):
     return entities
 
 #####################################################################################################################
+
 def visualize_context_graph(doc, lang):
     G = nx.Graph()
     entities = extract_entities(doc, lang)
@@ -153,25 +154,30 @@ def visualize_context_graph(doc, lang):
                 G.add_edge(sent_entities[i].text, sent_entities[j].text)
 
     # Visualize
-    plt.figure(figsize=(20, 15))
-    pos = nx.spring_layout(G, k=0.5, iterations=50)
+    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()]
 
-    nx.draw(G, pos, node_color=node_colors, with_labels=True, node_size=5000, 
-            font_size=12, font_weight='bold')
+    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) 
                        for category, color in color_map.items()]
-    plt.legend(handles=legend_elements, loc='upper left', bbox_to_anchor=(1, 1))
+    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=20)
+    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
 
 ############################################################################################################################################
+
 def visualize_semantic_relations(doc, lang):
     G = nx.Graph()
     word_freq = Counter(token.text.lower() for token in doc if token.pos_ not in ['PUNCT', 'SPACE'])
@@ -185,29 +191,35 @@ def visualize_semantic_relations(doc, lang):
         if token.text.lower() in top_words and token.head.text.lower() in top_words:
             G.add_edge(token.text, token.head.text, label=token.dep_)
 
-    plt.figure(figsize=(24, 18))
-    pos = nx.spring_layout(G, k=0.9, iterations=50)
+    plt.figure(figsize=(36, 27))  # Increased figure size
+    pos = nx.spring_layout(G, k=0.7, iterations=50)  # Adjusted layout
 
     node_colors = [POS_COLORS.get(G.nodes[node]['pos'], '#CCCCCC') for node in G.nodes()]
 
     nx.draw(G, pos, node_color=node_colors, with_labels=True, 
-            font_size=10, font_weight='bold', arrows=True, arrowsize=20, width=2, edge_color='gray')
+            node_size=10000,  # Increased node size
+            font_size=16,  # Increased font size
+            font_weight='bold', 
+            arrows=True, 
+            arrowsize=30,  # Increased arrow size
+            width=3,  # Increased edge width
+            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)
+    nx.draw_networkx_edge_labels(G, pos, edge_labels=edge_labels, font_size=14)  # Increased edge label font size
 
     title = {
         'es': "Relaciones Semánticas Relevantes",
         'en': "Relevant Semantic Relations",
         'fr': "Relations Sémantiques Pertinentes"
     }
-    plt.title(title[lang], fontsize=20, fontweight='bold')
+    plt.title(title[lang], fontsize=24, fontweight='bold')  # Increased title font size
     plt.axis('off')
 
     legend_elements = [plt.Rectangle((0,0),1,1, facecolor=POS_COLORS.get(pos, '#CCCCCC'), edgecolor='none', 
                        label=f"{POS_TRANSLATIONS[lang].get(pos, pos)}")
                        for pos in set(nx.get_node_attributes(G, 'pos').values())]
-    plt.legend(handles=legend_elements, loc='center left', bbox_to_anchor=(1, 0.5), fontsize=12)
+    plt.legend(handles=legend_elements, loc='center left', bbox_to_anchor=(1, 0.5), fontsize=16)  # Increased legend font size
 
     return plt