Update utils.py

utils.py

@@ -1,3 +1,4 @@
+import re
 import os
 import streamlit as st
 import requests
@@ -5,6 +6,8 @@ from transformers import pipeline
 from typing import Dict
 from together import Together
 from gtts import gTTS
+from gmail_smtplib_micro import GmailSMTPLib, Email
+
 
 # Image-to-text
 def img2txt(url: str) -> str:
@@ -71,3 +74,56 @@ def get_user_preferences() -> Dict[str, str]:
     preferences['ending'] = st.selectbox("Ending", ["Happy", "Bittersweet", "Open-ended", "Tragic"])
     
     return preferences
+
+
+def create_gmail_config():
+    """
+    Creates a Gmail configuration file if it doesn't exist
+    """
+    config_content = f"""[GMail]
+username = {os.getenv('SENDER_EMAIL', '')}
+password = {os.getenv('SENDER_PASSWORD', '')}
+"""
+    with open('pysnail.conf', 'w') as f:
+        f.write(config_content)
+
+def send_story_email(recipient_email: str, story_text: str, audio_file_path: str) -> bool:
+    """
+    Send the story text and audio file to the specified email address
+    Returns True if successful, False otherwise
+    """
+    try:
+        # Create email content
+        subject = "Your Generated Story"
+        body = f"""Here's your generated story:
+
+{story_text}
+
+Enjoy!"""
+        
+        # Create email object
+        email = Email(subject, body)
+        
+        # Add audio file attachment
+        email.add_attachment(audio_file_path)
+        
+        # Create Gmail configuration if it doesn't exist
+        if not os.path.exists('pysnail.conf'):
+            create_gmail_config()
+        
+        # Initialize Gmail client and send email
+        gmail = GmailSMTPLib('pysnail.conf')
+        gmail.send_object(email, recipient_email)
+        
+        return True
+    
+    except Exception as e:
+        print(f"Error sending email: {str(e)}")
+        return False
+
+def validate_email(email: str) -> bool:
+    """
+    Basic email validation
+    """
+    pattern = r'^[\w\.-]+@[\w\.-]+\.\w+$'
+    return re.match(pattern, email) is not None