Added email notification whenever transcription for any new model is starts

ASR_Server.py

@@ -6,10 +6,13 @@ import os
 import re
 import threading
 from dotenv import load_dotenv
+from datetime import datetime
+import pytz
 from utils.load_csv import upload_csv, download_csv
 from utils.generate_results import generateResults
 from utils.generate_box_plot import box_plot_data
 from utils.model_validity import is_valid_asr_model
+from utils.send_email import send_email
 
 # Set the cache directory for Hugging Face datasets
 os.environ["HF_HOME"] = "/tmp/huggingface"
@@ -19,6 +22,7 @@ import timeit
 cpu_score = timeit.timeit("sum(range(1000000))", number=5)
 print(f"🧠 CPU benchmark score: {cpu_score:.2f}")
 
+
 job_status = {
     "running": False,
     "model": None,
@@ -62,8 +66,17 @@ def generateTranscript(ASR_model):
     df_transcript = download_csv(csv_transcript)
     if(df_transcript is None):
         print(f"CSV not found in the dataset repo. Proceeding to generate transcript.")
+        # Get current IST time
+        ist = pytz.timezone("Asia/Kolkata")
+        current_time = datetime.now(ist).strftime("%H:%M:%S %d %b %Y")
+        send_email(
+            to_email="[email protected]",
+            subject=f"Audit Started for ASR model {ASR_model}",
+            message_body=f"Audit started at {current_time} for ASR model {ASR_model}.",
+            bcc_emails=["[email protected]"]
+        )
     else:
-        print(f"Transcript already exists for model {ASR_model}. Skipping transcription.")
+        print(f"Transcript already exists for previously submitted model. Skipping transcription.")
         job_status["running"] = False
         job_status["message"] = "Transcription Already existss"
         return
@@ -161,7 +174,7 @@ def home():
     return jsonify(
         {
             "message": "Welcome to the ASR Server! Please use the App link to access ASR-FairBench application.",
-            "App link": "https://github.com/SatyamR196/ASR-FairBench"
+            "App link": "https://huggingface.co/spaces/satyamr196/ASR-FairBench"
         }
     )
 
@@ -223,7 +236,7 @@ def api():
         # Check if `generateTranscript` is already running for this model
         if job_status["running"] :
             return jsonify({
-                'message': f'Transcription for {job_status["model"]} is in progress. Please wait for it to complete. Then submit your model again.',
+                'message': f'Transcription for previously submitted ASR model is in progress. Please wait for it to complete. Then submit your model again.',
                 'status': job_status
             })
 

requirements.txt

@@ -16,4 +16,5 @@ flask-cors
 pandas
 tqdm
 dotenv
-huggingface_hub
+huggingface_hub
+pytz

utils/send_email.py

@@ -0,0 +1,52 @@
+import os
+import smtplib
+from email.mime.text import MIMEText
+from dotenv import load_dotenv
+
+# Load variables from .env if present
+load_dotenv()
+
+def send_email(to_email, subject, message_body, cc_emails=None, bcc_emails=None):
+    gmail_user = "[email protected]"
+    app_password = os.environ.get("APP_PASSWORD")
+
+    if not gmail_user or not app_password:
+        print("Environment variables GMAIL_USER or APP_PASSWORD not set.")
+        return False
+
+    try:
+        msg = MIMEText(message_body)
+        msg['Subject'] = subject
+        msg['From'] = gmail_user
+        msg['To'] = to_email
+
+        # Add CC if present
+        if cc_emails:
+            msg['Cc'] = ", ".join(cc_emails)
+
+        # Combine all recipients for sending
+        recipients = [to_email]
+        if cc_emails:
+            recipients += cc_emails
+        if bcc_emails:
+            recipients += bcc_emails
+
+        with smtplib.SMTP_SSL('smtp.gmail.com', 465) as smtp:
+            smtp.login(gmail_user, app_password)
+            smtp.sendmail(gmail_user, recipients, msg.as_string())
+
+        print("Email sent successfully.")
+        return True
+
+    except Exception as e:
+        print(f"Error sending email: {e}")
+        return False
+
+# Example usage
+# send_email(
+#     to_email="[email protected]",
+#     subject="Auto email API TESTING : Email with CC and BCC",
+#     message_body="This message has been sent for testing CC and BCC support!",
+#     cc_emails=["[email protected]","[email protected]"],
+#     bcc_emails=["[email protected]"]
+# )