okay can you change the output on the repository to be CrudRepository<> and inside that instad of long take the type from the field with name id

script.js

@@ -128,10 +128,18 @@ document.addEventListener('DOMContentLoaded', function() {
         // Generate Repository
         let repositoryCode = `package com.example.repository;\n\n`;
         repositoryCode += `import com.example.entity.${className};\n`;
-        repositoryCode += `import org.springframework.data.jpa.repository.JpaRepository;\n`;
+        repositoryCode += `import org.springframework.data.repository.CrudRepository;\n`;
         repositoryCode += `import org.springframework.stereotype.Repository;\n\n`;
+        
+        // Find ID type (default to Long if not found)
+        let idType = 'Long';
+        const idField = fields.find(f => f.name.toLowerCase() === 'id');
+        if (idField) {
+            idType = idField.type;
+        }
+        
         repositoryCode += `@Repository\n`;
-        repositoryCode += `public interface ${className}Repository extends JpaRepository<${className}, Long> {\n\n`;
+        repositoryCode += `public interface ${className}Repository extends CrudRepository<${className}, ${idType}> {\n\n`;
         repositoryCode += `}\n`;
 // Generate Service
         let serviceCode = `package com.example.accessingdatajpa;\n\n`;