Update output_parser.py

output_parser.py

@@ -57,23 +57,18 @@ class OutputParser:
         attachment_styles = obj.get("Attachment Styles", {})
         big_five_traits = obj.get("Big Five Traits", {})
         personality_disorders = obj.get("Personality Disorders", {})
-
+    
         # Handle general_impression as a string
-        general_impression = obj.get("General Impression", "No general impression provided")
+        general_impression = obj.get("General Impression", {})
         if isinstance(general_impression, dict):
-            general_impression = json.dumps(general_impression)
-        elif isinstance(general_impression, str):
-            try:
-                # Check if it's a JSON string
-                json.loads(general_impression)
-                # If it parses successfully, it's likely a JSON string, so we'll keep it as is
-            except json.JSONDecodeError:
-                # If it's not a valid JSON string, we'll keep it as is (it's already a string)
-                pass
-
+            # Convert the dictionary to a formatted string
+            general_impression_str = "\n".join([f"{key}: {value}" for key, value in general_impression.items()])
+        else:
+            general_impression_str = str(general_impression)
+    
         return SpeakerAnalysis(
-            speaker=obj.get("Speaker", "Unknown Speaker"),
-            general_impression=general_impression,
+            speaker=str(obj.get("Speaker", "Unknown Speaker")),  # Convert to string
+            general_impression=general_impression_str,
             attachment_style=AttachmentStyle(
                 secured=attachment_styles.get("Secured", 0.0),
                 anxious_preoccupied=attachment_styles.get("Anxious-Preoccupied", 0.0),