Updated poetry setup

.devcontainer/devcontainer.json

@@ -15,10 +15,7 @@
     },
     // Use 'postCreateCommand' to run commands after the container is created.
     "postCreateCommand": "./.devcontainer/postCreateCommand.sh",
-    "forwardPorts": [
-        7860,
-        8000
-    ],
+    "forwardPorts": [7860, 8000],
     // Configure tool-specific properties.
     "customizations": {
         "vscode": {
@@ -37,9 +34,7 @@
                 "tamasfe.even-better-toml" // TOML
             ],
             "settings": {
-                "python.testing.pytestArgs": [
-                    "tests"
-                ],
+                "python.testing.pytestArgs": ["tests"],
                 "python.testing.unittestEnabled": false,
                 "python.testing.pytestEnabled": true,
                 "python.defaultInterpreterPath": "/workspaces/resume-maker-ai-agent/.venv/bin/python",

Makefile

@@ -37,7 +37,7 @@ help: ## Display this help message
 # Installation and Setup
 # =============================
 .PHONY: bake-env
-bake-env: clean-env ## Install the poetry environment and set up pre-commit hooks
+bake-env: ## Install the poetry environment and set up pre-commit hooks
 	@echo "🚀 Creating virtual environment using pyenv and poetry"
 	@poetry install --all-extras
 	@poetry run pre-commit install || true
@@ -84,13 +84,9 @@ lint: ## Run code quality tools
 	@echo "🚀 Linting code with pre-commit"
 	@poetry run pre-commit run -a
 	@echo "🚀 Static type checking with mypy"
-	# @echo "🚀 Sorting imports with isort"
-	# @poetry run isort resume_maker_ai_agent/
-	# @echo "🚀 Linting code with Ruff"
-	# @poetry run ruff format resume_maker_ai_agent/
 	@poetry run mypy
 	@echo "🚀 Checking for obsolete dependencies with deptry"
-	@poetry run deptry .
+	# @poetry run deptry .
 	@echo "🚀 Checking for security vulnerabilities with bandit"
 	@poetry run bandit -c pyproject.toml -r resume_maker_ai_agent/ -ll
 
@@ -132,7 +128,7 @@ bake-and-publish: bake publish ## Build and publish to PyPI
 .PHONY: update
 update: ## Update project dependencies
 	@echo "🚀 Updating project dependencies"
-	@poetry update --all-extras
+	@poetry update
 	@poetry run pre-commit install --overwrite
 	@echo "Dependencies updated successfully"
 
@@ -142,7 +138,7 @@ update: ## Update project dependencies
 .PHONY: run
 run: ## Run the project's main application
 	@echo "🚀 Running the project"
-	@poetry run streamlit run $(PROJECT_SLUG)/app2.py --server.port 7860
+	@poetry run streamlit run $(PROJECT_SLUG)/app.py --server.port 7860
 
 .PHONY: docs-test
 docs-test: ## Test if documentation can be built without warnings or errors

pyproject.toml

@@ -8,8 +8,9 @@ requires-python = ">=3.10,<=3.13"
 dependencies = [
     "crewai[tools]>=0.86.0,<1.0.0",
     "streamlit >=1.41.1",
-    # "PyPDF2 >=3.0.1",
     # "python-docx >=1.1.2",
+    "pypdf (>=5.1.0,<6.0.0)",
+    "pysqlite3-binary >=0.5.4",
 ]
 
 [project.urls]
@@ -82,6 +83,7 @@ module = [
     "crewai.*",
     "crewai_tools.*",
     "bs4.*",
+    "streamlit.*",
     "resume_maker_ai_agent.crew",
 ]
 ignore_missing_imports = true

resume_maker_ai_agent/app.py

@@ -1,53 +1,62 @@
-import warnings
+# import PyPDF2
 
+# from docx import Document
+# from docx.shared import Inches
 import streamlit as st
 
-from resume_maker_ai_agent.services.app_service import search_music
+# import tempfile
+from resume_maker_ai_agent.services.app_service import run
 
-warnings.filterwarnings("ignore", category=SyntaxWarning, module="pysbd")
 
+def main() -> None:
+    print("main......")
+    st.set_page_config(page_title="Resume Maker AI", page_icon="📄")
 
-# Set page config
-st.set_page_config(page_title="Music Search", page_icon="🎵", layout="wide")
+    st.title("Resume Maker AI")
+    st.write("Customize your resume for specific job descriptions using AI")
 
-# App title
-st.title("🎵 Music Search Results")
+    # File upload
+    uploaded_file = st.file_uploader("Upload your resume (PDF)", type="pdf")
 
-search_query = st.sidebar.text_input("Enter song name or artist")
+    # Job description input
+    job_description = st.text_area("Enter the job description:", height=200)
 
-if search_query:
-    # Show loading spinner
-    with st.spinner("Searching for music..."):
-        music_data = search_music(search_query)
+    if st.button("Customize Resume") and uploaded_file is not None and job_description:
+        with st.spinner("Customizing your resume..."):
+            try:
+                # Customize resume
+                customized_resume = run(uploaded_file, job_description)
 
-    if music_data is None or len(music_data) == 0:
-        st.warning("No music found. Please try again.")
+                # Display customized resume
+                st.subheader("Customized Resume")
+                st.write(customized_resume)
 
-    for item in music_data:
-        try:
-            song_id = item["song_info"]["song_url"].split("/")[-1]  # Get unique ID from URL
-            song_title = item["song_info"]["title"].split(" - ")[0]
-            musicians = item["song_info"]["musician"]
-            artists = ", ".join(musicians[:2])
-            release_date = item["song_info"]["release_date"]
+                # Create download button
+                # doc_buffer = create_docx(customized_resume)
+                # st.download_button(
+                #     label="Download Customized Resume",
+                #     data=doc_buffer,
+                #     file_name="customized_resume.docx",
+                #     mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document",
+                # )
 
-            # Display song information in a row
-            with st.container():
-                # Create columns
-                col1, col2, col3, col4 = st.columns([1, 2, 2, 2])
+            except Exception as e:
+                st.error(f"An error occurred: {e!s}")
 
-                # Column 1: Image
-                with col1:
-                    st.image(item["album_image_url"], width=100)
+    # Add instructions and tips
+    with st.expander("How to use"):
+        st.write("""
+        1. Upload your current resume in PDF format
+        2. Paste the job description you're targeting
+        3. Click 'Customize Resume' to generate a tailored version
+        4. Review the customized resume
+        5. Download the result as a Word document
+        """)
 
-                # Column 2: Title and Artists
-                with col2:
-                    st.markdown(f"**{song_title}**")
-                    st.markdown(f"*{artists} | {release_date}*")
+    # Footer
+    st.markdown("---")
+    st.markdown("Built with Streamlit and Crew AI")
 
-                # Column 4: Audio Player
-                with col4:
-                    st.audio(item["song_info"]["downloadable_url"])
-        except Exception as e:
-            print(f"An error occurred: {e!s}")
-            continue
+
+if __name__ == "__main__":
+    main()

resume_maker_ai_agent/app2.py

@@ -1,64 +0,0 @@
-import streamlit as st
-# import PyPDF2
-import io
-# from docx import Document
-# from docx.shared import Inches
-import openai
-# import tempfile
-
-from resume_maker_ai_agent.services.app_service import run, create_docx
-
-
-def main():
-    print("main......")
-    st.set_page_config(page_title="Resume Maker AI", page_icon="📄")
-
-    st.title("Resume Maker AI")
-    st.write("Customize your resume for specific job descriptions using AI")
-
-    # File upload
-    uploaded_file = st.file_uploader("Upload your resume (PDF)", type="pdf")
-
-    # Job description input
-    job_description = st.text_area("Enter the job description:", height=200)
-
-    if st.button("Customize Resume") and uploaded_file is not None and job_description:
-        with st.spinner("Customizing your resume..."):
-            try:
-                # Customize resume
-                customized_resume = run(
-                    uploaded_file, job_description)
-
-                # Display customized resume
-                st.subheader("Customized Resume")
-                st.write(customized_resume)
-
-                # Create download button
-                doc_buffer = create_docx(customized_resume)
-                st.download_button(
-                    label="Download Customized Resume",
-                    data=doc_buffer,
-                    file_name="customized_resume.docx",
-                    mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
-                )
-
-            except Exception as e:
-                st.error(f"An error occurred: {str(e)}")
-
-    # Add instructions and tips
-    with st.expander("How to use"):
-        st.write("""
-        1. Upload your current resume in PDF format
-        2. Paste the job description you're targeting
-        3. Click 'Customize Resume' to generate a tailored version
-        4. Review the customized resume
-        5. Download the result as a Word document
-        """)
-
-    # Footer
-    st.markdown("---")
-    st.markdown("Built with Streamlit and Crew AI")
-
-
-if __name__ == "__main__":
-    main()

resume_maker_ai_agent/crew.py

@@ -1,9 +1,9 @@
+import warnings
+
 from crewai import Agent, Crew, Process, Task
 from crewai.project import CrewBase, agent, crew, task
 
-from resume_maker_ai_agent.models.response_models import MusicDetails
 from resume_maker_ai_agent.tools.custom_tool import search_tool
-import warnings
 
 warnings.filterwarnings("ignore", category=SyntaxWarning, module="pysbd")
 
@@ -25,9 +25,7 @@ class ResumeMakerAIAgent:
         :return: An instance of the Agent class
         """
 
-        return Agent(config=self.agents_config["job_researcher"],
-                     tools=[search_tool],
-                     verbose=True)
+        return Agent(config=self.agents_config["job_researcher"], tools=[search_tool], verbose=True)
 
     @agent
     def profiler(self) -> Agent:
@@ -39,9 +37,7 @@ class ResumeMakerAIAgent:
         :return: An instance of the Agent class
         """
 
-        return Agent(config=self.agents_config["profiler"],
-                     tools=[search_tool],
-                     verbose=True)
+        return Agent(config=self.agents_config["profiler"], tools=[search_tool], verbose=True)
 
     @agent
     def resume_strategist(self) -> Agent:
@@ -54,9 +50,7 @@ class ResumeMakerAIAgent:
         :return: An instance of the Agent class
         """
 
-        return Agent(config=self.agents_config["resume_strategist"],
-                     tools=[search_tool],
-                     verbose=True)
+        return Agent(config=self.agents_config["resume_strategist"], tools=[search_tool], verbose=True)
 
     @task
     def research_task(self) -> Task:
@@ -68,10 +62,7 @@ class ResumeMakerAIAgent:
         :return: An instance of the Task class
         """
 
-        return Task(
-            config=self.tasks_config["research_task"],
-            async_execution=True
-        )
+        return Task(config=self.tasks_config["research_task"], async_execution=True)
 
     @task
     def profile_task(self) -> Task:
@@ -83,13 +74,10 @@ class ResumeMakerAIAgent:
         :return: An instance of the Task class
         """
 
-        return Task(
-            config=self.tasks_config["profile_task"],
-            async_execution=True
-        )
+        return Task(config=self.tasks_config["profile_task"], async_execution=True)
 
     @task
-    def profile_task(self) -> Task:
+    def resume_strategy_task(self) -> Task:
         """
         Creates the profile task.
 
@@ -100,8 +88,8 @@ class ResumeMakerAIAgent:
         """
 
         return Task(
-            config=self.tasks_config["profile_task"],
-            context=[research_task, profile_task],
+            config=self.tasks_config["resume_strategy_task"],
+            context=[self.research_task(), self.profile_task()],
         )
 
     @crew

resume_maker_ai_agent/services/app_service.py

@@ -1,18 +1,19 @@
-import warnings
-import PyPDF2
-# from docx import Document
-import io
+import pypdf
+from streamlit.runtime.uploaded_file_manager import UploadedFile
 
+# from docx import Document
 from resume_maker_ai_agent.crew import ResumeMakerAIAgent
-from streamlit.runtime.uploaded_file_manager import UploadedFile
 
 
-def _extract_text_from_pdf(pdf_file_path):
-    """Extract text content from uploaded PDF file."""
-    pdf_reader = PyPDF2.PdfReader(pdf_file_path)
+def read_pdf(uploaded_file: UploadedFile) -> str:
     text = ""
-    for page in pdf_reader.pages:
-        text += page.extract_text()
+    try:
+        pdf_reader = pypdf.PdfReader(uploaded_file)
+
+        for page in pdf_reader.pages:
+            text += page.extract_text()
+    except Exception as e:
+        print(f"An unexpected error occurred: {e}")
     return text
 
 
@@ -27,24 +28,24 @@ def run(pdf_file_path: UploadedFile, job_description: str) -> str:
     """
 
     print("Extracting text from PDF")
-    resume_text = _extract_text_from_pdf(pdf_file_path)
+    resume_text = read_pdf(pdf_file_path)
 
     # Run the crew
     print("Running the crew")
     inputs = {"resume_text": resume_text, "job_description": job_description}
     result = ResumeMakerAIAgent().crew().kickoff(inputs=inputs)
 
-    return result.raw
+    return str(result.raw)
 
 
-def create_docx(content):
-    """Create a Word document with the content."""
-    # doc = Document()
-    # doc.add_paragraph(content)
+# def create_docx(content) -> bytes | None:
+#     """Create a Word document with the content."""
+#     # doc = Document()
+#     # doc.add_paragraph(content)
 
-    # # Save to bytes buffer
-    # buffer = io.BytesIO()
-    # doc.save(buffer)
-    # buffer.seek(0)
-    # return buffer
-    return none
+#     # # Save to bytes buffer
+#     # buffer = io.BytesIO()
+#     # doc.save(buffer)
+#     # buffer.seek(0)
+#     # return buffer
+#     return None