File size: 1,540 Bytes
4880605 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
from flask import Flask, request, jsonify, render_template
from chains import Chain
from portfolio import Portfolio
from utils import clean_text
from langchain_community.document_loaders import WebBaseLoader
app = Flask(__name__)
chain = Chain()
portfolio = Portfolio()
@app.route('/')
def index():
return render_template('index.html')
@app.route('/generate-email', methods=['POST'])
def generate_email():
url = request.form.get('url')
if not url:
return jsonify({"error": "URL is required"}), 400
try:
# Load the webpage content
loader = WebBaseLoader([url])
data = clean_text(loader.load().pop().page_content)
# Load the portfolio into the vector database
portfolio.load_portfolio()
# Extract jobs from the cleaned text (use the first job found)
jobs = chain.extract_jobs(data)
if not jobs:
return jsonify({"error": "No jobs found on the provided URL"}), 404
# Generate a single email for the first job
job = jobs[0] # Take the first job if multiple are found
skills = job.get('skills', [])
links = portfolio.query_links(skills)
if not links:
links = "No relevant portfolio links found."
email = chain.write_mail(job, links)
return jsonify({"email": email})
except Exception as e:
return jsonify({"error": str(e)}), 500
if __name__ == '__main__':
app.run(debug=True)
|