Update visualization.py

visualization.py

@@ -1,6 +1,5 @@
 import gradio as gr
 import plotly.graph_objs as go
-import re
 
 def extract_data_and_explanation(text):
     speakers_data = {}
@@ -49,4 +48,75 @@ def extract_data_and_explanation(text):
     if current_speaker and explanation:
         speakers_data[current_speaker]["explanation"] = explanation
     
-    return speakers_data
+    return speakers_data
+
+def create_bar_chart(data, title):
+    fig = go.Figure(data=[go.Bar(
+        x=list(data.keys()), 
+        y=list(data.values()),
+        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")
+    return fig
+
+def create_radar_chart(data, title):
+    values = [data.get('Self', 0), data.get('Avoidance', 0), data.get('Others', 0), data.get('Anxiety', 0)]
+    fig = go.Figure(data=go.Scatterpolar(
+        r=values,
+        theta=['Self', 'Avoidance', 'Others', 'Anxiety'],
+        fill='toself'
+    ))
+    fig.update_layout(
+        polar=dict(
+            radialaxis=dict(visible=True, range=[0, max(values + [10])])
+        ),
+        showlegend=False,
+        title=title
+    )
+    return fig
+
+def update_visibility_and_charts(status, exec_time, lang, attachments, bigfive, personalities):
+    outputs = [
+        gr.update(value=status, visible=True),
+        gr.update(value=exec_time, visible=True),
+        gr.update(value=lang, visible=True),
+    ]
+    
+    for analysis_text, analysis_type in [(attachments, "Attachments"), (bigfive, "Big Five"), (personalities, "Personalities")]:
+        speakers_data = extract_data_and_explanation(analysis_text)
+        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"]}
+                    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()):  # Only create radar chart if there's data
+                        radar_fig = create_radar_chart(radar_data, f"Anxiety-Avoidance-Self-Others - {speaker}")
+                        outputs.append(gr.update(value=radar_fig, visible=True))
+                    else:
+                        outputs.append(gr.update(visible=False))
+                else:
+                    chart_data = {k: v for k, v in data.items() if k not in ["explanation"] 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))
+                    outputs.append(gr.update(visible=False))  # Placeholder for consistency
+                
+                explanation = data.get("explanation", "No explanation provided.")
+                outputs.append(gr.update(value=explanation, visible=True, label=f"{analysis_type} Explanation - {speaker}"))
+            else:
+                outputs.extend([gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)])
+    
+    # Pad the outputs to ensure we always return the same number of outputs
+    while len(outputs) < 21:  # 3 initial outputs + 6 plots + 6 radar charts + 6 explanations
+        outputs.append(gr.update(visible=False))
+    
+    print("Outputs generated:", outputs)
+    return outputs