update gradio test & rebase

Signed-off-by: Mustafa <[email protected]>

Author: MSCetin37

CodeGen/tests/test_compose_on_gaudi.sh

@@ -211,6 +211,18 @@ function validate_frontend() {
     fi
 }
 
+function validate_gradio() {
+    local URL="http://${ip_address}:5173/health"
+    local HTTP_STATUS=$(curl "$URL")
+    local SERVICE_NAME="Gradio"
+
+    if [ "$HTTP_STATUS" = '{"status":"ok"}' ]; then
+        echo "[ $SERVICE_NAME ] HTTP status is 200. UI server is running successfully..."
+    else
+        echo "[ $SERVICE_NAME ] UI server has failed..."
+    fi            
+}
+
 function stop_docker() {
     local docker_profile="$1"
 
@@ -249,8 +261,8 @@ function main() {
 
         validate_microservices "${docker_llm_container_names[${i}]}"
         validate_megaservice
-        # validate_frontend
-
+        validate_gradio
+        
         stop_docker "${docker_compose_profiles[${i}]}"
         sleep 5s
     done

CodeGen/tests/test_compose_on_xeon.sh

@@ -210,6 +210,17 @@ function validate_frontend() {
     fi
 }
 
+function validate_gradio() {
+    local URL="http://${ip_address}:5173/health"
+    local HTTP_STATUS=$(curl "$URL")
+    local SERVICE_NAME="Gradio"
+
+    if [ "$HTTP_STATUS" = '{"status":"ok"}' ]; then
+        echo "[ $SERVICE_NAME ] HTTP status is 200. UI server is running successfully..."
+    else
+        echo "[ $SERVICE_NAME ] UI server has failed..."
+    fi            
+}
 
 function stop_docker() {
     local docker_profile="$1"
@@ -249,13 +260,13 @@ function main() {
 
         validate_microservices "${docker_llm_container_names[${i}]}"
         validate_megaservice
-        # validate_frontend
+        validate_gradio
 
         stop_docker "${docker_compose_profiles[${i}]}"
         sleep 5s
     done
 
-    # echo y | docker system prune
+    echo y | docker system prune
 }
 
 main

CodeGen/ui/gradio/codegen_ui_gradio.py

@@ -335,6 +335,9 @@ def get_file_names(files):
                 delete_all_button = gr.Button("Delete All", variant="primary", size="sm")
                 delete_all_button.click(delete_all_files, outputs=upload_status)
 
+@app.get("/health")
+def health_check():
+    return {"status": "ok"}
 
 ui.queue()
 app = gr.mount_gradio_app(app, ui, path="/")