Update app.py

app.py

@@ -254,12 +254,16 @@ def certification(hf_username, first_name, last_name):
 
   df = df1[['passed', 'unit', 'env', 'min_result', 'best_result', 'best_model_id']]
 
-  certificate, message, pdf = verify_certification(results_certification, hf_username, first_name, last_name)
+  certificate, message, pdf, pass_ = verify_certification(results_certification, hf_username, first_name, last_name)
   print("MESSAGE", message)
 
-  
+  if pass_:
+    visible = True
+  else:
+    visible = False
 
-  return message, pdf, certificate, df #, output_row.update(visible=True)
+  
+  return message, pdf, certificate, df, output_row.update(visible=visible) 
 
 """
 Verify that the user pass.
@@ -275,6 +279,8 @@ def verify_certification(df, hf_username, first_name, last_name):
   # Check that we pass
   model_pass_nb = 0
   pass_percentage = 0
+  pass_ = False
+
   for unit in df:
     if unit["passed_"] is True:
       model_pass_nb += 1
@@ -283,6 +289,7 @@ def verify_certification(df, hf_username, first_name, last_name):
   print("pass_percentage", pass_percentage)
   
   if pass_percentage == 100:
+    pass_ = True
     # Generate a certificate of excellence
     certificate, pdf = generate_certificate("./certificate_models/certificate-excellence.png", first_name, last_name)
 
@@ -298,6 +305,7 @@ def verify_certification(df, hf_username, first_name, last_name):
     """
 
   elif pass_percentage < 100 and pass_percentage >= 80:
+    pass_ = True
     # Certificate of completion
     certificate, pdf = generate_certificate("./certificate_models/certificate-completion.png", first_name, last_name)
 
@@ -316,7 +324,7 @@ def verify_certification(df, hf_username, first_name, last_name):
   else:
     # Not pass yet
     certificate = Image.new("RGB", (100, 100), (255, 255, 255))
-    pdf = ""
+    certificate, pdf = generate_certificate(certificate.filename, first_name, last_name)
       
     # Add a message
     message = """
@@ -324,7 +332,7 @@ def verify_certification(df, hf_username, first_name, last_name):
     Check below which units you need to do again to get your certificate 💪
     """
   print("return certificate")
-  return certificate, message, pdf
+  return certificate, message, pdf, pass_
     
 
 def generate_certificate(certificate_model, first_name, last_name):
@@ -396,12 +404,12 @@ with gr.Blocks() as demo:
     last_name = gr.Textbox(placeholder="Doe", label="Your Last Name")
     #email = gr.Textbox(placeholder="[email protected]", label="Your Email (to receive your certificate)")
     check_progress_button = gr.Button(value="Check if I pass")
-    #with gr.Row(visible=False) as output_row:
     output_text = gr.components.Textbox()
-    output_pdf = gr.File()
-    output_img = gr.components.Image(type="pil")
+    with gr.Row(visible=True) as output_row:
+        output_pdf = gr.File()
+        output_img = gr.components.Image(type="pil")
     output_dataframe = gr.components.Dataframe(headers=["Pass?", "Unit", "Environment", "Baseline", "Your best result", "Your best model id"], datatype=["markdown", "markdown", "markdown", "number", "number", "markdown", "bool"]) #value= certification(hf_username, first_name, last_name), 
-    check_progress_button.click(fn=certification, inputs=[hf_username, first_name, last_name], outputs=[output_text, output_pdf, output_img, output_dataframe])#, output_row])#[output1, output2])
+    check_progress_button.click(fn=certification, inputs=[hf_username, first_name, last_name], outputs=[output_text, output_pdf, output_img, output_dataframe, output_row])#[output1, output2])
 
     
 demo.launch(debug=True)