Update app.py

app.py

@@ -311,30 +311,9 @@ def main():
             st.subheader("Semantic Network")
             semantic_graph = analyzer.create_semantic_network(text)
             
-            # Convert NetworkX graph to Plotly
-            edge_x = []
-            edge_y = []
-            node_x = []
-            node_y = []
-            node_text = []
-            
-            # Add edges
-            for edge in semantic_graph.edges():
-                x0, y0 = semantic_graph.nodes[edge[0]]['pos']
-                x1, y1 = semantic_graph.nodes[edge[1]]['pos']
-                edge_x.extend([x0, x1, None])
-                edge_y.extend([y0, y1, None])
-            
-            # Add nodes
-            for node in semantic_graph.nodes():
-                x, y = semantic_graph.nodes[node]['pos']
-                node_x.append(x)
-                node_y.append(y)
-                node_text.append(node)
-            
             # Create the network visualization
             network_fig = go.Figure()
-
+        
             # Add edges with varying widths
             for edge in semantic_graph.edges():
                 x0, y0 = semantic_graph.nodes[edge[0]]['pos']
@@ -348,7 +327,7 @@ def main():
                     line=dict(width=width, color='#888'),
                     hoverinfo='none'
                 ))
-            
+        
             # Add nodes with varying sizes
             for node in semantic_graph.nodes():
                 x, y = semantic_graph.nodes[node]['pos']
@@ -358,20 +337,26 @@ def main():
                     x=[x],
                     y=[y],
                     mode='markers+text',
-                    marker=dict(size=10 + size/2),
+                    marker=dict(
+                        size=10 + size/2,
+                        color='#1f77b4',
+                        line=dict(width=2, color='white')
+                    ),
                     text=[node],
                     textposition="top center",
                     hoverinfo='text'
                 ))
-            
+        
             network_fig.update_layout(
                 showlegend=False,
                 hovermode='closest',
                 margin=dict(b=0,l=0,r=0,t=0),
                 xaxis=dict(showgrid=False, zeroline=False, showticklabels=False),
-                yaxis=dict(showgrid=False, zeroline=False, showticklabels=False)
+                yaxis=dict(showgrid=False, zeroline=False, showticklabels=False),
+                plot_bgcolor='white'
             )
-        
+            
+            st.plotly_chart(network_fig)    
         with tab5:
             st.subheader("Advanced NLP Analysis")