Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,470 @@
-import altair as alt
-import numpy as np
-import pandas as pd
 import streamlit as st
+import pandas as pd
+import json
+import os
+from pydantic import BaseModel, Field
+from typing import List, Set, Dict, Any, Optional
+import time
+from langchain_openai import ChatOpenAI
+from langchain_core.messages import HumanMessage
+from langchain_core.prompts import ChatPromptTemplate
+from langchain_core.output_parsers import StrOutputParser
+from langchain_core.prompts import PromptTemplate
+import gspread
+from google.oauth2 import service_account
+
+st.set_page_config(
+    page_title="Candidate Matching App",
+    page_icon="👨‍💻🎯",
+    layout="wide"
+)
+
+# Define pydantic model for structured output
+class Shortlist(BaseModel):
+    fit_score: float = Field(description="A score between 0 and 10 indicating how closely the candidate profile matches the job requirements.")
+    candidate_name: str = Field(description="The name of the candidate.")
+    candidate_url: str = Field(description="The URL of the candidate's LinkedIn profile.")
+    candidate_summary: str = Field(description="A brief summary of the candidate's skills and experience along with its educational background.")
+    candidate_location: str = Field(description="The location of the candidate.")
+    justification: str = Field(description="Justification for the shortlisted candidate with the fit score")
+
+# Function to parse and normalize tech stacks
+def parse_tech_stack(stack):
+    if pd.isna(stack) or stack == "" or stack is None:
+        return set()
+    if isinstance(stack, set):
+        return stack
+    try:
+        # Handle potential string representation of sets
+        if isinstance(stack, str) and stack.startswith("{") and stack.endswith("}"):
+            # This could be a string representation of a set
+            items = stack.strip("{}").split(",")
+            return set(item.strip().strip("'\"") for item in items if item.strip())
+        return set(map(lambda x: x.strip().lower(), str(stack).split(',')))
+    except Exception as e:
+        st.error(f"Error parsing tech stack: {e}")
+        return set()
+
+def display_tech_stack(stack_set):
+    if isinstance(stack_set, set):
+        return ", ".join(sorted(stack_set))
+    return str(stack_set)
+
+def get_matching_candidates(job_stack, candidates_df):
+    """Find candidates with matching tech stack for a specific job"""
+    matched = []
+    job_stack_set = parse_tech_stack(job_stack)
+    
+    for _, candidate in candidates_df.iterrows():
+        candidate_stack = parse_tech_stack(candidate['Key Tech Stack'])
+        common = job_stack_set & candidate_stack
+        if len(common) >= 2:
+            matched.append({
+                "Name": candidate["Full Name"],
+                "URL": candidate["LinkedIn URL"],
+                "Degree & Education": candidate["Degree & University"],
+                "Years of Experience": candidate["Years of Experience"],
+                "Current Title & Company": candidate['Current Title & Company'],
+                "Key Highlights": candidate["Key Highlights"],
+                "Location": candidate["Location (from most recent experience)"],
+                "Experience": str(candidate["Experience"]),
+                "Tech Stack": candidate_stack
+            })
+    return matched
+
+def setup_llm():
+    """Set up the LangChain LLM with structured output"""
+    # Create LLM instance
+    llm = ChatOpenAI(
+        model="gpt-4o-mini",
+        temperature=0,
+        max_tokens=None,
+        timeout=None,
+        max_retries=2,
+    )
+    
+    # Create structured output
+    sum_llm = llm.with_structured_output(Shortlist)
+    
+    # Create system prompt
+    system = """You are an expert Recruitor, your task is to analyse the Candidate profile and determine if it matches with the job details and provide a score(out of 10) indicating how compatible the
+    the profile is according to job.
+Try to ensure following points while estimating the candidate's fit score:
+For education:
+Tier1 - MIT, Stanford, CMU, UC Berkeley, Caltech, Harvard, IIT Bombay, IIT Delhi, Princeton, UIUC, University of Washington, Columbia, University of Chicago, Cornell, University of Michigan (Ann Arbor), UT Austin - Maximum points
+Tier2 - UC Davis, Georgia Tech, Purdue, UMass Amherst,etc - Moderate points
+Tier3 - Unknown or unranked institutions - Lower points or reject
+Startup Experience Requirement:
+Candidates must have worked  as a direct employee at a VC-backed startup (Seed to series C/D)
+preferred - Y Combinator, Sequoia,a16z,Accel,Founders Fund,LightSpeed,Greylock,Benchmark,Index Ventures,etc.   
+    The fit score signifies based on following metrics:
+    1–5 - Poor Fit - Auto-reject
+    6–7 - Weak Fit - Auto-reject
+    8.0–8.7 - Moderate Fit - Auto-reject
+    8.8–10 - STRONG Fit - Include in results
+    """
+    
+    # Create query prompt
+    query_prompt = ChatPromptTemplate.from_messages([
+        ("system", system),
+        ("human", """
+    You are an expert Recruitor, your task is to determine if the user is a correct match for the given job or not.
+          For this you will be provided with the follwing inputs of job and candidates:
+    Job Details
+    Company: {Company}
+    Role: {Role}
+    About Company: {desc}
+    Locations: {Locations}
+    Tech Stack: {Tech_Stack}
+    Industry: {Industry}
+    
+    Candidate Details:
+    Full Name: {Full_Name}
+    LinkedIn URL: {LinkedIn_URL}
+    Current Title & Company: {Current_Title_Company}
+    Years of Experience: {Years_of_Experience}
+    Degree & University: {Degree_University}
+    Key Tech Stack: {Key_Tech_Stack}
+    Key Highlights: {Key_Highlights}
+    Location (from most recent experience): {cand_Location}
+    Past_Experience: {Experience}
+    Answer in the structured manner as per the schema.
+    If any parameter is Unknown try not to include in the summary, only include those parameters which are known.
+    """),
+    ])
+    
+    # Chain the prompt and LLM
+    cat_class = query_prompt | sum_llm
+    
+    return cat_class
+
+def call_llm(candidate_data, job_data, llm_chain):
+    """Call the actual LLM to evaluate the candidate"""
+    try:
+        # Convert tech stacks to strings for the LLM payload
+        job_tech_stack = job_data.get("Tech_Stack", set())
+        candidate_tech_stack = candidate_data.get("Tech Stack", set())
+        
+        if isinstance(job_tech_stack, set):
+            job_tech_stack = ", ".join(sorted(job_tech_stack))
+            
+        if isinstance(candidate_tech_stack, set):
+            candidate_tech_stack = ", ".join(sorted(candidate_tech_stack))
+        
+        # Prepare payload for LLM
+        payload = {
+            "Company": job_data.get("Company", ""),
+            "Role": job_data.get("Role", ""),
+            "desc": job_data.get("desc", ""),
+            "Locations": job_data.get("Locations", ""),
+            "Tech_Stack": job_tech_stack,
+            "Industry": job_data.get("Industry", ""),
+
+            "Full_Name": candidate_data.get("Name", ""),
+            "LinkedIn_URL": candidate_data.get("URL", ""),
+            "Current_Title_Company": candidate_data.get("Current Title & Company", ""),
+            "Years_of_Experience": candidate_data.get("Years of Experience", ""),
+            "Degree_University": candidate_data.get("Degree & Education", ""),
+            "Key_Tech_Stack": candidate_tech_stack,
+            "Key_Highlights": candidate_data.get("Key Highlights", ""),
+            "cand_Location": candidate_data.get("Location", ""),
+            "Experience": candidate_data.get("Experience", "")
+        }
+        
+        # Call LLM
+        response = llm_chain.invoke(payload)
+        print(candidate_data.get("Experience", ""))
+        
+        # Return response in expected format
+        return {
+            "candidate_name": response.candidate_name,
+            "candidate_url": response.candidate_url,
+            "candidate_summary": response.candidate_summary,
+            "candidate_location": response.candidate_location,
+            "fit_score": response.fit_score,
+            "justification": response.justification
+        }
+    except Exception as e:
+        st.error(f"Error calling LLM: {e}")
+        # Fallback to a default response
+        return {
+            "candidate_name": candidate_data.get("Name", "Unknown"),
+            "candidate_url": candidate_data.get("URL", ""),
+            "candidate_summary": "Error processing candidate profile",
+            "candidate_location": candidate_data.get("Location", "Unknown"),
+            "fit_score": 0.0,
+            "justification": f"Error in LLM processing: {str(e)}"
+        }
+
+def process_candidates_for_job(job_row, candidates_df, llm_chain=None):
+    """Process candidates for a specific job using the LLM"""
+    if llm_chain is None:
+        with st.spinner("Setting up LLM..."):
+            llm_chain = setup_llm()
+    
+    selected_candidates = []
+    
+    try:
+        # Get job-specific data
+        job_data = {
+            "Company": job_row["Company"],
+            "Role": job_row["Role"],
+            "desc": job_row.get("One liner", ""),
+            "Locations": job_row.get("Locations", ""),
+            "Tech_Stack": job_row["Tech Stack"],
+            "Industry": job_row.get("Industry", "")
+        }
+        
+        # Find matching candidates for this job
+        with st.spinner("Finding matching candidates based on tech stack..."):
+            matching_candidates = get_matching_candidates(job_row["Tech Stack"], candidates_df)
+        
+        if not matching_candidates:
+            st.warning("No candidates with matching tech stack found for this job.")
+            return []
+        
+        st.success(f"Found {len(matching_candidates)} candidates with matching tech stack.")
+        
+        # Create progress elements
+        candidates_progress = st.progress(0)
+        candidate_status = st.empty()
+        
+        # Process each candidate
+        for i, candidate_data in enumerate(matching_candidates):
+            # Update progress
+            candidates_progress.progress((i + 1) / len(matching_candidates))
+            candidate_status.text(f"Evaluating candidate {i+1}/{len(matching_candidates)}: {candidate_data.get('Name', 'Unknown')}")
+            
+            # Process the candidate with the LLM
+            response = call_llm(candidate_data, job_data, llm_chain)
+            
+            response_dict = {
+                "Name": response["candidate_name"],
+                "LinkedIn": response["candidate_url"],
+                "summary": response["candidate_summary"],
+                "Location": response["candidate_location"],
+                "Fit Score": response["fit_score"],
+                "justification": response["justification"],
+                # Add back original candidate data for context
+                "Educational Background": candidate_data.get("Degree & Education", ""),
+                "Years of Experience": candidate_data.get("Years of Experience", ""),
+                "Current Title & Company": candidate_data.get("Current Title & Company", "")
+            }
+            
+            # Add to selected candidates if score is high enough
+            if response["fit_score"] >= 8.8:
+                selected_candidates.append(response_dict)
+                st.markdown(response_dict)
+            else:
+                st.write(f"Rejected candidate: {response_dict['Name']} with score: {response['fit_score']}")
+        
+        # Clear progress indicators
+        candidates_progress.empty()
+        candidate_status.empty()
+        
+        # Show results
+        if selected_candidates:
+            st.success(f"✅ Found {len(selected_candidates)} suitable candidates for this job!")
+        else:
+            st.info("No candidates met the minimum fit score threshold for this job.")
+        
+        return selected_candidates
+        
+    except Exception as e:
+        st.error(f"Error processing job: {e}")
+        return []
+
+def main():
+    st.title("👨‍💻 Candidate Matching App")
+    
+    # Initialize session state
+    if 'processed_jobs' not in st.session_state:
+        st.session_state.processed_jobs = {}
+    
+    st.write("""
+    This app matches job listings with candidate profiles based on tech stack and other criteria.
+    Select a job to find matching candidates.
+    """)
+    
+    # API Key input
+    with st.sidebar:
+        st.header("API Configuration")
+        api_key = st.text_input("Enter OpenAI API Key", type="password")
+        if api_key:
+            os.environ["OPENAI_API_KEY"] = api_key
+            st.success("API Key set!")
+        else:
+            st.warning("Please enter OpenAI API Key to use LLM features")
+            
+    # Show API key warning if not set
+    secret_content = os.getenv("GCP_SERVICE_ACCOUNT")
+    secret_content = secret_content.replace("\n", "\\n")
+    secret_content = json.loads(secret_content)
+    SCOPES = ['https://www.googleapis.com/auth/spreadsheets']
+    creds = service_account.Credentials.from_service_account_info(secret_content, scopes=SCOPES)
+    gc = gspread.authorize(creds)
+    job_sheet = gc.open_by_key('1BZlvbtFyiQ9Pgr_lpepDJua1ZeVEqrCLjssNd6OiG9k')
+    candidates_sheet = gc.open_by_key('1u_9o5f0MPHFUSScjEcnA8Lojm4Y9m9LuWhvjYm6ytF4')
+
+    if not api_key:
+        st.warning("⚠️ You need to provide an OpenAI API key in the sidebar to use this app.")
+
+    if api_key:
+        try:
+            # Load data from Google Sheets
+            job_worksheet = job_sheet.worksheet('paraform_jobs_formatted')
+            job_data = job_worksheet.get_all_values()
+            candidate_worksheet = candidates_sheet.worksheet('transformed_candidates_updated')
+            candidate_data = candidate_worksheet.get_all_values()
+            
+            # Convert to DataFrames
+            jobs_df = pd.DataFrame(job_data[1:], columns=job_data[0])
+            candidates_df = pd.DataFrame(candidate_data[1:], columns=candidate_data[0])
+            candidates_df = candidates_df.fillna("Unknown")
+            
+            # Display data preview
+            with st.expander("Preview uploaded data"):
+                st.subheader("Jobs Data Preview")
+                st.dataframe(jobs_df.head(3))
+                
+                st.subheader("Candidates Data Preview")
+                st.dataframe(candidates_df.head(3))
+            
+            # Map column names if needed
+            column_mapping = {
+                "Full Name": "Full Name",
+                "LinkedIn URL": "LinkedIn URL",
+                "Current Title & Company": "Current Title & Company",
+                "Years of Experience": "Years of Experience",
+                "Degree & University": "Degree & University",
+                "Key Tech Stack": "Key Tech Stack", 
+                "Key Highlights": "Key Highlights",
+                "Location (from most recent experience)": "Location (from most recent experience)"
+            }
+            
+            # Rename columns if they don't match expected
+            candidates_df = candidates_df.rename(columns={
+                col: mapping for col, mapping in column_mapping.items() 
+                if col in candidates_df.columns and col != mapping
+            })
+            
+            # Now, instead of processing all jobs upfront, we'll display job selection
+            # and only process the selected job when the user chooses it
+            display_job_selection(jobs_df, candidates_df)
+            
+        except Exception as e:
+            st.error(f"Error processing files: {e}")
+    
+    st.divider()
+    
+
+def display_job_selection(jobs_df, candidates_df):
+    # Store the LLM chain as a session state to avoid recreating it
+    if 'llm_chain' not in st.session_state:
+        st.session_state.llm_chain = None
+    
+    st.subheader("Select a job to view potential matches")
+    
+    # Create job options - but don't compute matches yet
+    job_options = []
+    for i, row in jobs_df.iterrows():
+        job_options.append(f"{row['Role']} at {row['Company']}")
+    
+    if job_options:
+        selected_job_index = st.selectbox("Jobs:", 
+                                      range(len(job_options)),
+                                      format_func=lambda x: job_options[x])
+        
+        # Display job details
+        job_row = jobs_df.iloc[selected_job_index]
+        
+        # Parse tech stack for display
+        job_row_stack = parse_tech_stack(job_row["Tech Stack"])
+        
+        col1, col2 = st.columns([2, 1])
+        
+        with col1:
+            st.subheader(f"Job Details: {job_row['Role']}")
+            
+            job_details = {
+                "Company": job_row["Company"],
+                "Role": job_row["Role"],
+                "Description": job_row.get("One liner", "N/A"),
+                "Locations": job_row.get("Locations", "N/A"),
+                "Industry": job_row.get("Industry", "N/A"),
+                "Tech Stack": display_tech_stack(job_row_stack)
+            }
+            
+            for key, value in job_details.items():
+                st.markdown(f"**{key}:** {value}")
+        
+        # Create a key for this job in session state
+        job_key = f"job_{selected_job_index}_processed"
+        
+        if job_key not in st.session_state:
+            st.session_state[job_key] = False
+        
+        # Add a process button for this job
+        if not st.session_state[job_key]:
+            if st.button(f"Find Matching Candidates for this Job"):
+                if "OPENAI_API_KEY" not in os.environ or not os.environ["OPENAI_API_KEY"]:
+                    st.error("Please enter your OpenAI API key in the sidebar before processing")
+                else:
+                    # Process candidates for this job (only when requested)
+                    selected_candidates = process_candidates_for_job(
+                        job_row, 
+                        candidates_df,
+                        st.session_state.llm_chain
+                    )
+                    
+                    # Store the results and set as processed
+                    if 'Selected_Candidates' not in st.session_state:
+                        st.session_state.Selected_Candidates = {}
+                    st.session_state.Selected_Candidates[selected_job_index] = selected_candidates
+                    st.session_state[job_key] = True
+                    
+                    # Store the LLM chain for reuse
+                    if st.session_state.llm_chain is None:
+                        st.session_state.llm_chain = setup_llm()
+                    
+                    # Force refresh
+                    st.rerun()
+        
+        # Display selected candidates if already processed
+        if st.session_state[job_key] and 'Selected_Candidates' in st.session_state:
+            selected_candidates = st.session_state.Selected_Candidates.get(selected_job_index, [])
+            
+            # Display selected candidates
+            st.subheader("Selected Candidates")
+            
+            if len(selected_candidates) > 0:
+                for i, candidate in enumerate(selected_candidates):
+                    with st.expander(f"{i+1}. {candidate['Name']} (Score: {candidate['Fit Score']})"):
+                        col1, col2 = st.columns([3, 1])
+                        
+                        with col1:
+                            st.markdown(f"**Summary:** {candidate['summary']}")
+                            st.markdown(f"**Current:** {candidate['Current Title & Company']}")
+                            st.markdown(f"**Education:** {candidate['Educational Background']}")
+                            st.markdown(f"**Experience:** {candidate['Years of Experience']}")
+                            st.markdown(f"**Location:** {candidate['Location']}")
+                            st.markdown(f"**[LinkedIn Profile]({candidate['LinkedIn']})**")
+                        
+                        with col2:
+                            st.markdown(f"**Fit Score:** {candidate['Fit Score']}")
+                        
+                        st.markdown("**Justification:**")
+                        st.info(candidate['justification'])
+            else:
+                st.info("No candidates met the minimum score threshold (8.8) for this job.")
+                
+                # We don't show tech-matched candidates here since they are generated
+                # during the LLM matching process now
+            
+            # Add a reset button to start over
+            if st.button("Reset and Process Again"):
+                st.session_state[job_key] = False
+                st.rerun()
 
-"""
-# Welcome to Streamlit!
-
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
-
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
-
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+if __name__ == "__main__":
+    main()