Update utility/script_generator.py

utility/script_generator.py

@@ -44,16 +44,7 @@ def generate_script(topic):
     # Extract the generated text
     generated_text = response[0]['generated_text']
 
-    try:
-        # Attempt to parse the generated text as JSON
-        script = json.loads(generated_text)["script"]
-    except json.JSONDecodeError:
-        print("JSONDecodeError. Attempting to extract JSON from the response.")
-        json_start = generated_text.find('{')
-        json_end = generated_text.rfind('}') + 1
-        script = json.loads(generated_text[json_start:json_end])["script"]
-    
-    return str(script)
+    return generated_text
 
 # # Example usage
 # topic = "Weird facts"