Update app.py

app.py

@@ -256,7 +256,7 @@ def certification(hf_username, first_name, last_name):
 
   certificate, message = verify_certification(results_certification, hf_username, first_name, last_name)
   print("MESSAGE", message)
-  return message, df, certificate
+  return message, pdf, df, certificate
 
 """
 Verify that the user pass.
@@ -281,7 +281,7 @@ def verify_certification(df, hf_username, first_name, last_name):
   
   if pass_percentage == 100:
     # Generate a certificate of excellence
-    certificate = generate_certificate("./certificate_models/certificate-excellence.png", first_name, last_name)
+    certificate, pdf = generate_certificate("./certificate_models/certificate-excellence.png", first_name, last_name)
 
     # Add this user to our database
     add_certified_user(hf_username, first_name, last_name, pass_percentage)
@@ -294,7 +294,7 @@ def verify_certification(df, hf_username, first_name, last_name):
 
   elif pass_percentage < 100 and pass_percentage >= 80:
     # Certificate of completion
-    certificate = generate_certificate("./certificate_models/certificate-completion.png", first_name, last_name)
+    certificate, pdf = generate_certificate("./certificate_models/certificate-completion.png", first_name, last_name)
 
     # Add this user to our database
     add_certified_user(hf_username, first_name, last_name, pass_percentage)
@@ -309,6 +309,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 = ""
       
     # Add a message
     message = """
@@ -316,7 +317,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
+  return certificate, message, pdf
     
 
 def generate_certificate(certificate_model, first_name, last_name):
@@ -342,9 +343,9 @@ def generate_certificate(certificate_model, first_name, last_name):
     # Date of certification
     d.text((1480, 1170), str(date.today()), fill="black", anchor="mm", font=date_font)
   
-    im.save("certificate_"+".png")
+    pdf = im.save("certificate_"+".png")
 
-    return im
+    return im, pdf
 
 
 def add_certified_user(hf_username, first_name, last_name, pass_percentage):
@@ -383,8 +384,9 @@ with gr.Blocks() as demo:
     #email = gr.Textbox(placeholder="jane.doe@gmail.com", label="Your Email (to receive your certificate)")
     check_progress_button = gr.Button(value="Check if I pass")
     output_text = gr.components.Textbox()
-    output_pdf = gr.components.Image(type="pil")
+    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_dataframe, output_pdf])#[output1, output2])
+    check_progress_button.click(fn=certification, inputs=[hf_username, first_name, last_name], outputs=[output_text, output_dataframe, output_pdf, output_img])#[output1, output2])
 
 demo.launch(debug=True)