첨부된 화일의 확장자가 jpg나 png 일 경우 본문에서 다운로드 버튼이 뜨지 않도록 수정하고 싶습니다.

<script>
function checkFileTypeAndHide(elementId, fileName) {
var fileExtension = fileName.split('.').pop().toLowerCase();
if (fileExtension === 'jpg' || fileExtension === 'png') {
var element = document.getElementById(elementId);
if (element) {
element.style.display = 'none';
}
}
}
</script>
<!--// 첨부파일 목록 -->
<!--@foreach($oDocument->getUploadedFiles() as $key => $file)-->
<li style="list-style: none;">
<div class="file_info" id="file_info_{{$key}}">
<div class="file_name">
{$file->source_filename}
</div>
<p>{FileHandler::filesize($file->file_size)}</p>

{@
if(!$is_download_able):
$download_action = 'onclick="show_download_info(); return false;"';
endif;
}
<div class="btn_download_product" {$download_action}>
<a href="{getUrl('')}{$file->download_url}">
다운로드 받기
</a>
</div>
</div>
</li>
<script>
checkFileTypeAndHide('file_info_{{$key}}', '{{$file->source_filename}}');
</script>
<!--@endforeach-->