Update app.py

app.py

@@ -1,20 +1,31 @@
 import gradio as gr
 import requests
+from serpapi import GoogleSearch  # Import SerpAPI for searching
 import os
 from datetime import datetime  # Import datetime for date validation
 
-# Load Groq Cloud API key securely from environment variables
+# Load API keys securely from environment variables
+serpapi_key = os.getenv("SERPAPI_API_KEY")  # Add your SerpAPI key to your environment variables
 groq_api_key = os.getenv("GROQ_CLOUD_API_KEY")
 
-# Scraping function to fetch user public data (for demo purposes, we will simulate this)
-def fetch_public_data(name, dob, city):
-    # Simulate fetched bio data for demonstration purposes
-    bio = f"{name} is a software engineer from {city} with over 10 years of experience. Known for work in AI, cloud computing, and leadership in various engineering teams."
+# Function to use SerpAPI to get the LinkedIn URL
+def get_linkedin_profile_via_serpapi(name, city):
+    query = f"{name} LinkedIn {city}"
+    params = {
+        "q": query,
+        "hl": "en",
+        "gl": "us",
+        "api_key": serpapi_key,
+    }
+    search = GoogleSearch(params)
+    results = search.get_dict()
     
-    # Since we're asking the user to input their LinkedIn URL, we'll no longer scrape LinkedIn.
-    linkedin_profile = None
+    # 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 bio, linkedin_profile
+    return "LinkedIn profile not found"
 
 # Helper function to call Groq Cloud LLM API to generate and correct the email
 def generate_and_correct_email(bio, company_name, role):
@@ -63,13 +74,16 @@ 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, linkedin_profile):
+def create_email(name, dob, city, company_name, role, email, phone):
     # 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 public data based on user info (no need to scrape LinkedIn anymore)
-    bio, _ = fetch_public_data(name, dob, city)  # We'll use the provided LinkedIn URL
+    # 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 2: Generate the email using Groq Cloud LLM
     generated_email = generate_and_correct_email(bio, company_name, role)
@@ -92,7 +106,6 @@ 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 input 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)
@@ -100,7 +113,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, linkedin_input],
+        inputs=[name_input, dob_input, city_input, company_name_input, role_input, email_input, phone_input],
         outputs=[email_output],
         title="Email Writing AI Agent",
         description="Generate a professional email for a job application by providing your basic info.",