Update app.py

app.py

@@ -4,24 +4,24 @@ import os
 from datetime import datetime  # Import datetime for date validation
 
 # Load API keys securely from environment variables
-serpapi_key = os.getenv("SERPAPI_API_KEY")  # Add your SerpAPI key to your environment variables
+proxycurl_api_key = os.getenv("PROXYCURL_API_KEY")  # Add your Proxycurl API key to your environment variables
 groq_api_key = os.getenv("GROQ_CLOUD_API_KEY")
 
-# Function to use SerpAPI to get the LinkedIn URL via HTTP requests
-def get_linkedin_profile_via_serpapi(name, city):
-    query = f"{name} LinkedIn {city}"
-    url = f"https://serpapi.com/search.json?engine=google&q={query}&hl=en&gl=us&api_key={serpapi_key}"
+# Function to use Proxycurl API to get the LinkedIn profile data
+def get_linkedin_profile_via_proxycurl(linkedin_profile_url):
+    headers = {
+        "Authorization": f"Bearer {proxycurl_api_key}",
+    }
+    url = f"https://nubela.co/proxycurl/api/v2/linkedin?url={linkedin_profile_url}"
+    
+    response = requests.get(url, headers=headers)
     
-    response = requests.get(url)
     if response.status_code == 200:
-        results = response.json()
-        # Look for LinkedIn profile in search results
-        for result in results.get("organic_results", []):
-            if "linkedin.com" in result.get("link", ""):
-                return result["link"]
-        return "LinkedIn profile not found"
+        data = response.json()
+        bio = data.get("summary", "No bio available")
+        return bio
     else:
-        return "Error: Unable to fetch results from SerpAPI"
+        return "Error: Unable to fetch LinkedIn profile"
 
 # Helper function to call Groq Cloud LLM API to generate and correct the email
 def generate_and_correct_email(bio, company_name, role):
@@ -70,22 +70,22 @@ def validate_dob(dob):
         return False
 
 # Main function to create the email and allow for saving, editing, or copying
-def create_email(name, dob, city, company_name, role, email, phone):
+def create_email(name, dob, city, company_name, role, email, phone, linkedin_profile_url):
     # Validate the DOB format (DD-MM-YYYY)
     if not validate_dob(dob):
         return "Invalid Date of Birth format. Please use DD-MM-YYYY."
 
-    # Step 1: Fetch LinkedIn profile using SerpAPI
-    linkedin_profile = get_linkedin_profile_via_serpapi(name, city)
-    
-    # Use a placeholder bio (you can extend this to fetch real bio data from LinkedIn)
-    bio = f"{name} is a professional in {city}."
+    # Step 1: Fetch LinkedIn profile using Proxycurl API if LinkedIn URL is provided
+    if linkedin_profile_url:
+        bio = get_linkedin_profile_via_proxycurl(linkedin_profile_url)
+    else:
+        bio = f"{name} is a professional in {city}."  # Default bio if no LinkedIn URL is provided
     
     # Step 2: Generate the email using Groq Cloud LLM
     generated_email = generate_and_correct_email(bio, company_name, role)
     
     # Step 3: Add the user's email, phone number, and LinkedIn profile to the signature
-    signature = f"\n\nBest regards,\n{name}\nEmail: {email}\nPhone: {phone}\nLinkedIn: {linkedin_profile}"
+    signature = f"\n\nBest regards,\n{name}\nEmail: {email}\nPhone: {phone}\nLinkedIn: {linkedin_profile_url if linkedin_profile_url else 'Not provided'}"
 
     # Return the final polished email with the signature
     return generated_email + signature
@@ -102,6 +102,7 @@ def gradio_ui():
     
     email_input = gr.Textbox(label="Email Address", placeholder="Enter your email address")
     phone_input = gr.Textbox(label="Phone Number", placeholder="Enter your phone number")
+    linkedin_input = gr.Textbox(label="LinkedIn URL", placeholder="Enter your LinkedIn profile URL")  # New field for LinkedIn URL
     
     # Define output for the generated email
     email_output = gr.Textbox(label="Generated Email", placeholder="Your generated email will appear here", lines=10)
@@ -109,7 +110,7 @@ def gradio_ui():
     # Create the Gradio interface
     demo = gr.Interface(
         fn=create_email,  # Function to call when the user submits
-        inputs=[name_input, dob_input, city_input, company_name_input, role_input, email_input, phone_input],
+        inputs=[name_input, dob_input, city_input, company_name_input, role_input, email_input, phone_input, linkedin_input],
         outputs=[email_output],
         title="Email Writing AI Agent",
         description="Generate a professional email for a job application by providing your basic info.",