Spaces:
Sleeping
Sleeping
| import streamlit as st | |
| import email_utils | |
| import LLM | |
| if "model" not in st.session_state.keys(): | |
| st.session_state["model"] = LLM.init() | |
| model = st.session_state["model"] | |
| if "credentials_set" not in st.session_state.keys(): | |
| st.session_state['credentials_set'] = False | |
| def start(): | |
| with st.sidebar: | |
| st.markdown("# Email Summarizer") | |
| st.subheader("Email Credentials") | |
| from_email = st.text_input("Email Address") | |
| from_password = st.text_input("App Password", type="password") | |
| if st.button("Set Credentials"): | |
| if not from_email or not from_password: | |
| st.error("Please provide both email address and password.") | |
| else: | |
| if email_utils.set_credentials(from_email, from_password): | |
| st.session_state['credentials_set'] = True | |
| st.success(f"Credentials set successfully. Email: {from_email}") | |
| email_ids = email_utils.fetch_emails_from_imap(from_email, from_password) | |
| st.session_state.update(email_ids=email_ids) | |
| else: | |
| st.error("Failed to set credentials") | |
| if st.session_state['credentials_set']: | |
| if st.button("Fetch Emails"): | |
| try: | |
| with st.spinner('Loading...'): | |
| render_emails(from_email, from_password) | |
| except Exception as e: | |
| print(e) | |
| else: | |
| st.button("Fetch Emails", disabled=True) | |
| st.warning("Please set credentials first", icon="⚠️",) | |
| def render_emails(from_email, from_password, page_size=10): | |
| """Renders the email messages in a Streamlit application with pagination. | |
| Args: | |
| email_messages: A list of email messages. | |
| page_number: The current page number. | |
| page_size: The number of emails to display per page. | |
| """ | |
| page_number = st.session_state.get("page", 1) | |
| email_ids = st.session_state.email_ids | |
| start_index = (page_number - 1) * page_size | |
| end_index = start_index + page_size | |
| email_messages = email_utils.decode_emails(email_ids, start_index, end_index, | |
| from_email, from_password) | |
| for email_message in email_messages: | |
| content = email_utils.strip_tags(email_message["content"]) | |
| summary = LLM.summarize(content, model[0]) | |
| tags = LLM.get_tags(content, model[2]) | |
| # spam = LLM.detect_spam(content, model[1]) | |
| # Add a redirect button that links to the original email | |
| redirect_url = f'https://mail.google.com/mail/u/0/#search/rfc822msgid%3A{email_message["Message ID"]}' | |
| EMAIL_FROM= email_message['from'] | |
| EMAIL_SUBJECT = email_message['subject'] if email_message['subject'].strip() else "No Subject" | |
| EMAIL_TAGS = tags[0]['generated_text'] | |
| EMAIL_SUMMARY = summary[0]['summary_text'] | |
| with st.expander( | |
| f"**From**:\n{EMAIL_FROM}\n\n**Subject**:\n{EMAIL_SUBJECT}\n\n**Tags**:\n{EMAIL_TAGS}\n\n" | |
| ): | |
| # tag_html = ''.join([f'<span style="display: inline-block; background-color: rgba(230, 230, 230, 0.2); padding: 3px 6px; margin-right: 8px; border-radius: 5px;">{tag}</span>' for tag in tags[0]['generated_text'].split(',')]) | |
| st.markdown(f"**Summary**:\n {EMAIL_SUMMARY}") | |
| st.markdown(f"**[Read full e-mail]({redirect_url})**") | |
| # Handle the button click event | |
| # if redirect_button_clicked: | |
| # st.write(f"Redirecting to Gmail: {redirect_url}") | |
| # st.experimental_rerun() # Refresh the app to open the link | |
| total = len(st.session_state.email_ids) | |
| # Add buttons to allow the user to navigate between pages. | |
| if page_number > 1: | |
| st.button('Previous page', | |
| on_click=lambda: (st.session_state.update(page=page_number - 1), | |
| render_emails(from_email, from_password))) | |
| if page_number < total: | |
| st.button('Next page', | |
| on_click=lambda: (st.session_state.update(page=page_number + 1), | |
| render_emails(from_email, from_password))) | |
| start() | |