Update app.py

app.py

@@ -311,24 +311,37 @@ def main():
             st.subheader("Semantic Network")
             semantic_graph = analyzer.create_semantic_network(text)
             
-            # Create the network visualization
             network_fig = go.Figure()
         
-            # Add edges with varying widths
+            # Add edges with enhanced visual encoding
             for edge in semantic_graph.edges():
                 x0, y0 = semantic_graph.nodes[edge[0]]['pos']
                 x1, y1 = semantic_graph.nodes[edge[1]]['pos']
-                width = semantic_graph.edges[edge]['width']
+                weight = semantic_graph.edges[edge]['weight']
+                max_weight = max(d['weight'] for _, _, d in semantic_graph.edges(data=True))
+                
+                # Normalize weight for visual encoding
+                normalized_weight = weight / max_weight
+                
+                # Enhanced width scaling (more pronounced differences)
+                width = 2 + (normalized_weight * 8)
+                
+                # Color gradient from light to dark based on weight
+                color = f'rgba(31, 119, 180, {0.3 + normalized_weight * 0.7})'
                 
                 network_fig.add_trace(go.Scatter(
                     x=[x0, x1, None],
                     y=[y0, y1, None],
                     mode='lines',
-                    line=dict(width=width, color='#888'),
-                    hoverinfo='none'
+                    line=dict(
+                        width=width,
+                        color=color
+                    ),
+                    hoverinfo='text',
+                    hovertext=f'Relationship strength: {weight:.2f}'
                 ))
         
-            # Add nodes with varying sizes
+            # Enhanced nodes with better visibility
             for node in semantic_graph.nodes():
                 x, y = semantic_graph.nodes[node]['pos']
                 size = semantic_graph.nodes[node]['size']
@@ -338,25 +351,31 @@ def main():
                     y=[y],
                     mode='markers+text',
                     marker=dict(
-                        size=10 + size/2,
-                        color='#1f77b4',
-                        line=dict(width=2, color='white')
+                        size=15 + size/2,  # Increased base size
+                        color='#ffffff',
+                        line=dict(width=2, color='#1f77b4'),
+                        symbol='circle'
                     ),
                     text=[node],
                     textposition="top center",
-                    hoverinfo='text'
+                    textfont=dict(size=12, color='black'),
+                    hoverinfo='text',
+                    hovertext=f'Term: {node}<br>Frequency: {size}'
                 ))
         
             network_fig.update_layout(
                 showlegend=False,
                 hovermode='closest',
-                margin=dict(b=0,l=0,r=0,t=0),
+                margin=dict(b=20, l=20, r=20, t=20),
                 xaxis=dict(showgrid=False, zeroline=False, showticklabels=False),
                 yaxis=dict(showgrid=False, zeroline=False, showticklabels=False),
-                plot_bgcolor='white'
+                plot_bgcolor='white',
+                width=800,
+                height=600
             )
             
-            st.plotly_chart(network_fig)    
+            st.plotly_chart(network_fig, use_container_width=True)
+    
         with tab5:
             st.subheader("Advanced NLP Analysis")