diff --git a/ballerina/build.gradle b/ballerina/build.gradle
index a3529a2137..10366d3f83 100644
--- a/ballerina/build.gradle
+++ b/ballerina/build.gradle
@@ -80,6 +80,19 @@ task unpackDevTools(type: Copy) {
     }
 }
 
+task downloadDocUi {
+    def response = new JsonSlurper().parseText(new URL(docUiApi).text)
+    def zipFileUrl = response.fileURL
+    def zipInputStream = new URL(zipFileUrl).openStream()
+    def outputFile = new File("${buildDir}/target/extracted-distributions/doc-ui-zip/ballerina-doc-ui.zip")
+    doLast {
+        outputFile.parentFile.mkdirs()
+        outputFile.withOutputStream { outputStream ->
+            outputStream << zipInputStream
+        }
+    }
+}
+
 task unpackAwsLambdaBala(type: Copy) {
     group = "unpack_dependencies"
     configurations.awsLambdaBala.resolvedConfiguration.resolvedArtifacts.each { artifact ->
@@ -271,8 +284,9 @@ task copyDevToolsCoverageReport(type: Copy) {
     into "$project.buildDir/target/extracted-distributions/jballerina-tools-zip/jballerina-tools-${ballerinaLangVersion}/lib/tools/coverage"
     rename("testerina-report-tools-${devToolsVersion}.zip", "report.zip")
 }
+
 task copyDevToolsDocUi(type: Copy) {
-    from zipTree("$project.buildDir/target/extracted-distributions/ballerina-dev-tools-zip/docs/docerina-ui-${devToolsVersion}.zip")
+    from zipTree("$project.buildDir/target/extracted-distributions/doc-ui-zip/ballerina-doc-ui.zip")
     into "$project.buildDir/target/extracted-distributions/jballerina-tools-zip/jballerina-tools-${ballerinaLangVersion}/lib/tools/doc-ui"
 }
 
@@ -1502,6 +1516,7 @@ extractJreForWindows.dependsOn extractJreForMacArm
 copyOtherRepos.dependsOn extractJreForWindows
 copyDevToolsCoverageReport.dependsOn copyOtherRepos
 copyDevToolsDocUi.dependsOn copyDevToolsCoverageReport
+copyDevToolsDocUi.dependsOn downloadDocUi
 buildDistRepo.dependsOn copyDevToolsDocUi
 buildDistRepo.dependsOn generateCache
 filterApiDocs.dependsOn buildDistRepo
diff --git a/gradle.properties b/gradle.properties
index 09ec6272ca..e582cbeac1 100644
--- a/gradle.properties
+++ b/gradle.properties
@@ -79,6 +79,9 @@ persistToolVersion=1.2.0-20230908-152500-2ba8c45
 devToolsVersion=1.2.1-20230914-153500-664cd40
 ballerinaCommandVersion=1.4.2
 
+# API Doc UI
+docUiApi=https://api.dev-central.ballerina.io/2.0/docs/doc-ui
+
 # GraphQL Tool
 graphqlVersion=0.8.1