Update output_parser.py

output_parser.py

@@ -54,12 +54,44 @@ class OutputParser:
             raise ValueError(f"Error parsing output: {str(e)}")
 
     def parse_speaker_analysis(self, obj: dict) -> SpeakerAnalysis:
+        attachment_styles = obj.get("Attachment Styles", {})
+        big_five_traits = obj.get("Big Five Traits", {})
+        personality_disorders = obj.get("Personality Disorders", {})
+
         return SpeakerAnalysis(
             speaker=obj.get("Speaker", "Unknown Speaker"),
             general_impression=obj.get("General Impression", "No general impression provided"),
-            attachment_style=AttachmentStyle(**(obj.get("Attachment Styles", {}))),
-            big_five_traits=BigFiveTraits(**(obj.get("Big Five Traits", {}))),
-            personality_disorder=PersonalityDisorder(**(obj.get("Personality Disorders", {})))
+            attachment_style=AttachmentStyle(
+                secured=attachment_styles.get("Secured", 0.0),
+                anxious_preoccupied=attachment_styles.get("Anxious-Preoccupied", 0.0),
+                dismissive_avoidant=attachment_styles.get("Dismissive-Avoidant", 0.0),
+                fearful_avoidant=attachment_styles.get("Fearful-Avoidant", 0.0),
+                self_model=attachment_styles.get("Self", 0),
+                others_model=attachment_styles.get("Others", 0),
+                anxiety=attachment_styles.get("Anxiety", 0),
+                avoidance=attachment_styles.get("Avoidance", 0),
+                explanation=attachment_styles.get("Explanation", "No explanation provided")
+            ),
+            big_five_traits=BigFiveTraits(
+                extraversion=big_five_traits.get("Extraversion", 0),
+                agreeableness=big_five_traits.get("Agreeableness", 0),
+                conscientiousness=big_five_traits.get("Conscientiousness", 0),
+                neuroticism=big_five_traits.get("Neuroticism", 0),
+                openness=big_five_traits.get("Openness", 0),
+                explanation=big_five_traits.get("Explanation", "No explanation provided")
+            ),
+            personality_disorder=PersonalityDisorder(
+                depressed=personality_disorders.get("Depressed", 0),
+                paranoid=personality_disorders.get("Paranoid", 0),
+                schizoid_schizotypal=personality_disorders.get("Schizoid-Schizotypal", 0),
+                antisocial_psychopathic=personality_disorders.get("Antisocial-Psychopathic", 0),
+                borderline_dysregulated=personality_disorders.get("Borderline-Dysregulated", 0),
+                narcissistic=personality_disorders.get("Narcissistic", 0),
+                anxious_avoidant=personality_disorders.get("Anxious-Avoidant", 0),
+                dependent_victimized=personality_disorders.get("Dependent-Victimized", 0),
+                obsessional=personality_disorders.get("Obsessional", 0),
+                explanation=personality_disorders.get("Explanation", "No explanation provided")
+            )
         )
 
 output_parser = OutputParser()