Update src/streamlit_app.py

src/streamlit_app.py

@@ -66,8 +66,10 @@ category_mapping = {
     "Temporal & Events": ["event", "date"],
     "Digital & Products": ["platform", "product", "media_type", "url"],
 }
+
+# FIX: Corrected the dictionary comprehension to avoid redundant iteration variable (preventing UnboundLocalError)
 reverse_category_mapping = {label: category
-for category, label_list in category_mapping.items() for label in label in label_list}
+for category, label_list in category_mapping.items() for label in label_list}
 
 
 # --- Utility Functions for Analysis and Plotly ---
@@ -251,7 +253,7 @@ def generate_network_graph(df, raw_text):
         showlegend=False,
         marker=dict(
             size=unique_entities['frequency'] * 5 + 10,
-            color=[entity_color_map.get(label, '#cccccc') for label in unique_entities['label']],
+            color=[entity_color_map.get(label, '#cccccc') for label in unique_entities['label']], # 'label' is safe here (local to list comprehension)
             line_width=1,
             line_color='black',
             opacity=0.9
@@ -269,7 +271,7 @@ def generate_network_graph(df, raw_text):
     legend_traces = []
     seen_labels = set()
     for index, row in unique_entities.iterrows():
-        label = row['label']
+        label = row['label'] # 'label' is defined here
         if label not in seen_labels:
             seen_labels.add(label)
             color = entity_color_map.get(label, '#cccccc')
@@ -830,7 +832,3 @@ if st.session_state.show_results:
             height=800,
             scrolling=True
         )
-
-
-
-