import plotly.graph_objects as go
from plotly.subplots import make_subplots

def create_charts(results):
    charts = {}
    explanations = {}

    # Loop through each detected speaker in the results
    for speaker_id, speaker_data in results['speakers'].items():
        speaker_charts = {}
        speaker_explanations = {}

        # Attachment Styles for each speaker
        attachment_data = speaker_data['attachments']
        fig_attachment = go.Figure(go.Bar(
            x=['Secured', 'Anxious-Preoccupied', 'Dismissive-Avoidant', 'Fearful-Avoidant'],
            y=[attachment_data.secured, attachment_data.anxious_preoccupied,
               attachment_data.dismissive_avoidant, attachment_data.fearful_avoidant],
            marker_color=['blue', 'orange', 'green', 'red']
        ))
        fig_attachment.update_layout(title_text=f"Attachment Styles - Speaker {speaker_id}", showlegend=False)
        speaker_charts["attachment"] = fig_attachment
        speaker_explanations["attachment"] = attachment_data.explanation

        # Attachment Dimensions (Radar Chart) for each speaker
        fig_dimensions = go.Figure(go.Scatterpolar(
            r=[attachment_data.avoidance, attachment_data.anxiety, attachment_data.self_rating, attachment_data.others_rating],
            theta=['Avoidance', 'Anxiety', 'Self', 'Others'],
            fill='toself'
        ))
        fig_dimensions.update_layout(title_text=f"Attachment Dimensions - Speaker {speaker_id}", showlegend=False)
        speaker_charts["dimensions"] = fig_dimensions

        # Big Five Traits for each speaker
        bigfive_data = speaker_data['bigfive']
        fig_bigfive = go.Figure(go.Bar(
            x=['Extraversion', 'Agreeableness', 'Conscientiousness', 'Neuroticism', 'Openness'],
            y=[bigfive_data.extraversion, bigfive_data.agreeableness,
               bigfive_data.conscientiousness, bigfive_data.neuroticism, bigfive_data.openness],
            marker_color=['blue', 'green', 'red', 'purple', 'orange']
        ))
        fig_bigfive.update_layout(title_text=f"Big Five Traits - Speaker {speaker_id}", showlegend=False)
        speaker_charts["bigfive"] = fig_bigfive
        speaker_explanations["bigfive"] = bigfive_data.explanation

        # Personality Disorders for each speaker
        personality_data = speaker_data['personalities']
        fig_personality = go.Figure(go.Bar(
            x=['Antisocial', 'Narcissistic', 'Depressed', 'Anxious-Avoidant',
               'Obsessive', 'Paranoid', 'Borderline', 'Dependent', 'Schizoid-Schizotypal'],
            y=[personality_data.antisocial_psychopathic, personality_data.narcissistic,
               personality_data.depressed, personality_data.anxious_avoidant,
               personality_data.obsessional, personality_data.paranoid,
               personality_data.borderline_dysregulated, personality_data.dependent_victimized,
               personality_data.schizoid_schizotypal],
            marker_color=['black', 'orange', 'gray', 'green', 'brown', 'purple', 'red', 'cyan', 'magenta']
        ))
        fig_personality.update_layout(title_text=f"Personality Disorders - Speaker {speaker_id}", showlegend=False)
        speaker_charts["personality"] = fig_personality
        speaker_explanations["personality"] = 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 for each speaker
        charts[speaker_id] = speaker_charts
        explanations[speaker_id] = speaker_explanations

    return charts, explanations