Add email support, update .gitignore and .env_example (#10)
**Add email support for video summaries** This PR allow us to send video summary results via email instead of printing them. It also updates .env_example so that the template for environment variable is up to date. **Note:** Due to limitations with my OpenAI API token (insufficient credits), I was only able to successfully test the email sending independently. Integration of email with the transcription and summarization output has not been tested. **Closes #9** Co-authored-by: iakn0001 <imran.aknaf@ulb.be> Reviewed-on: #10 Reviewed-by: Eric Meehan <eric@eom.dev> Co-authored-by: Imran <imran.aknaf.pro@gmail.com> Co-committed-by: Imran <imran.aknaf.pro@gmail.com>
This commit is contained in:
parent
46cb748bc6
commit
52972d2668
@ -5,3 +5,8 @@ OPENAI_TRANSCRIPTION_MODEL=
|
||||
OPENAI_CHAT_SYSTEM_PROMPT=
|
||||
OPENAI_CHAT_MODEL=
|
||||
OPENAI_CHAT_N=
|
||||
SENDER_EMAIL=
|
||||
RECEIVER_EMAIL=
|
||||
SENDER_APP_PASSWORD=
|
||||
SMTP_HOST=
|
||||
SMTP_PORT=
|
1
.gitignore
vendored
1
.gitignore
vendored
@ -1,4 +1,5 @@
|
||||
__pycache__/*
|
||||
tmp/*
|
||||
venv/*
|
||||
.venv/*
|
||||
.env
|
||||
|
39
app.py
39
app.py
@ -7,6 +7,9 @@ from pydub import AudioSegment
|
||||
|
||||
from prompt import DEFAULT_PROMPT
|
||||
|
||||
import smtplib
|
||||
from email.message import EmailMessage
|
||||
|
||||
VIDEO_URL = os.getenv('INPUT_VIDEO_URL', None)
|
||||
OUTPUT_PATH = os.getenv('OUTPUT_PATH', 'tmp')
|
||||
AUDIO_SEGMENT_DURATION = 30000
|
||||
@ -16,13 +19,19 @@ OPENAI_TRANSCRIPTION_MODEL = os.getenv('OPENAI_TRANSCRIPTION_MODEL', 'whisper-1'
|
||||
OPENAI_CHAT_SYSTEM_PROMPT = os.getenv('OPENAI_CHAT_SYSTEM_PROMPT', DEFAULT_PROMPT)
|
||||
OPENAI_CHAT_MODEL = os.getenv('OPENAI_CHAT_MODEL', 'whisper-1')
|
||||
OPENAI_CHAT_N = int(os.getenv('OPENAI_CHAT_N', '3'))
|
||||
SENDER_EMAIL = os.getenv("SENDER_EMAIL", None)
|
||||
RECEIVER_EMAIL = os.getenv("RECEIVER_EMAIL", None)
|
||||
SENDER_APP_PASSWORD = os.getenv("SENDER_APP_PASSWORD", None)
|
||||
SMTP_HOST = os.getenv("SMTP_HOST", "smtp.gmail.com")
|
||||
SMTP_PORT = int(os.getenv("SMTP_PORT", "465"))
|
||||
|
||||
def main():
|
||||
openai_client = OpenAI(
|
||||
base_url = OPENAI_BASE_URL,
|
||||
api_key = OPENAI_API_KEY
|
||||
)
|
||||
return summarize_transcription(
|
||||
|
||||
summaries = summarize_transcription(
|
||||
openai_client,
|
||||
transcribe_audio(
|
||||
openai_client,
|
||||
@ -31,6 +40,14 @@ def main():
|
||||
)
|
||||
)
|
||||
)
|
||||
|
||||
email_body=""
|
||||
for i, summary in enumerate(summaries) :
|
||||
email_body += f"Option {i+1} :\n{summary.text}\n\n"
|
||||
|
||||
send_email(subject="Video summaries results", body=email_body)
|
||||
|
||||
return summaries
|
||||
|
||||
def get_video_from_url():
|
||||
filename = VIDEO_URL.split('/')[-1]
|
||||
@ -65,6 +82,21 @@ def summarize_transcription(openai_client, transcription):
|
||||
prompt=OPENAI_CHAT_SYSTEM_PROMPT.format(transcription)
|
||||
).choices
|
||||
|
||||
def send_email(subject, body, sender_email=SENDER_EMAIL, receiver_email=RECEIVER_EMAIL, sender_password=SENDER_APP_PASSWORD):
|
||||
msg = EmailMessage()
|
||||
msg.set_content(body)
|
||||
msg['Subject'] = subject
|
||||
msg['From'] = sender_email
|
||||
msg['To'] = receiver_email
|
||||
|
||||
try:
|
||||
with smtplib.SMTP_SSL(SMTP_HOST, SMTP_PORT) as smtp:
|
||||
smtp.login(sender_email, sender_password)
|
||||
smtp.send_message(msg)
|
||||
print("Email sent successfully!")
|
||||
except Exception as e:
|
||||
print(f"Email sending error: {e}")
|
||||
|
||||
def setup():
|
||||
from dotenv import load_dotenv
|
||||
load_dotenv()
|
||||
@ -74,6 +106,7 @@ def cleanup():
|
||||
|
||||
if __name__ == '__main__':
|
||||
setup()
|
||||
for each in main():
|
||||
|
||||
for summary in main():
|
||||
print("========")
|
||||
print(each.text)
|
||||
print(summary .text)
|
||||
|
Loading…
Reference in New Issue
Block a user