Add email sending functionality and refactor post creation to include wait time parameter

app.py

@@ -2,9 +2,6 @@ import gradio as gr
 from main import post_blogpost
 from post_blog import test
 
-def clear_everything(title, category, summary, mindmap, citation, status, access_key):
-    return None, None, None, None, None, None, None
-
 theme = gr.themes.Soft(
     primary_hue="purple",
     secondary_hue="cyan",
@@ -25,9 +22,12 @@ with gr.Blocks(theme=theme, title="ReXplore Backend", fill_height=True) as app:
         """)
     with gr.Row():
         with gr.Column():
-            access_key = gr.Textbox(label="Access Key", placeholder="Enter the Access Key", type="password")
-            start = gr.Button(value="Start", variant="primary")
-            test_btn = gr.Button(value="Test")
+            with gr.Row():
+                access_key = gr.Textbox(label="Access Key", placeholder="Enter the Access Key", type="password")
+                wait_time = gr.Number(value=5, minimum=3, maximum=15, step=1, label="Wait Time", interactive=True)
+            with gr.Row():
+                start = gr.Button(value="Start", variant="primary")
+                test_btn = gr.Button(value="Test", variant="secondary")
         status = gr.Textbox(label="Data", placeholder="Enter the data", lines=7)
     start.click(
         post_blogpost,
@@ -46,6 +46,6 @@ with gr.Blocks(theme=theme, title="ReXplore Backend", fill_height=True) as app:
         scroll_to_output=True,
         show_api=True,
         api_name="rexplore_backend_test",
-        show_progress="full",)
+        show_progress="full")
 
 app.queue(default_concurrency_limit=25).launch(show_api=True, show_error=True, debug=True)

main.py

@@ -10,7 +10,10 @@ dotenv.load_dotenv()
 summarizer_api_key = os.getenv("SUMMARIZER_API_KEY")
 access_key = os.getenv("ACCESS_KEY")
 
-def paper_data(paper_data):
+def send_mail():
+    return "Mail Sent"
+
+def paper_data(paper_data, wait_time=3):
     data = {"status": "success"}
     data['data'] = {}
     paper_data = json.loads(paper_data)
@@ -26,7 +29,7 @@ def paper_data(paper_data):
                     print(f"Skipping paper with ID: {paper_id} (missing details)")
                     continue
                 summary, mindmap = summarize_paper(pdf_url, paper_id, summarizer_api_key)
-                post_blog(title, category, summary, mindmap, citation, access_key)
+                post_blog(title, category, summary, mindmap, citation, access_key, wait_time)
                 data['data'][category][paper_id] = {
                     "id": paper_id,
                     "doi": doi,
@@ -36,16 +39,19 @@ def paper_data(paper_data):
                     "summary": summary,
                     "mindmap": mindmap,
                 }
-    output_file = "paper_data_with_summary.json"
     data = json.dumps(data, indent=4, ensure_ascii=False)
-    with open(output_file, "w", encoding="utf-8") as file:
-            json.dump(data, file, indent=4)
-    print(f"Processed data saved to {output_file}")
     return data
 
-def post_blogpost(uaccess_key):
+def post_blogpost(uaccess_key, wait_time=3):
     if uaccess_key != access_key:
         return False
     data = fetch_paper_data_with_category(uaccess_key)
-    pdata = paper_data(data)
+    pdata = paper_data(data, wait_time)
+    try:
+        send_mail()
+        print("\n-------------------------------------------------------\nMail Sent\n-------------------------------------------------------\n")
+    except Exception as e:
+        print(f"\n-------------------------------------------------------\nError sending mail: {e}\n-------------------------------------------------------\n")
+    finally:
+        print("\n-------------------------------------------------------\nProcess Completed\n-------------------------------------------------------\n")
     return pdata

post_blog.py

@@ -99,15 +99,15 @@ def post_post(title, category, body, image):
         print(f"Error posting {title}: {e}")
         return False
     
-def post_blog(title, category, summary, mindmap, citation, uaccess_key):
+def post_blog(title, category, summary, mindmap, citation, uaccess_key, wait_time=3):
     if uaccess_key != access_key:
         return False
     else:
         status = True
         post_title, post_category, post_body, post_image = create_post(title, category, summary, mindmap, citation)
         status = post_post(post_title, post_category, post_body, post_image)
-        print(f"Waiting for {2*60} seconds...")
-        time.sleep(2*60)
+        print(f"Waiting for {wait_time*60} seconds...")
+        time.sleep(wait_time*60)
         if status:
             print('Post created successfully')
             return True
@@ -142,7 +142,7 @@ def test(access_key):
                     mindmap = paperdata['mindmap']
                     citation = paperdata['citation']
                     access_key = access_key
-                    status = post_blog(title, category, summary, mindmap, citation, access_key)
+                    status = post_blog(title, category, summary, mindmap, citation, access_key, 0)
                     print(status)
                     return status
                 

send_mail.py

@@ -0,0 +1,69 @@
+from email import encoders
+from email.mime.base import MIMEBase
+import os
+from datetime import datetime
+from pytz import timezone
+import pytz
+import sib_api_v3_sdk
+from sib_api_v3_sdk.rest import ApiException
+from dotenv import load_dotenv
+
+load_dotenv()
+mail_api = os.getenv("MAIL_API")
+
+def mail_body(generation_details):
+    body = f"""
+    Hello,
+    These are the details of the Blogs Posted at ReXplore: Science @ Fingertips.
+
+    Date & Time: {get_current_time()}
+
+
+    {generation_details}
+
+
+    Regards,
+    Nayan Kasturi (Raanna),
+    Developer & Maintainer,
+    ReXplore.
+    """
+    return body
+
+def get_current_time():
+    fmt = "%d-%m-%Y %H:%M:%S %Z%z"
+    now_utc = datetime.now(timezone('UTC'))
+    now_asia = now_utc.astimezone(timezone('Asia/Kolkata'))
+    return now_asia.strftime(fmt)
+
+def create_attachment(content, filename):
+    attachment = MIMEBase('application', 'octet-stream')
+    attachment.set_payload(content)
+    encoders.encode_base64(attachment)
+    attachment.add_header('Content-Disposition', f'attachment; filename="{filename}"')
+    return attachment
+
+def send_email(generation_details):
+    configuration = sib_api_v3_sdk.Configuration()
+    configuration.api_key['api-key'] = mail_api
+    api_instance = sib_api_v3_sdk.TransactionalEmailsApi(sib_api_v3_sdk.ApiClient(configuration))
+
+    data = mail_body(generation_details)
+    data_attchment = create_attachment(data.encode('utf-8'), "data.txt")
+
+    subject = "New Blog Batch Published to ReXplore at " + get_current_time()
+    sender = {"name": "Project Gatekeeper", "email": "projectgatekeeper@silerudaagartha.eu.org"}
+    reply_to = {"name": "Project Gatekeeper", "email": "gatekeeper@raannakasturi.eu.org"}
+    text_content = data
+    attachments = [
+        {"content": data_attchment.get_payload(), "name": data_attchment.get_filename()},
+    ]
+    to = [{"email": "raannakasturi@proton.me"}]
+    send_smtp_email = sib_api_v3_sdk.SendSmtpEmail(to=to, reply_to=reply_to, attachment=attachments, text_content=text_content, sender=sender, subject=subject)
+    try:
+        api_instance.send_transac_email(send_smtp_email)
+        print("Email Sent")
+        return True
+    except ApiException as e:
+        print("Can't send email")
+        print("Exception when calling SMTPApi->send_transac_email: %s\n" % e)
+        return False