Update app.py

app.py

@@ -11,11 +11,10 @@ from utils import (
     get_user_preferences, 
     send_story_email, 
     validate_email,
-    create_gmail_config
+    validate_gmail_credentials
 )
 
 def main():
-    # Page configuration
     st.set_page_config(
         page_title="🎨 Image-to-Audio Story 🎧",
         page_icon="🖼️",
@@ -23,21 +22,38 @@ def main():
     )
     st.title("Turn the Image into Audio Story")
 
-    # Check for Gmail configuration in sidebar
-    if not os.path.exists('pysnail.conf'):
-        st.sidebar.markdown("## 📧 Email Configuration")
-        with st.sidebar.expander("Configure Gmail Settings", expanded=True):
-            sender_email = st.text_input("Gmail Address:", type="default", key="gmail")
-            sender_password = st.text_input("App Password:", type="password", key="password")
-            
-            if st.button("Save Configuration"):
-                if sender_email and sender_password:
+    # Email Configuration in Sidebar
+    st.sidebar.markdown("## 📧 Email Configuration")
+    with st.sidebar.expander("Configure Gmail Settings", expanded=not bool(os.getenv('SENDER_EMAIL'))):
+        sender_email = st.text_input(
+            "Gmail Address:",
+            value=os.getenv('SENDER_EMAIL', ''),
+            type="default",
+            key="gmail"
+        )
+        sender_password = st.text_input(
+            "App Password:",
+            value=os.getenv('SENDER_PASSWORD', ''),
+            type="password",
+            key="password",
+            help="Use an App Password from your Google Account settings"
+        )
+        
+        if st.button("Save Configuration"):
+            if not sender_email or not sender_password:
+                st.error("Please provide both email and password.")
+            elif not validate_email(sender_email):
+                st.error("Please enter a valid Gmail address.")
+            else:
+                # Validate credentials
+                error = validate_gmail_credentials(sender_email, sender_password)
+                if error:
+                    st.error(f"Failed to validate credentials: {error}")
+                else:
+                    # Save credentials to environment variables
                     os.environ['SENDER_EMAIL'] = sender_email
                     os.environ['SENDER_PASSWORD'] = sender_password
-                    create_gmail_config()
                     st.success("Email configuration saved successfully!")
-                else:
-                    st.error("Please provide both email and password.")
 
     # LLM Parameters in sidebar
     st.sidebar.markdown("# ⚙️ LLM Configuration")
@@ -54,8 +70,7 @@ def main():
         st.markdown("## 📷 Upload Image")
         uploaded_file = st.file_uploader(
             "Choose an image...", 
-            type=["jpg", "jpeg", "png"],
-            help="Upload an image to generate a story from"
+            type=["jpg", "jpeg", "png"]
         )
 
         # Story preferences section
@@ -108,11 +123,9 @@ def main():
                             st.audio("audio_story.mp3")
                         
                         # Email section
-                        st.markdown("---")
-                        st.markdown("## 📧 Get Story via Email")
-                        
-                        # Only show email input if configuration exists
-                        if os.path.exists('pysnail.conf'):
+                        if os.getenv('SENDER_EMAIL') and os.getenv('SENDER_PASSWORD'):
+                            st.markdown("---")
+                            st.markdown("## 📧 Get Story via Email")
                             email = st.text_input(
                                 "Enter your email address:",
                                 help="We'll send you the story text and audio file"