Refactor gen_ssl function to include email status

app.py

@@ -12,12 +12,12 @@ def gen_ssl(i_domains, wildcard, email, ca_server, key_type, key_size=None, key_
         key_curve = None
     if key_size is not None:
         key_size = int(key_size)
-    pvt, csr, cert = main(i_domains, wildcard, email, ca_server, key_type, key_size, key_curve)
+    pvt, csr, cert, email_status = main(i_domains, wildcard, email, ca_server, key_type, key_size, key_curve)
     print("SSL Certificate generated successfully")
     try:
-        return pvt.decode('utf-8'), csr.decode('utf-8'), cert.decode('utf-8')
+        return pvt.decode('utf-8'), csr.decode('utf-8'), cert.decode('utf-8'), email_status
     except:
-        return pvt, csr, cert
+        return pvt, csr, cert, email_status
 
 def update_key_options(key_type):
     if key_type == "rsa":
@@ -60,7 +60,8 @@ def app():
                 csr = gr.Textbox(label="Your CSR", placeholder="Your CSR will appear here, after successful SSL generation", type="text", interactive=False, show_copy_button=True, lines=10, max_lines=10)
             with gr.Column():
                 crt = gr.Textbox(label="Your SSL Certificate", placeholder="Your SSL Certificate will appear here, after successful SSL generation", type="text", interactive=False, show_copy_button=True, lines=10, max_lines=10)
-        btn.click(gen_ssl, inputs=[domains_input, wildcard, email_input, ca_server, key_type, key_size_dropdown, key_curve_dropdown], outputs=[pvt, csr, crt])
+        email_status = gr.Textbox(label="Email Status", placeholder="Email status will appear here, after sending email", type="text", interactive=False)
+        btn.click(gen_ssl, inputs=[domains_input, wildcard, email_input, ca_server, key_type, key_size_dropdown, key_curve_dropdown], outputs=[pvt, csr, crt, email_status])
     try:
         webui.queue(default_concurrency_limit=15).launch()
     except Exception as e:

main.py

@@ -126,8 +126,8 @@ def main(i_domains, wildcard, email, ca_server, key_type, key_size=None, key_cur
             cf_non_wildcard(verification_tokens, email, exchange)
     except Exception as e:
         print(f"Error adding TXT records: {e}")
-    for i in range(60):
-        print(f"Waiting for {60-i} seconds", end="\r")
+    for i in range(30):
+        print(f"Waiting for {30-i} seconds", end="\r")
         time.sleep(1)
     while not verify_txt_records(verification_tokens, exchange):
         print("TXT records not verified yet")
@@ -151,8 +151,11 @@ def main(i_domains, wildcard, email, ca_server, key_type, key_size=None, key_cur
     Key Size = {key_size}
     For more details, visit: https://projectgatekeeper.vercel.app/tool/decode.html
     """
-    send_email(email, private_key, csr, cert, generation_details)
-    return private_key, csr, cert
+    if send_email(email, private_key, csr, cert, generation_details):
+        email_status = f"Email Sent to {email}"
+    else:
+        email_status = f"Can't sent email to {email}"
+    return private_key, csr, cert, email_status
 
 if __name__ == "__main__":
     DOMAINS = 'raannakasturi.eu.org'