Update src/streamlit_app.py

src/streamlit_app.py

@@ -177,8 +177,10 @@ def create_topic_word_bubbles(df_topic_data):
 
 def generate_network_graph(df, raw_text, entity_color_map):
     """Generates a network graph visualization (Node Plot) with edges based on entity co-occurrence in sentences."""
+    df = df.reset_index(drop=True)
     entity_counts = df['text'].value_counts().reset_index()
     entity_counts.columns = ['text', 'frequency']
+    representative_entities = df.sort_values('score', ascending=False).drop_duplicates(subset=['text'])[['text', 'label', 'score']]
     unique_entities = df.drop_duplicates(subset=['text', 'label']).merge(entity_counts, on='text')
     
     if unique_entities.shape[0] < 2: