Update app.py

app.py

@@ -9,11 +9,20 @@ from together import Together
 from email.mime.multipart import MIMEMultipart
 from email.mime.text import MIMEText
 from email.mime.audio import MIMEAudio
+#from dotenv import load_dotenv
+
+# Load environment variables
+#load_dotenv()
 
 # Image-to-text function
 def img2txt(url: str) -> str:
+    print("Initializing captioning model...")
     captioning_model = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
+    
+    print("Generating text from the image...")
     text = captioning_model(url, max_new_tokens=20)[0]["generated_text"]
+    
+    print(text)
     return text
 
 # Text-to-story generation function
@@ -43,6 +52,7 @@ def txt2story(prompt: str, top_k: int, top_p: float, temperature: float) -> str:
 
 # Text-to-speech function
 def txt2speech(text: str) -> None:
+    print("Converting text to speech using gTTS...")
     tts = gTTS(text=text, lang='en')
     tts.save("audio_story.mp3")
 
@@ -61,6 +71,47 @@ def get_user_preferences() -> Dict[str, str]:
     }
     return preferences
 
+# Function to send the story via email using smtplib
+def send_story_email(recipient_email: str, story_text: str, audio_file_path: str) -> bool:
+    try:
+        # Email configuration
+        smtp_server = os.environ.get("SMTP_SERVER")  # e.g., "smtp.gmail.com"
+        smtp_port = int(os.environ.get("SMTP_PORT", 587))  # e.g., 587 for TLS
+        sender_email = os.environ.get("SENDER_EMAIL")  # Your email
+        sender_password = os.environ.get("SENDER_PASSWORD")  # Your email password
+        
+        # Create the email
+        msg = MIMEMultipart()
+        msg['From'] = sender_email
+        msg['To'] = recipient_email
+        msg['Subject'] = "Your Generated Story"
+
+        # Attach the story text
+        msg.attach(MIMEText(f"Here's your generated story:\n\n{story_text}\n\nEnjoy!", 'plain'))
+
+        # Attach the audio file
+        with open(audio_file_path, 'rb') as audio_file:
+            audio_part = MIMEAudio(audio_file.read(), _subtype='mp3')
+            audio_part.add_header('Content-Disposition', 'attachment', filename=os.path.basename(audio_file_path))
+            msg.attach(audio_part)
+
+        # Send the email
+        with smtplib.SMTP(smtp_server, smtp_port) as server:
+            server.starttls()  # Upgrade to a secure connection
+            server.login(sender_email, sender_password)
+            server.send_message(msg)
+        
+        return True
+        
+    except Exception as e:
+        print(f"Error sending email: {str(e)}")
+        return False
+
+# Basic email validation function
+def validate_email(email: str) -> bool:
+    pattern = r'^[\w\.-]+@[\w\.-]+\.\w+$'
+    return re.match(pattern, email) is not None
+
 # Main Streamlit application
 def main():
     st.set_page_config(
@@ -92,6 +143,10 @@ def main():
 
     with col2:
         if uploaded_file is not None:
+            # Reset session state variables when a new image is uploaded
+            st.session_state.story = ""
+            st.session_state.audio_file_path = ""
+
             # Display uploaded image
             st.markdown("## 🖼️ Your Image")
             bytes_data = uploaded_file.read()
@@ -99,43 +154,53 @@ def main():
                 file.write(bytes_data)
             st.image(uploaded_file, use_column_width=True)
 
-            # Process image and generate story
-            if st.button("🎨 Generate Story"):
+            # Generate image caption as soon as image is uploaded
+            with st.spinner("📝 Generating image caption..."):
                 try:
-                    # Step 1: Generate Image Caption
-                    with st.spinner("🔍 Generating Image Caption..."):
-                        scenario = img2txt("uploaded_image.jpg")
-                        st.session_state.caption = scenario  # Store caption in session state
-
-                    # Step 2: Generate Story
-                    with st.spinner("📝 Generating Story..."):
-                        prompt = f"""Based on the image description: '{scenario}', 
-                        create a {preferences['genre']} story set in {preferences['setting']} 
-                        in {preferences['continent']}. The story should have a {preferences['tone']} 
-                        tone and explore the theme of {preferences['theme']}. The main conflict 
-                        should be {preferences['conflict']}. The story should have a {preferences['twist']} 
-                        and end with a {preferences['ending']} ending."""
-                        
-                        story = txt2story(prompt, top_k=5, top_p=0.8, temperature=1.5)
-                        st.session_state.story = story  # Store story in session state
-
-                    # Step 3: Generate Audio Version
-                    with st.spinner("🎧 Generating Audio Version..."):
-                        txt2speech(story)
-                        st.session_state.audio_file_path = "audio_story.mp3"  # Store audio path in session state
-
+                    scenario = img2txt("uploaded_image.jpg")
+                    st.session_state.caption = scenario  # Store caption in session state
+                    st.success("Image caption generated.")
                 except Exception as e:
-                    st.error(f"An error occurred: {str(e)}")
+                    st.error(f"An error occurred while generating image caption: {str(e)}")
                     st.warning("Please try again or contact support if the problem persists.")
 
+            # Display image caption
+            if st.session_state.caption:
+                st.markdown("## 📜 Image Caption")
+                st.write(st.session_state.caption)
+
+            # Generate Story button
+            if st.session_state.caption:
+                if st.button("🎨 Generate Story"):
+                    with st.spinner("📝 Generating story..."):
+                        try:
+                            # Create story prompt using st.session_state.caption
+                            prompt = f"""Based on the image description: '{st.session_state.caption}', 
+                            create a {preferences['genre']} story set in {preferences['setting']} 
+                            in {preferences['continent']}. The story should have a {preferences['tone']} 
+                            tone and explore the theme of {preferences['theme']}. The main conflict 
+                            should be {preferences['conflict']}. The story should have a {preferences['twist']} 
+                            and end with a {preferences['ending']} ending."""
+                            
+                            # Generate story
+                            story = txt2story(prompt, top_k=5, top_p=0.8, temperature=1.5)
+                            st.session_state.story = story  # Store story in session state
+                            st.success("Story generated.")
+
+                            # Convert to audio
+                            with st.spinner("🔊 Generating audio story..."):
+                                txt2speech(story)
+                                st.session_state.audio_file_path = "audio_story.mp3"  # Store audio path in session state
+                                st.success("Audio story generated.")
+
+                        except Exception as e:
+                            st.error(f"An error occurred: {str(e)}")
+                            st.warning("Please try again or contact support if the problem persists.")
+
         # Display results if story exists in session state
         if st.session_state.story:
             st.markdown("---")
             
-            # Image caption
-            with st.expander("📜 Image Caption", expanded=True):
-                st.write(st.session_state.caption)
-            
             # Story text
             with st.expander("📖 Generated Story", expanded=True):
                 st.write(st.session_state.story)
@@ -144,5 +209,25 @@ def main():
             with st.expander("🎧 Audio Version", expanded=True):
                 st.audio(st.session_state.audio_file_path)
 
+            # Email section
+            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"
+            )
+            
+            if st.button("📤 Send to Email"):
+                if not email:
+                    st.warning("Please enter an email address.")
+                elif not validate_email(email):
+                    st.error("Please enter a valid email address.")
+                else:
+                    with st.spinner("📨 Sending email..."):
+                        if send_story_email(email, st.session_state.story, st.session_state.audio_file_path):
+                            st.success(f"Email sent to: {email}")
+                        else:
+                            st.error("❌ Failed to send email. Please try again.")
+
 if __name__ == '__main__':
     main()