feat: integrate job recommendation functionality into API response

.github/workflows/ci_be.yml

@@ -62,10 +62,14 @@ jobs:
         run: |
           sed -i '/<!-- Pytest Coverage Comment:Begin -->/,/<!-- Pytest Coverage Comment:End -->/c\<!-- Pytest Coverage Comment:Begin -->\n\${{ steps.coverageComment.outputs.coverageHtml }}\n<!-- Pytest Coverage Comment:End -->' ../README.md
 
+      - name: Debug current branch
+        run: |
+          echo "head ref: ${{ github.head_ref }} "
+          echo "ref name: ${{ github.ref_name }} "
+
       - name: Commit & Push changes to Readme
-        uses: actions-js/push@master
+        uses: ad-m/github-push-action@master
         with:
-          branch: |
-            ${{ github.ref_name == 'main' && 'dev/backend' || github.ref_name.startsWith('be/') && github.ref_name || 'dev/backend' }}
+          branch: ${{ github.head_ref || github.ref_name }}
           message: Update coverage on Readme
           github_token: ${{ secrets.GITHUB_TOKEN }}

be_repo/app.py

@@ -9,6 +9,7 @@
 from configs.database import get_resume_database, get_user_database
 from graphs.qa_graph import create_graph
 from modules.evaluator import evaluate_resume, evaluate_resume_with_jd
+from modules.job_recommendation_system import job_recommend
 from modules.langgraph_qa import get_answer_from_langgraph
 from modules.upload import upload_parse_resume
 
@@ -159,9 +160,17 @@ def ask_question():
         return jsonify({"error": "No user ID provided."}), 400
     if not question:
         return jsonify({"error": "No question provided."}), 400
+    # Load resume from database
+    resume = resume_collection.find_one({"user_id": user_id})
+    if not resume:
+        return jsonify({"error": "No resume found for this user."}), 404
+
+    resume_text = resume.get('resume_text', '')
+    if not resume_text:
+        return jsonify({"error": "Resume text is empty."}), 400
 
     # Get answer using LangGraph
-    response = get_answer_from_langgraph(qa_graph, resume_collection, user_state_collection, user_id, question)
+    response = get_answer_from_langgraph(qa_graph, resume_text, user_state_collection, user_id, question)
 
     return jsonify({"response": response}), 200
 
@@ -202,8 +211,17 @@ def interview_question_suggestion():
     if not user_id:
         return jsonify({"error": "No user ID provided."}), 400
 
+    # Load resume from database
+    resume = resume_collection.find_one({"user_id": user_id})
+    if not resume:
+        return jsonify({"error": "No resume found for this user."}), 404
+
+    resume_text = resume.get('resume_text', '')
+    if not resume_text:
+        return jsonify({"error": "Resume text is empty."}), 400
+
     # Get answer using LangGraph
-    response = get_answer_from_langgraph(qa_graph, resume_collection, user_state_collection, user_id, prompt)
+    response = get_answer_from_langgraph(qa_graph, resume_text, user_state_collection, user_id, prompt)
 
     return jsonify({"response": response}), 200
 
@@ -218,10 +236,17 @@ def job_suggestion():
     if not user_id:
         return jsonify({"error": "No user ID provided."}), 400
 
+    # Load resume from database
+    resume = resume_collection.find_one({"user_id": user_id})
+    if not resume:
+        return jsonify({"error": "No resume found for this user."}), 404
+
     # Get answer using LangGraph
-    response = 'Example response'
+    resume_text = resume.get('resume_text', '')
+    if not resume_text:
+        return jsonify({"error": "Resume text is empty."}), 400
 
-    return jsonify({"response": response}), 200
+    return jsonify({"response": job_recommend(resume_text, user_id)}), 200
 
 
 if __name__ == '__main__':

be_repo/modules/job_recommendation_system.py

@@ -1,75 +1,55 @@
 # job_recommendation_system.py
 
-from neo4j_model import Neo4jModel
-from resume_processor import ResumeProcessor
-from retrieval_engine import RetrievalEngine
-from recommendation_generator import RecommendationGenerator
-from view import CLIView
-import sys
+import logging
 
-def main():
-
-
-    # Redirect standard output to a file
-    sys.stdout = open('output.log', 'w')
-
-    # Your code here
-    print("Lots of output")
+from .neo4j_model import Neo4jModel
+from .recommendation_generator import RecommendationGenerator
+from .resume_processor import ResumeProcessor
+from .retrieval_engine import RetrievalEngine
+from .view import CLIView
 
 
+def job_recommend(resume_text, user_id):
     # Setup Logging
-    import logging
     logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
     logger = logging.getLogger(__name__)
-
+
+    # Get Resume Input from User
+    if not resume_text.strip():
+        logger.error(f'No resume text provided, user_id: {user_id}.')
+        return 'Error: No resume text provided.'
+
     # Neo4j Connection Details
     NEO4J_URI = "neo4j+ssc://7bf5a48e.databases.neo4j.io"  # Replace with your Neo4j URI
-    NEO4J_USERNAME = "neo4j"             # Replace with your Neo4j username
-    NEO4J_PASSWORD = "oxsK7V5_86emZlYQlvCfQHfVWS95wXz29OhtU8GAdFc"          # Replace with your Neo4j password
+    NEO4J_USERNAME = "neo4j"  # Replace with your Neo4j username
+    NEO4J_PASSWORD = "oxsK7V5_86emZlYQlvCfQHfVWS95wXz29OhtU8GAdFc"  # Replace with your Neo4j password
 
     # Initialize Model
     neo4j_model = Neo4jModel(
         uri=NEO4J_URI,
         username=NEO4J_USERNAME,
         password=NEO4J_PASSWORD
     )
-    
+
     # Initialize Controller Components
     resume_processor = ResumeProcessor()
     retrieval_engine = RetrievalEngine(resume_processor, neo4j_model)
     recommendation_generator = RecommendationGenerator()
-    
+
     # Initialize View
     view = CLIView()
-
-    # Get Resume Input from User
-    resume_text = view.get_resume_input()
-
-    if not resume_text.strip():
-        logger.error("No resume text provided.")
-        print("Error: No resume text provided.")
-        return
-
-    # Perform Mixed Retrieval for 'JD' Node Label
-    node_label = "JD"  # Adjust as needed; could be dynamic based on user input or other criteria
-    similar_docs, graph_results = retrieval_engine.perform_mixed_retrieval(resume_text, node_label=node_label)
-
+
+    # Perform Mixed Retrieval
+    similar_docs, graph_results = retrieval_engine.perform_mixed_retrieval(resume_text, node_label='JTitle')
+
     if not similar_docs and not graph_results:
-        print("No job recommendations found based on your resume.")
-        return
-
+        return 'No job recommendations found based on your resume.'
+
     # Generate Recommendations
     try:
         recommendations = recommendation_generator.generate_recommendations(similar_docs, graph_results)
     except Exception as e:
-        print("Error: Failed to generate job recommendations.")
-        return
-
-    # Display Recommendations
-    view.display_recommendations(recommendations)
+        return 'Error: Failed to generate job recommendations.'
 
-    # Close the file
-    sys.stdout.close()
-
-if __name__ == "__main__":
-    main()
+    # Display Recommendations
+    return view.display_recommendations(recommendations)

be_repo/modules/langgraph_qa.py

@@ -1,27 +1,24 @@
-def get_answer_from_langgraph(qa_graph, resume_collection, user_state_collection, user_id, question):
-    resume = resume_collection.find_one({"user_id": user_id})
+def get_answer_from_langgraph(qa_graph, resume_text, user_state_collection, user_id, question):
     user_state = user_state_collection.find_one({"user_id": user_id})
-    resume_text = resume.get('resume_text', '')
-    state = user_state.get('state', '')
     thread_id = user_state.get('thread_id', '')
 
     config = {"configurable": {"thread_id": user_id + thread_id}}
 
     # If state is 0, send resume to LLM first
-    if state == '0':
-        events = qa_graph.stream(
-            {"messages": [("user", resume_text)]}, config, stream_mode="values"
-        )
-        # Update state to 1
-        new_state = {
-            "user_id": user_id,
-            "state": '1',
-            "thread_id": thread_id
-        }
-        user_state_collection.replace_one({"user_id": user_id}, new_state, upsert=True)
-        for event in events:
-            if event["messages"][-1].type == "ai":
-                print('User ask for the first time!')
+    # if state == '0':
+    events = qa_graph.stream(
+        {"messages": [("user", resume_text)]}, config, stream_mode="values"
+    )
+    # Update state to 1
+    new_state = {
+        "user_id": user_id,
+        "state": '1',
+        "thread_id": thread_id
+    }
+    user_state_collection.replace_one({"user_id": user_id}, new_state, upsert=True)
+    for event in events:
+        if event["messages"][-1].type == "ai":
+            print('User ask for the first time!')
     # Then send the question
     events = qa_graph.stream(
         {"messages": [("user", question)]}, config, stream_mode="values"

be_repo/modules/retrieval_engine.py

@@ -1,12 +1,10 @@
 # retrieval_engine.py
 
-from langchain_neo4j import GraphCypherQAChain
-from langchain_openai import ChatOpenAI
-from langchain.chains.retrieval import create_retrieval_chain 
 from langchain.chains.combine_documents import create_stuff_documents_chain
-from configs.openai_key import get_openai_api_key  # New import
+from langchain.chains.retrieval import create_retrieval_chain
 from langchain.prompts import PromptTemplate
 
+
 class RetrievalEngine:
     def __init__(self, resume_processor, neo4j_model):
         """
@@ -21,7 +19,7 @@ def __init__(self, resume_processor, neo4j_model):
 
         # Initialize Language Model (already initialized in Neo4jModel)
         self.llm = self.neo4j_model.llm
-        
+
         # Initialize GraphCypherQAChain (already initialized in Neo4jModel)
         self.graph_chain = self.neo4j_model.get_graph_chain()
 
@@ -53,12 +51,12 @@ def __init__(self, resume_processor, neo4j_model):
             {context}
             \"\"\"
             """,
-            input_variables=["input"]  
+            input_variables=["input"]
         )
 
-         # Create a documents chain
+        # Create a documents chain
         self.combine_docs_chain = create_stuff_documents_chain(self.llm, prompt=prompt)
-        
+
         # Initialize Retrieval Chain
         # Default node_label is 'JD'; can be adjusted as needed
         self.retrieval_chain = create_retrieval_chain(
@@ -79,13 +77,13 @@ def perform_mixed_retrieval(self, resume_text, node_label="JD"):
         """
         # Process resume into a Document
         doc = self.resume_processor.process_resume(resume_text)
-        
+
         if not doc:
             return [], {}
-        
+
         # Store the Document in the appropriate vector store
         self.neo4j_model.store_documents([doc], node_label=node_label)
-        
+
         # Access the schema property correctly
         schema = self.neo4j_model.graph.get_schema
 
@@ -94,16 +92,15 @@ def perform_mixed_retrieval(self, resume_text, node_label="JD"):
         similar_docs = similar_docs_result.get("output", [])
         print("similar_docs_result:", similar_docs_result)
         print("Keys in similar_docs_result:", similar_docs_result.keys())
-
-
 
         for doc in similar_docs:
             print("Document Metadata:", doc.metadata)
 
-        query = f"Based on the following resume, recommend relevant job positions: {resume_text}"
+        query = (f"Based on the following resume, recommend relevant job positions based on skills and experience, "
+                 f"while ignoring the location: {resume_text}")
         graph_response = self.graph_chain.invoke({"query": query, "schema": schema})
         # After graph query
         print("Graph Response:")
         print(graph_response)
-       
-        return similar_docs, graph_response
+
+        return similar_docs, graph_response

be_repo/modules/view.py

@@ -28,8 +28,8 @@ def display_recommendations(self, recommendations):
         Display job recommendations to the user.
         """
         if not recommendations:
-            print("No job recommendations found based on your resume.")
-            return
-        print("\nRecommended Jobs for You:")
+            return 'No job recommendations found based on your resume.'
+        res = '\nRecommended Jobs for You:\n'
         for idx, job in enumerate(recommendations, start=1):
-            print(f"{idx}. {job}")
+            res += f'{idx}. {job}\n'
+        return res

be_repo/requirements.txt

@@ -1,18 +1,22 @@
-Flask==3.0.3
+chromedriver_autoinstaller==0.6.4
+Flask==3.1.0
 Flask_Cors==5.0.0
-openai==1.54.4
-pandas==2.2.3
+langchain==0.3.9
+langchain_community==0.3.8
+langchain_neo4j==0.1.1
+langchain_openai==0.2.10
+langgraph==0.2.53
+neo4j==5.27.0
+openai==1.55.3
+pandas==1.5.3
+protobuf==3.20.2
 pymongo==4.10.1
 PyPDF2==3.0.1
-qdrant_client==1.12.0
+pytest==8.3.4
+PyVirtualDisplay==3.0
+qdrant_client==1.12.1
+selenium==4.27.1
 tqdm==4.66.2
-selenium==4.26.1
-google-auth==2.36.0
-google-auth-oauthlib==1.2.1
-chromedriver-autoinstaller==0.6.4
-langgraph==0.2.48
-langchain==0.3.7
-langsmith==0.1.143
-langchain_community==0.3.7
-langchain-openai==0.2.8
-pyvirtualdisplay
+google-auth>=2.0.0
+google-auth-oauthlib>=0.4.0
+google-auth-httplib2>=0.1.0

be_repo/tests/test_e2e.py

@@ -42,7 +42,7 @@ def driver():
 
 @pytest.fixture
 def wait(driver):
-    return WebDriverWait(driver, 30)
+    return WebDriverWait(driver, 50)
 
 
 def upload_resume(driver, wait):

fe_repo/src/App.tsx

@@ -108,7 +108,7 @@ function App() {
     };
     // show loading message
     setMessages((messages) => [...messages, {
-      text: 'Suggest jobs',
+      text: 'Suggested jobs',
       isUser: true
     }, sendingMessage]);
     // show loading message
@@ -118,7 +118,7 @@ function App() {
     } else {
       sendingMessage.text = (
         <>
-          <div className="font-bold">Suggested questions:<br/>{response.response}</div>
+          <div className="font-bold">Suggested jobs:<br/>{response.response}</div>
         </>
       );
     }

fe_repo/src/functions/api.test.ts

@@ -1,5 +1,5 @@
 import {expect, test} from 'vitest';
-import {analyze, getHasResume, getUserId, login, sendMessage, suggest, uploadFile, suggestJob} from "./api.ts";
+import {analyze, getHasResume, getUserId, login, sendMessage, suggest, suggestJob, uploadFile} from "./api.ts";
 
 test('sendMessage', () => {
   expect(sendMessage("I'm adam")).toBeDefined();