HEHEBOIOG commited on
Commit
c692f95
·
verified ·
1 Parent(s): 6195a3d

Upload main.py

Browse files
Files changed (1) hide show
  1. main.py +35 -0
main.py ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from langchain_community.document_loaders import WebBaseLoader
3
+
4
+ from chains import Chain
5
+ from portfolio import Portfolio
6
+ from utils import clean_text
7
+
8
+
9
+ def create_streamlit_app(llm, portfolio, clean_text):
10
+ st.title("📧 Cold Mail Generator")
11
+ url_input = st.text_input("Enter a URL:", value="https://jobs.nike.com/job/R-33460")
12
+ submit_button = st.button("Submit")
13
+
14
+ if submit_button:
15
+ try:
16
+ loader = WebBaseLoader([url_input])
17
+ data = clean_text(loader.load().pop().page_content)
18
+ portfolio.load_portfolio()
19
+ jobs = llm.extract_jobs(data)
20
+ for job in jobs:
21
+ skills = job.get('skills', [])
22
+ links = portfolio.query_links(skills)
23
+ email = llm.write_mail(job, links)
24
+ st.code(email, language='markdown')
25
+ except Exception as e:
26
+ st.error(f"An Error Occurred: {e}")
27
+
28
+
29
+ if __name__ == "__main__":
30
+ chain = Chain()
31
+ portfolio = Portfolio()
32
+ st.set_page_config(layout="wide", page_title="Cold Email Generator", page_icon="📧")
33
+ create_streamlit_app(chain, portfolio, clean_text)
34
+
35
+