Update app.py

app.py

@@ -2,90 +2,13 @@ import gradio as gr
 from llm_loader import load_model
 from processing import process_input
 from transcription_diarization import diarize_audio
+from visualization import create_charts
 import time
 from config import openai_api_key
-import plotly.graph_objs as go
-from plotly.subplots import make_subplots
 
 # Load the model
 llm = load_model(openai_api_key)
 
-def create_charts(results):
-    charts = {}
-    explanations = {}
-    general_impressions = {}
-    
-    for speaker_id, data in results.items():
-        charts[speaker_id] = {}
-        explanations[speaker_id] = {}
-        
-        # Extract general impression
-        general_impressions[speaker_id] = data.get('general_impression', "No general impression provided.")
-
-        # Attachment Styles
-        attachment_data = data['attachments']
-        labels = ['Secured', 'Anxious-Preoccupied', 'Dismissive-Avoidant', 'Fearful-Avoidant']
-        values = [getattr(attachment_data, 'secured', 0), 
-                  getattr(attachment_data, 'anxious_preoccupied', 0), 
-                  getattr(attachment_data, 'dismissive_avoidant', 0), 
-                  getattr(attachment_data, 'fearful_avoidant', 0)]
-        colors = ['blue', 'orange', 'green', 'red']
-        
-        fig = go.Figure(data=[go.Bar(x=labels, y=values, marker_color=colors)])
-        fig.update_layout(title=f'{speaker_id}: Attachment Styles', yaxis_range=[0, 1])
-        charts[speaker_id]['attachment'] = fig
-        explanations[speaker_id]['attachment'] = attachment_data.explanation
-        
-        # Big Five Traits
-        bigfive_data = data['bigfive']
-        labels = ['Extraversion', 'Agreeableness', 'Conscientiousness', 'Neuroticism', 'Openness']
-        values = [bigfive_data.extraversion, bigfive_data.agreeableness, 
-                  bigfive_data.conscientiousness, bigfive_data.neuroticism, bigfive_data.openness]
-        colors = ['blue', 'green', 'red', 'purple', 'orange']
-        
-        fig = go.Figure(data=[go.Bar(x=labels, y=values, marker_color=colors)])
-        fig.update_layout(title=f'{speaker_id}: Big Five Traits', yaxis_range=[0, 10])
-        charts[speaker_id]['bigfive'] = fig
-        explanations[speaker_id]['bigfive'] = bigfive_data.explanation
-        
-        # Personality Disorders
-        personality_data = data['personalities']
-        labels = ['Depressed', 'Paranoid', 'Schizoid-Schizotypal', 'Antisocial-Psychopathic',
-                  'Borderline-Dysregulated', 'Narcissistic', 'Anxious-Avoidant', 'Dependent-Victimized', 'Obsessional']
-        values = [personality_data.depressed, personality_data.paranoid,
-                  personality_data.schizoid_schizotypal, personality_data.antisocial_psychopathic,
-                  personality_data.borderline_dysregulated, personality_data.narcissistic,
-                  personality_data.anxious_avoidant, personality_data.dependent_victimized,
-                  personality_data.obsessional]
-        colors = ['gray', 'purple', 'gold', 'black', 'red', 'orange', 'green', 'cyan', 'brown']
-        
-        fig = go.Figure(data=[go.Bar(x=labels, y=values, marker_color=colors)])
-        fig.update_layout(title=f'{speaker_id}: Personality Disorders', yaxis_range=[0, 5])
-        charts[speaker_id]['personality'] = fig
-        explanations[speaker_id]['personality'] = personality_data.explanation
-
-        # Attachment Dimensions (Radar Chart)
-        dimensions_data = data['attachments']
-        labels = ['Self', 'Others', 'Anxiety', 'Avoidance']
-        values = [dimensions_data.self_model, dimensions_data.others_model, 
-                  dimensions_data.anxiety, dimensions_data.avoidance]
-        
-        fig = go.Figure(data=go.Scatterpolar(
-          r=values,
-          theta=labels,
-          fill='toself'
-        ))
-        fig.update_layout(
-          polar=dict(
-            radialaxis=dict(visible=True, range=[0, 10])
-          ),
-          showlegend=False,
-          title=f'{speaker_id}: Attachment Dimensions'
-        )
-        charts[speaker_id]['dimensions'] = fig
-    
-    return charts, explanations, general_impressions
-
 def analyze_video(video_path, progress=gr.Progress()):
     start_time = time.time()
     if not video_path: