Update app.py
Browse files
app.py
CHANGED
@@ -7,6 +7,7 @@ import shutil
|
|
7 |
import os
|
8 |
from difflib import SequenceMatcher
|
9 |
import json
|
|
|
10 |
|
11 |
def construct_query(row):
|
12 |
"""Constructs the Google search query using applicant data."""
|
@@ -73,41 +74,36 @@ def fetch_linkedin_links(query, api_key, applicant_name):
|
|
73 |
"Content-Type": "application/json",
|
74 |
"Authorization": f"Bearer {api_key}"
|
75 |
}
|
76 |
-
|
|
|
|
|
|
|
77 |
payload = {
|
78 |
-
"zone": "serp_api2", #
|
79 |
-
"url":
|
80 |
-
"format": "
|
81 |
}
|
82 |
-
|
83 |
response = requests.post("https://api.brightdata.com/request", headers=headers, data=json.dumps(payload))
|
84 |
response.raise_for_status()
|
85 |
-
|
86 |
-
|
87 |
-
|
88 |
-
|
89 |
-
|
90 |
-
|
91 |
-
else:
|
92 |
-
# Fallback: Parse raw HTML if format is "raw"
|
93 |
-
results = []
|
94 |
-
|
95 |
-
# Search for LinkedIn links in response content (raw or parsed)
|
96 |
-
links = re.findall(linkedin_regex, response.text)
|
97 |
-
for link in links:
|
98 |
profile_name = get_name_from_url(link)
|
99 |
if profile_name:
|
100 |
similarity = calculate_similarity(applicant_name, profile_name)
|
101 |
if similarity >= 0.5:
|
102 |
return link
|
103 |
-
|
104 |
return None
|
105 |
|
106 |
except Exception as e:
|
107 |
st.error(f"Error fetching link for query '{query}': {e}")
|
108 |
return None
|
109 |
|
110 |
-
|
111 |
|
112 |
def process_file(file, api_key):
|
113 |
"""Processes the uploaded Excel file to fetch LinkedIn profile links."""
|
|
|
7 |
import os
|
8 |
from difflib import SequenceMatcher
|
9 |
import json
|
10 |
+
from urllib.parse import quote_plus
|
11 |
|
12 |
def construct_query(row):
|
13 |
"""Constructs the Google search query using applicant data."""
|
|
|
74 |
"Content-Type": "application/json",
|
75 |
"Authorization": f"Bearer {api_key}"
|
76 |
}
|
77 |
+
|
78 |
+
encoded_query = quote_plus(query)
|
79 |
+
search_url = f"https://www.google.com/search?q={encoded_query}"
|
80 |
+
|
81 |
payload = {
|
82 |
+
"zone": "serp_api2", # Must match your BrightData dashboard zone
|
83 |
+
"url": search_url,
|
84 |
+
"format": "raw" # Use "raw" if "json" gives problems
|
85 |
}
|
86 |
+
|
87 |
response = requests.post("https://api.brightdata.com/request", headers=headers, data=json.dumps(payload))
|
88 |
response.raise_for_status()
|
89 |
+
|
90 |
+
html = response.text
|
91 |
+
linkedin_links = re.findall(r'https://(www|[a-z]{2})\.linkedin\.com/in/[a-zA-Z0-9\-]+', html)
|
92 |
+
linkedin_links = list(set(["https://" + link for link in linkedin_links])) # De-duplicate
|
93 |
+
|
94 |
+
for link in linkedin_links:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
95 |
profile_name = get_name_from_url(link)
|
96 |
if profile_name:
|
97 |
similarity = calculate_similarity(applicant_name, profile_name)
|
98 |
if similarity >= 0.5:
|
99 |
return link
|
100 |
+
|
101 |
return None
|
102 |
|
103 |
except Exception as e:
|
104 |
st.error(f"Error fetching link for query '{query}': {e}")
|
105 |
return None
|
106 |
|
|
|
107 |
|
108 |
def process_file(file, api_key):
|
109 |
"""Processes the uploaded Excel file to fetch LinkedIn profile links."""
|