Update modules/syntax_analysis.py

modules/syntax_analysis.py

@@ -117,25 +117,29 @@ def create_syntax_graph(doc, lang):
 def visualize_syntax_graph(doc, lang):
     G, word_colors = create_syntax_graph(doc, lang)
 
-    plt.figure(figsize=(20, 15))
-    pos = nx.spring_layout(G, k=2, iterations=100)
+    plt.figure(figsize=(24, 18))  # Increase figure size
+    pos = nx.spring_layout(G, k=0.9, iterations=50)  # Adjust layout parameters
 
     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)
+    nx.draw(G, pos, with_labels=False, node_color=node_colors, node_size=node_sizes, arrows=True, 
+            arrowsize=20, width=2, edge_color='gray')  # Adjust node and edge appearance
 
-    nx.draw_networkx_labels(G, pos, {node: data['label'] for node, data in G.nodes(data=True)}, font_size=8)
+    nx.draw_networkx_labels(G, pos, {node: data['label'] for node, data in G.nodes(data=True)}, 
+                            font_size=10, font_weight='bold')  # Increase font size and make bold
 
     edge_labels = nx.get_edge_attributes(G, 'label')
     nx.draw_networkx_edge_labels(G, pos, edge_labels=edge_labels, font_size=8)
 
-    plt.title("Syntactic Analysis" if lang == 'en' else "Analyse Syntaxique" if lang == 'fr' else "Análisis Sintáctico")
+    plt.title("Syntactic Analysis" if lang == 'en' else "Analyse Syntaxique" if lang == 'fr' else "Análisis Sintáctico",
+              fontsize=20, fontweight='bold')  # Increase title font size
     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]})")
+    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())]
-    plt.legend(handles=legend_elements, loc='center left', bbox_to_anchor=(1, 0.5))
+    plt.legend(handles=legend_elements, loc='center left', bbox_to_anchor=(1, 0.5), fontsize=12)  # Increase legend font size
 
     return plt