Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from llm_loader import load_model
 from processing import process_input
 from transcription_diarization import process_video
-import plotly.graph_objects as go
+from visualization import create_charts
 import os
 import time
 from config import hf_token, openai_api_key
@@ -21,79 +21,6 @@ LANGUAGE_MAP = {
     "Arabic": "ar"
 }
 
-def create_charts(results):
-    charts = {}
-    explanations = {}
-    
-    speaker_id = "Speaker 1"  # Default speaker ID
-    speaker_charts = {}
-    speaker_explanations = {}
-
-    # Attachment Styles
-    attachment_data = results.get('attachments')
-    if attachment_data:
-        fig_attachment = go.Figure(go.Bar(
-            x=['Secured', 'Anxious-Preoccupied', 'Dismissive-Avoidant', 'Fearful-Avoidant'],
-            y=[getattr(attachment_data, 'secured', 0), getattr(attachment_data, 'anxious_preoccupied', 0),
-               getattr(attachment_data, 'dismissive_avoidant', 0), getattr(attachment_data, 'fearful_avoidant', 0)],
-            marker_color=['blue', 'orange', 'green', 'red']
-        ))
-        fig_attachment.update_layout(title_text=f"Attachment Styles - {speaker_id}", showlegend=False)
-        speaker_charts["attachment"] = fig_attachment
-        speaker_explanations["attachment"] = getattr(attachment_data, 'explanation', '')
-
-        # Attachment Dimensions (Radar Chart)
-        fig_dimensions = go.Figure(go.Scatterpolar(
-            r=[getattr(attachment_data, 'avoidance', 0), getattr(attachment_data, 'anxiety', 0), 
-               getattr(attachment_data, 'self_rating', 0), getattr(attachment_data, 'others_rating', 0)],
-            theta=['Avoidance', 'Anxiety', 'Self', 'Others'],
-            fill='toself'
-        ))
-        fig_dimensions.update_layout(title_text=f"Attachment Dimensions - {speaker_id}", showlegend=False)
-        speaker_charts["dimensions"] = fig_dimensions
-
-    # Big Five Traits
-    bigfive_data = results.get('bigfive')
-    if bigfive_data:
-        fig_bigfive = go.Figure(go.Bar(
-            x=['Extraversion', 'Agreeableness', 'Conscientiousness', 'Neuroticism', 'Openness'],
-            y=[getattr(bigfive_data, 'extraversion', 0), getattr(bigfive_data, 'agreeableness', 0),
-               getattr(bigfive_data, 'conscientiousness', 0), getattr(bigfive_data, 'neuroticism', 0), 
-               getattr(bigfive_data, 'openness', 0)],
-            marker_color=['blue', 'green', 'red', 'purple', 'orange']
-        ))
-        fig_bigfive.update_layout(title_text=f"Big Five Traits - {speaker_id}", showlegend=False)
-        speaker_charts["bigfive"] = fig_bigfive
-        speaker_explanations["bigfive"] = getattr(bigfive_data, 'explanation', '')
-
-    # Personality Disorders
-    personality_data = results.get('personalities')
-    if personality_data:
-        fig_personality = go.Figure(go.Bar(
-            x=['Antisocial', 'Narcissistic', 'Depressed', 'Anxious-Avoidant',
-               'Obsessive', 'Paranoid', 'Borderline', 'Dependent', 'Schizoid-Schizotypal'],
-            y=[getattr(personality_data, 'antisocial_psychopathic', 0), getattr(personality_data, 'narcissistic', 0),
-               getattr(personality_data, 'depressed', 0), getattr(personality_data, 'anxious_avoidant', 0),
-               getattr(personality_data, 'obsessional', 0), getattr(personality_data, 'paranoid', 0),
-               getattr(personality_data, 'borderline_dysregulated', 0), getattr(personality_data, 'dependent_victimized', 0),
-               getattr(personality_data, 'schizoid_schizotypal', 0)],
-            marker_color=['black', 'orange', 'gray', 'green', 'brown', 'purple', 'red', 'cyan', 'magenta']
-        ))
-        fig_personality.update_layout(title_text=f"Personality Disorders - {speaker_id}", showlegend=False)
-        speaker_charts["personality"] = fig_personality
-        speaker_explanations["personality"] = getattr(personality_data, 'explanation', '')
-
-    # Update all charts to take full width
-    for fig in speaker_charts.values():
-        fig.update_layout(height=400, width=None, margin=dict(l=50, r=50, t=100, b=50))
-
-    # Store the charts and explanations
-    charts[speaker_id] = speaker_charts
-    explanations[speaker_id] = speaker_explanations
-
-    return charts, explanations
-
-
 def analyze_video(video_path, language_display_name, max_speakers, progress=gr.Progress()):
     start_time = time.time()
 
@@ -132,11 +59,6 @@ def analyze_video(video_path, language_display_name, max_speakers, progress=gr.P
     end_time = time.time()
     execution_time = end_time - start_time
 
-    # Prepare outputs for each speaker
-    output_components = []
-    for speaker_id, speaker_charts in charts.items():
-        speaker_explanations = explanations[speaker_id]
-
     # Prepare outputs for each speaker
     output_components = []
     for speaker_id, speaker_charts in charts.items():