Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

(#757) Allow Users to Manually Edit the Generated Email in Cold Email Generator App #765

Merged
merged 3 commits into from
Oct 27, 2024
Merged

(#757) Allow Users to Manually Edit the Generated Email in Cold Email Generator App #765

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
5e604d4
Update app.py
shreyasparaj Oct 21, 2024
5fd7b36
feature to save email as txt file
shreyasparaj Oct 24, 2024
74c5919
Update app.py
shreyasparaj Oct 25, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
161 changes: 26 additions & 135 deletions Generative-AI/Cold-Email-Generator/app/app.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,33 +8,22 @@
from langchain_core.exceptions import OutputParserException
import re


# Load environment variables
load_dotenv()


def clean_text(text):
# Remove HTML tags
text = re.sub(r'<[^>]*?>', '', text)
# Remove URLs
text = re.sub(r'http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\\(\\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+', '', text)
# Remove special characters
text = re.sub(r'[^a-zA-Z0-9 ]', '', text)
# Replace multiple spaces with a single space
text = re.sub(r'\\s{2,}', ' ', text)
# Trim leading and trailing whitespace
text = text.strip()
# Remove extra whitespace
text = ' '.join(text.split())
return text


# Chain class handling the LLM processing
class Chain:
def __init__(self):
self.llm = ChatGroq(temperature=0, groq_api_key=os.getenv("GROQ_API_KEY"), model_name="llama-3.1-70b-versatile")


def extract_jobs(self, cleaned_text):
prompt_extract = PromptTemplate.from_template(
"""
Expand All @@ -56,205 +45,107 @@ def extract_jobs(self, cleaned_text):
raise OutputParserException("Context too big. Unable to parse jobs.")
return res if isinstance(res, list) else [res]


def write_mail(self, job, links, user_name, user_about):
prompt_email = PromptTemplate.from_template(
"""
### JOB DESCRIPTION:
{job_description}

### INSTRUCTION:
You are {user_name}. {user_about}
Your job is to write a cold email to the client regarding the job mentioned above, describing how you can contribute to fulfilling their needs.
Also, add the most relevant ones from the following links to showcase portfolio: {link_list}
Do not provide a preamble.
### EMAIL (NO PREAMBLE):


"""
)
chain_email = prompt_email | self.llm
res = chain_email.invoke({"job_description": str(job), "link_list": links, "user_name": user_name, "user_about": user_about})
return res.content


# Portfolio class using temporary in-memory storage
class Portfolio:
def __init__(self):
# Initialize a dictionary to store skills and portfolio links temporarily
if 'portfolio' not in st.session_state:
st.session_state['portfolio'] = []


def add_to_portfolio(self, skills, links):
"""Add the user's skills and portfolio links to temporary storage."""
if skills and links:
st.session_state['portfolio'].append({"skills": skills, "links": links})


def query_links(self, required_skills):
"""Query the temporary storage for relevant links based on provided skills."""
if not required_skills:
return []


# Find relevant portfolio entries based on skills
matched_links = []
for entry in st.session_state['portfolio']:
portfolio_skills = entry['skills']
if any(skill in portfolio_skills for skill in required_skills):
matched_links.append(entry['links'])

return matched_links[:2]

return matched_links[:2] # Return up to 2 matched links


# Function to create the Streamlit app interface
def create_streamlit_app(llm, portfolio, clean_text):
st.set_page_config(page_title="Cold Email Generator", page_icon="", layout="wide")


st.markdown("""
<style>
.main {
background-color: #121212;
color: #e0e0e0;
font-family: 'Arial', sans-serif;
}
.title {
text-align: center;
color: #e0e0e0;
font-size: 2.5em;
font-weight: bold;
}
.subtitle {
text-align: center;
color: #b0b0b0;
font-size: 1.2em;
}
.container {
padding: 20px;
max-width: 800px;
margin: 0 auto;
background-color: #1e1e1e;
border-radius: 8px;
box-shadow: 0 4px 8px rgba(0, 0, 0, 0.5);
}
.input-box {
margin-bottom: 20px;
}
.input-box input, .input-box textarea {
width: 100%;
padding: 10px;
border-radius: 5px;
border: 1px solid #333;
background-color: #2c2c2c;
color: #e0e0e0;
box-sizing: border-box;
}
.button {
background-color: #007BFF;
color: #e0e0e0;
padding: 10px 20px;
border: none;
border-radius: 5px;
cursor: pointer;
font-size: 1em;
}
.button:hover {
background-color: #0056b3;
}
.code-block {
background-color: #2c2c2c;
padding: 10px;
border-radius: 5px;
color: #e0e0e0;
}
#MainMenu {visibility: hidden;}
footer {visibility: hidden;}
header {visibility: hidden;}
.stTextInput>div>div>textarea {min-height: 100px;}
.stButton>button {background-color: #007BFF;}
.stButton>button:hover {background-color: #0056b3;}
.footer {
text-align: center;
color: #e0e0e0;
font-size: 1em;
margin-top: 20px;
padding: 10px;
background-color: #333;
border-radius: 5px;
}
.footer a {
color: #66b3ff;
text-decoration: none;
}
.footer a:hover {
text-decoration: underline;
}
</style>
""", unsafe_allow_html=True)


st.markdown("<div class='title'>Cold Email Generator</div>", unsafe_allow_html=True)
st.markdown("<div class='subtitle'>Effortlessly craft professional cold emails for job applications based on job postings.</div>", unsafe_allow_html=True)


st.markdown("<div class='container'>", unsafe_allow_html=True)


user_name = st.text_input("Enter your name:", value=" ")
user_about = st.text_area(
"Enter a brief description about yourself:",
value=" "
)


url_input = st.text_input("Enter a Job Post URL:", value=" ")

user_name = st.text_input("Enter your name:", value=st.session_state.get('user_name', ""))
user_about = st.text_area("Enter a brief description about yourself:", value=st.session_state.get('user_about', ""))
url_input = st.text_input("Enter a Job Post URL:", value=st.session_state.get('url_input', ""))

st.subheader("Enter Your Skills and Portfolio Links")
skills_input = st.text_area("Enter your skills (comma separated):", value="")
links_input = st.text_area("Enter your portfolio links (comma separated):", value="")

skills_input = st.text_area("Enter your skills (comma separated):", value=st.session_state.get('skills_input', ""))
links_input = st.text_area("Enter your portfolio links (comma separated):", value=st.session_state.get('links_input', ""))

submit_button = st.button("Submit", key='submit_button', help="Click to generate the cold email")


if submit_button:
try:
st.session_state['user_name'] = user_name
st.session_state['user_about'] = user_about
st.session_state['url_input'] = url_input
st.session_state['skills_input'] = skills_input
st.session_state['links_input'] = links_input

skills_list = [skill.strip() for skill in skills_input.split(",")]
links_list = [link.strip() for link in links_input.split(",")]


portfolio.add_to_portfolio(skills_list, links_list)


loader = WebBaseLoader([url_input])
data = clean_text(loader.load().pop().page_content)
jobs = llm.extract_jobs(data)


for job in jobs:
job_skills = job.get('skills', [])
links = portfolio.query_links(job_skills)
email = llm.write_mail(job, links, user_name, user_about)
st.markdown(f"<div class='code-block'><pre>{email}</pre></div>", unsafe_allow_html=True)

st.session_state['draft_email'] = email

except Exception as e:
st.error(f"An Error Occurred: {e}")

if 'draft_email' in st.session_state:
st.text_area("Edit Email", value=st.session_state['draft_email'], key='edit_email', height=300)
if st.button("Save Draft"):
st.session_state['final_email'] = st.session_state['edit_email']
st.success("Draft saved successfully!")

st.markdown("</div>", unsafe_allow_html=True)
if 'final_email' in st.session_state:
st.markdown(f"<div class='code-block'><pre>{st.session_state['final_email']}</pre></div>", unsafe_allow_html=True)

# Add a copy button using Streamlit built-in feature
st.code(st.session_state['final_email'], language='text')


# Download button for the email
st.download_button(label="Download Email", data=st.session_state['final_email'], file_name="cold_email.txt", mime="text/plain")

st.markdown("</div>", unsafe_allow_html=True)

# Main function to run the app
if __name__ == "__main__":
chain = Chain()
portfolio = Portfolio()


create_streamlit_app(chain, portfolio, clean_text)