Update app.py

app.py

@@ -123,6 +123,26 @@ def easyphoto_infer(model_id, selected_template_images, additional_prompt, seed,
     model_id = str(model_id).strip()
     print('model id: ' + model_id)
 
+    if job_id is None:
+        output_info = 'Status: no job id provided, please do model training first!'
+        return output_info, []
+    
+    job_id = str(job_id).strip()
+    print('job id: ' + job_id)
+
+    check_client = ApiClient(host, appId, token)
+    check_api = AiServiceJobApi(check_client)
+    try:
+        job_id = int(str(job_id).strip())
+        response = check_api.get_async_job_with_id(job_id)
+        message = response.data['job']['message']
+        if not message == 'success':
+            output = 'Status: model training incomplete!'
+            return output, []
+    except:
+            output = 'Status: checking training status failed, please do model training first!'
+            return output, []
+
     for image_url in image_urls:
         try:
             params = {
@@ -161,6 +181,7 @@ def easyphoto_infer(model_id, selected_template_images, additional_prompt, seed,
 
 with gr.Blocks() as easyphoto_demo:
     model_id = gr.Textbox(visible=False)
+    job_id = gr.Textbox(visible=False)
     with gr.TabItem('Training'):
         with gr.Blocks():
             with gr.Row():
@@ -184,7 +205,6 @@ with gr.Blocks() as easyphoto_demo:
                         '''
                     )
 
-            job_id = gr.Textbox(visible=False)
             with gr.Row():
                 run_button = gr.Button('Submit My Training Job')
                 check_button = gr.Button('Check My Training Job Status')