app.py

import gradio as gr
import requests
import language_tool_python
from bs4 import BeautifulSoup
import os

# Load Groq Cloud API key from Hugging Face secrets
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."
    return bio

# Helper function to call Groq Cloud LLM API to generate email
def generate_email_from_groq(bio, company_name, role):
    url = "https://api.groq.cloud/generate"  # Adjust based on Groq's API documentation
    headers = {
        "Authorization": f"Bearer {groq_api_key}",
        "Content-Type": "application/json",
    }
    prompt = f"Write a professional email applying for a {role} position at {company_name}. Use this bio: {bio}. The email should include an introduction, relevant experience, skills, and a closing."
    data = {
        "model": "groq-model",  # Adjust the model name based on Groq documentation
        "prompt": prompt,
        "max_tokens": 300
    }
    
    response = requests.post(url, headers=headers, json=data)
    
    if response.status_code == 200:
        return response.json().get("choices")[0].get("text").strip()
    else:
        return "Error generating email. Please check your API key or try again later."

# Grammar and Tone Checker Function
def check_grammar(email_text):
    tool = language_tool_python.LanguageTool('en-US')
    matches = tool.check(email_text)
    corrected_text = language_tool_python.utils.correct(email_text, matches)
    return corrected_text

# Main function to create the email and allow for saving, editing, or copying
def create_email(name, dob, city, company_name, role):
    # Step 1: Fetch public data based on user info
    bio = 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: Run grammar and tone check
    polished_email = check_grammar(generated_email)

    # Return the final polished email
    return polished_email

# Define interface with Gradio
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)")
    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")
    role_input = gr.Textbox(label="Role", placeholder="Enter the role you are applying for")

    # Define output for the generated email
    email_output = gr.Textbox(label="Generated Email", placeholder="Your generated email will appear here", lines=10)
    
    # 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],
        outputs=[email_output],
        title="Email Writing AI Agent",
        description="Generate a professional email for a job application by providing your basic info.",
        allow_flagging="never"  # Disable flagging
    )
    
    # Launch the Gradio app
    demo.launch()

# Start the Gradio app when running the script
if __name__ == "__main__":
    gradio_ui()