Update app.py

app.py

@@ -5,15 +5,14 @@ from email.mime.text import MIMEText
 from groq import Groq
 from gtts import gTTS
 import tempfile
-import pygame
 import streamlit as st
 from dotenv import load_dotenv
-import threading
+from pydub import AudioSegment
+
 # ==========================
-# 🔐 Secrets (set your own)
+# 🔐 Secrets (stored in .env)
 # ==========================
 load_dotenv()
-
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 EMAIL_ADDRESS = os.getenv("EMAIL_ADDRESS")
 EMAIL_PASSWORD = os.getenv("EMAIL_PASSWORD")
@@ -58,8 +57,7 @@ def translate_email(content, target_language):
             model="llama3-8b-8192",
             messages=[{"role": "user", "content": prompt}]
         )
-        translated = response.choices[0].message.content.strip()
-        return translated
+        return response.choices[0].message.content.strip()
     except Exception as e:
         return f"Translation failed: {e}"
 
@@ -67,19 +65,16 @@ def translate_email(content, target_language):
 # 📤 Email Sender Agent
 # ==========================
 def email_sender_agent(context):
-    email_address = os.environ["EMAIL_ADDRESS"]
-    email_password = os.environ["EMAIL_PASSWORD"]
-
     message = MIMEText(context["email_content"])
     message["Subject"] = "AI Agent Email"
-    message["From"] = email_address
+    message["From"] = EMAIL_ADDRESS
 
     success = 0
     for recipient in context["recipient_list"]:
         message["To"] = recipient
         try:
             with smtplib.SMTP_SSL("smtp.gmail.com", 465) as server:
-                server.login(email_address, email_password)
+                server.login(EMAIL_ADDRESS, EMAIL_PASSWORD)
                 server.send_message(message)
                 success += 1
         except Exception as e:
@@ -95,9 +90,9 @@ def speak_and_download(text, lang="en"):
     try:
         tts = gTTS(text=text, lang=lang)
         with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as fp:
-            temp_path = fp.name
-            tts.save(temp_path)
-        return temp_path
+            path = fp.name
+            tts.save(path)
+        return path
     except Exception as e:
         st.error(f"TTS error: {e}")
         return None
@@ -105,17 +100,11 @@ def speak_and_download(text, lang="en"):
 # ==========================
 # 📜 Whisper Transcriber
 # ==========================
-from pydub import AudioSegment
-
 def transcribe_audio(file):
     try:
-        # Handle if file is a path (str)
         if isinstance(file, str):
-            file_path = file
-            audio = AudioSegment.from_file(file_path)
+            audio = AudioSegment.from_file(file)
         else:
-            # Handle file-like object from uploader
-            file_path = file.name
             if file.name.endswith(".mp3"):
                 audio = AudioSegment.from_mp3(file)
             elif file.name.endswith(".wav"):
@@ -124,11 +113,10 @@ def transcribe_audio(file):
                 st.error("Unsupported file type.")
                 return ""
 
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as temp_wav:
-            audio.export(temp_wav.name, format="wav")
-            result = whisper_model.transcribe(temp_wav.name)
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp:
+            audio.export(tmp.name, format="wav")
+            result = whisper_model.transcribe(tmp.name)
             return result["text"]
-
     except Exception as e:
         st.error(f"❌ Transcription error: {e}")
         return ""
@@ -139,24 +127,25 @@ def transcribe_audio(file):
 st.set_page_config(page_title="Email Generator AI", layout="centered")
 st.title("📧 AI Email Generator with Voice Input")
 
+# Session state init
+if "context" not in st.session_state:
+    st.session_state.context = None
+
 mode = st.radio("Choose Input Mode", ["Text Input", "Upload Audio File", "Paste Full Email"])
 recipient_input = st.text_input("Recipient Emails (comma-separated)", "")
 tone = st.text_input("Email Tone (Formal, Friendly, Apologetic, etc.)", "Formal")
-
 recipient_list = [r.strip() for r in recipient_input.split(",") if r.strip()]
 
-context = None
-
 if mode == "Text Input":
     user_request = st.text_area("What should the email say?")
     if st.button("Generate Email"):
-        context = init_context(user_request, recipient_list, tone)
-        context = email_generator_agent(context)
+        st.session_state.context = init_context(user_request, recipient_list, tone)
+        st.session_state.context = email_generator_agent(st.session_state.context)
 
 elif mode == "Paste Full Email":
     full_email = st.text_area("Paste your complete email")
     if st.button("Use Email"):
-        context = {
+        st.session_state.context = {
             "user_input": "",
             "recipient_list": recipient_list,
             "email_type": "manual",
@@ -165,24 +154,24 @@ elif mode == "Paste Full Email":
         }
 
 elif mode == "Upload Audio File":
-    uploaded_audio = st.file_uploader("Upload a .wav audio file", type=["wav"])
+    uploaded_audio = st.file_uploader("Upload an audio file (.wav or .mp3)", type=["wav", "mp3"])
     if uploaded_audio and st.button("Transcribe and Generate Email"):
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp:
-            tmp.write(uploaded_audio.read())
-            tmp_path = tmp.name
-        transcribed = transcribe_audio(tmp_path)
+        transcribed = transcribe_audio(uploaded_audio)
         st.success(f"Transcribed Text: {transcribed}")
-        context = init_context(transcribed, recipient_list, tone)
-        context = email_generator_agent(context)
+        st.session_state.context = init_context(transcribed, recipient_list, tone)
+        st.session_state.context = email_generator_agent(st.session_state.context)
 
-if context:
+# ========== Show Email Output ========== #
+if st.session_state.context:
+    context = st.session_state.context
     st.subheader("📨 Generated Email")
     st.text_area("Email Content", context["email_content"], height=200)
-    
+
     audio_path = speak_and_download(context["email_content"])
     if audio_path:
         st.audio(audio_path)
 
+    # Translation
     if st.checkbox("🌐 Translate Email"):
         lang = st.text_input("Translate to (e.g., Urdu, Spanish, German)")
         lang_map = {
@@ -193,6 +182,7 @@ if context:
             "english": "en"
         }
         target_lang_code = lang_map.get(lang.lower(), "en")
+
         if lang and st.button("Translate"):
             translated = translate_email(context["email_content"], lang)
             st.success("Translated Email:")
@@ -200,8 +190,9 @@ if context:
             translated_audio_path = speak_and_download(translated, lang=target_lang_code)
             if translated_audio_path:
                 st.audio(translated_audio_path)
-            context["email_content"] = translated
+            st.session_state.context["email_content"] = translated
 
+    # Email Sending
     if st.button("✉️ Send Email"):
-        context = email_sender_agent(context)
-        st.success(context["status"])
+        st.session_state.context = email_sender_agent(st.session_state.context)
+        st.success(st.session_state.context["status"])