Update utils.py

utils.py

@@ -5,6 +5,8 @@ import os
 import tiktoken
 from gtts import gTTS
 import tempfile
+import json
+import re
 
 groq_client = Groq(api_key=os.environ["GROQ_API_KEY"])
 tokenizer = tiktoken.get_encoding("cl100k_base")
@@ -35,10 +37,27 @@ def generate_script(system_prompt: str, input_text: str, tone: str):
         temperature=0.7
     )
     
+    # Extract content and remove any markdown code block syntax
+    content = response.choices[0].message.content
+    content = re.sub(r'```json\s*|\s*```', '', content)
+    
     try:
-        dialogue = Dialogue.model_validate_json(response.choices[0].message.content)
+        # First, try to parse as JSON
+        json_data = json.loads(content)
+        dialogue = Dialogue.model_validate(json_data)
+    except json.JSONDecodeError as json_error:
+        # If JSON parsing fails, try to extract JSON from the text
+        match = re.search(r'\{.*\}', content, re.DOTALL)
+        if match:
+            try:
+                json_data = json.loads(match.group())
+                dialogue = Dialogue.model_validate(json_data)
+            except (json.JSONDecodeError, ValidationError) as e:
+                raise ValueError(f"Failed to parse dialogue JSON: {e}\nContent: {content}")
+        else:
+            raise ValueError(f"Failed to find valid JSON in the response: {content}")
     except ValidationError as e:
-        raise ValueError(f"Failed to parse dialogue JSON: {e}")
+        raise ValueError(f"Failed to validate dialogue structure: {e}\nContent: {content}")
     
     return dialogue