Update app.py

app.py

@@ -3,6 +3,7 @@ import requests
 import language_tool_python
 from bs4 import BeautifulSoup
 import os
+from datetime import datetime  # Import datetime for date validation
 
 # Load Groq Cloud API key from Hugging Face secrets
 groq_api_key = os.getenv("GROQ_CLOUD_API_KEY")
@@ -11,7 +12,11 @@ groq_api_key = os.getenv("GROQ_CLOUD_API_KEY")
 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."
-    return bio
+    
+    # Simulate LinkedIn profile scraping (replace this with actual scraping logic)
+    linkedin_profile = f"https://linkedin.com/in/{name.replace(' ', '').lower()}"
+    
+    return bio, linkedin_profile
 
 # Helper function to call Groq Cloud LLM API to generate email
 def generate_email_from_groq(bio, company_name, role):
@@ -41,16 +46,30 @@ def check_grammar(email_text):
     corrected_text = language_tool_python.utils.correct(email_text, matches)
     return corrected_text
 
+# Function to validate the DOB format (DD-MM-YYYY)
+def validate_dob(dob):
+    try:
+        # Attempt to parse the DOB to the expected format
+        datetime.strptime(dob, "%d-%m-%Y")
+        return True
+    except ValueError:
+        # Return False if the format is invalid
+        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):
+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
-    bio = fetch_public_data(name, dob, city)
+    bio, linkedin_profile = fetch_public_data(name, dob, city)
     
     # Step 2: Generate the email using Groq Cloud LLM
     generated_email = generate_email_from_groq(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}"
+    signature = f"\n\nBest regards,\n{name}\nEmail: {email}\nPhone: {phone}\nLinkedIn: {linkedin_profile}"
 
     # Step 4: Run grammar and tone check
     polished_email = check_grammar(generated_email + signature)
@@ -62,7 +81,7 @@ def create_email(name, dob, city, company_name, role, email, phone, linkedin):
 def gradio_ui():
     # Define inputs
     name_input = gr.Textbox(label="Name", placeholder="Enter your name")
-    dob_input = gr.Textbox(label="Date of Birth", placeholder="Enter your date of birth (YYYY-MM-DD)")
+    dob_input = gr.Textbox(label="Date of Birth", placeholder="Enter your date of birth (DD-MM-YYYY)")
     city_input = gr.Textbox(label="City", placeholder="Enter your city of residence")
     
     company_name_input = gr.Textbox(label="Company Name", placeholder="Enter the name of the company you are applying to")
@@ -70,7 +89,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 Profile URL", placeholder="Enter your LinkedIn profile link")
     
     # Define output for the generated email
     email_output = gr.Textbox(label="Generated Email", placeholder="Your generated email will appear here", lines=10)
@@ -78,7 +96,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.",