add model service check to chatbot

chatbot/ai_applications/chat_ui.py

@@ -1,5 +1,7 @@
 import gradio as gr
 from gradio_client import Client
+import requests
+import time
 import os
 
 
@@ -17,8 +19,22 @@ def ask(self, prompt, history):
                 yield r
         yield str(job.outputs()[-1])
 
+def checking_model_service(model_service):
+    print("Waiting for Model Service Availability...")
+    ready = False
+    while not ready:
+        try:
+            request = requests.get(f'{model_service}')
+            if request.status_code == 200:
+                ready = True
+        except:
+            pass
+        time.sleep(1) 
+    print("Model Service Available")
+
 if __name__ == "__main__":
     model_endpoint = os.getenv('MODEL_ENDPOINT', "http://0.0.0.0:7860")
+    checking_model_service(model_endpoint)
     chat = Chat(model_endpoint)
     demo = gr.ChatInterface(chat.ask)
     demo.launch(server_name="0.0.0.0", server_port=8080)