Create app.py

app.py

@@ -0,0 +1,109 @@
+import gradio as gr
+import os
+import sys
+import uuid
+import tempfile
+import chromadb
+from langchain_groq import ChatGroq
+from langchain_community.document_loaders import WebBaseLoader, UnstructuredFileLoader
+from langchain_text_splitters import RecursiveCharacterTextSplitter
+from langchain_core.prompts import PromptTemplate
+
+# Get API key from Hugging Face Secrets
+GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
+
+def generate_content(resume_file, job_url):
+    """
+    Main function to generate the cover letter.
+    """
+    if not GROQ_API_KEY:
+        return "❌ Error: Groq API key is not set in Hugging Face secrets. Please add it to your Space settings."
+    if not resume_file:
+        return "❌ Error: Please upload a resume."
+    if not job_url:
+        return "❌ Error: Please provide a job description URL."
+        
+    # --- 1. Validate Groq API Key ---
+    try:
+        llm = ChatGroq(
+            temperature=0,
+            groq_api_key=GROQ_API_KEY,
+            model_name="llama3-70b-8192"
+        )
+        llm.invoke("Test LLM connection.")
+    except Exception as e:
+        return f"❌ Error: Invalid Groq API key or model unavailable. Details: {e}"
+
+    # --- 2. Process Resume ---
+    try:
+        # Gradio's File component provides a NamedTemporaryFile
+        loader = UnstructuredFileLoader(resume_file.name)
+        resume_text = loader.load()[0].page_content
+        text_splitter = RecursiveCharacterTextSplitter(chunk_size=500, chunk_overlap=50)
+        resume_chunks = text_splitter.split_text(resume_text)
+    except Exception as e:
+        return f"❌ Error processing the resume file. Ensure it's a valid PDF. Error: {e}"
+
+    # --- 3. Set up the Resume Vector Database ---
+    client = chromadb.PersistentClient('resume_vectorstore')
+    collection = client.get_or_create_collection(name="resume_content")
+    
+    # Clear old data before adding new
+    if collection.count() > 0:
+        collection.delete(ids=collection.get()['ids'])
+    
+    ids = [str(uuid.uuid4()) for _ in range(len(resume_chunks))]
+    collection.add(documents=resume_chunks, ids=ids)
+
+    # --- 4. Web Scraping and JD Extraction ---
+    try:
+        loader = WebBaseLoader(job_url)
+        jd_text = loader.load().pop().page_content
+    except Exception as e:
+        return f"❌ Error scraping the URL. Please check the URL. Error: {e}"
+    
+    prompt_extract = PromptTemplate.from_template(
+        """### SCRAPED TEXT FROM WEBSITE: {page_data}
+        ### INSTRUCTION: Extract key skills, technologies, and responsibilities.
+        Return them as a list of strings. ### OUTPUT:"""
+    )
+    chain_extract = prompt_extract | llm
+    jd_requirements = chain_extract.invoke(input={'page_data': jd_text}).content.split('\n')
+
+    # --- 5. Find Relevant Resume Content ---
+    relevant_resume_chunks = collection.query(
+        query_texts=jd_requirements,
+        n_results=5
+    ).get('documents', [])
+
+    # --- 6. Generate Cover Letter/Resume Content ---
+    prompt_content = PromptTemplate.from_template(
+        """### JOB REQUIREMENTS: {jd_requirements}
+        ### YOUR RESUME CONTENT: {resume_content}
+        ### INSTRUCTION: You are a career consultant. Write a professional and compelling cover letter.
+        ### COVER LETTER:"""
+    )
+    chain_content = prompt_content | llm
+    generated_content = chain_content.invoke(
+        input={
+            'jd_requirements': "\n".join(jd_requirements),
+            'resume_content': "\n".join([item for sublist in relevant_resume_chunks for item in sublist])
+        }
+    ).content
+
+    return generated_content
+
+# --- Gradio UI ---
+iface = gr.Interface(
+    fn=generate_content,
+    inputs=[
+        gr.File(label="Upload your resume (PDF)"),
+        gr.Textbox(label="Job Posting URL"),
+    ],
+    outputs=gr.Textbox(label="Generated Cover Letter"),
+    title="AI Resume Matcher and Content Generator",
+    description="Upload your resume and a job description to get a personalized cover letter.",
+    theme="huggingface"
+)
+
+iface.launch()