Update visualization.py

visualization.py

@@ -9,8 +9,9 @@ def extract_data_and_explanation(text):
         line = line.strip()
         if line.startswith("-------"):
             if current_speaker and explanation:
-                speakers_data[current_speaker]["explanation"] = explanation
+                speakers_data[current_speaker]["explanation"] = explanation.strip()
             explanation = ""
+            current_speaker = None
             continue
         if line.startswith("Speaker"):
             current_speaker = line.split(':')[0].strip()
@@ -20,7 +21,7 @@ def extract_data_and_explanation(text):
             key = key.strip()
             value = value.strip()
             if key.lower() == "explanation":
-                explanation = value
+                explanation += value + " "
             elif '|' in value:
                 for part in value.split('|'):
                     part = part.strip()
@@ -32,20 +33,17 @@ def extract_data_and_explanation(text):
                             sub_key, sub_value = parts
                         else:
                             continue
-                    try:
-                        speakers_data[current_speaker][sub_key.strip()] = float(sub_value.strip())
-                    except ValueError:
-                        speakers_data[current_speaker][sub_key.strip()] = sub_value.strip()
+                    speakers_data[current_speaker][sub_key.strip()] = float(sub_value.strip())
             else:
                 try:
                     speakers_data[current_speaker][key] = float(value)
                 except ValueError:
                     speakers_data[current_speaker][key] = value
         elif line and current_speaker:
-            speakers_data[current_speaker][line] = 0
+            explanation += line + " "
     
     if current_speaker and explanation:
-        speakers_data[current_speaker]["explanation"] = explanation
+        speakers_data[current_speaker]["explanation"] = explanation.strip()
 
     return speakers_data
 
@@ -53,9 +51,12 @@ def create_bar_chart(data, title):
     fig = go.Figure(data=[go.Bar(
         x=list(data.keys()), 
         y=list(data.values()),
+        text=list(data.values()),
+        textposition='auto',
         marker_color=['red', 'green', 'blue', 'yellow', 'purple', 'orange', 'pink', 'cyan', 'magenta', 'brown'][:len(data)]
     )])
     fig.update_layout(title=title, xaxis_title="Traits", yaxis_title="Score")
+    fig.update_xaxes(tickangle=45)
     return fig
 
 def create_radar_chart(data, title):
@@ -86,15 +87,15 @@ def update_visibility_and_charts(status, exec_time, lang, attachments, bigfive,
         for speaker, data in speakers_data.items():
             if data:
                 if analysis_type == "Attachments":
-                    chart_data = {k: v for k, v in data.items() if k in ["Secured", "Anxious-Preoccupied", "Dismissive-Avoidant", "Fearful-Avoidant"]}
+                    chart_data = {k: v for k, v in data.items() if k in ["Secured", "Anxious-Preoccupied", "Dismissive-Avoidant", "Fearful-Avoidant"] and isinstance(v, (int, float))}
                     if chart_data:
                         fig = create_bar_chart(chart_data, f"{analysis_type} Analysis - {speaker}")
                         outputs.append(gr.update(value=fig, visible=True))
                     else:
                         outputs.append(gr.update(visible=False))
                     
-                    radar_data = {k: v for k, v in data.items() if k in ["Anxiety", "Avoidance", "Self", "Others"]}
-                    if any(radar_data.values()):
+                    radar_data = {k: v for k, v in data.items() if k in ["Anxiety", "Avoidance", "Self", "Others"] and isinstance(v, (int, float))}
+                    if radar_data:
                         radar_fig = create_radar_chart(radar_data, f"Anxiety-Avoidance-Self-Others - {speaker}")
                         outputs.append(gr.update(value=radar_fig, visible=True))
                     else:
@@ -116,5 +117,4 @@ def update_visibility_and_charts(status, exec_time, lang, attachments, bigfive,
     while len(outputs) < 21:
         outputs.append(gr.update(visible=False))
     
-    print("Outputs generated:", outputs)
     return outputs