Added input for additional instructions from user for writer agent, in addition to stock prompt from configuration.py

.gitignore

@@ -1,3 +1,4 @@
 __pycache__
 .vscode
-.venv
+.venv
+data

functions/gradio.py

@@ -14,7 +14,7 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 
-def process_inputs(linkedin_pdf, github_url, job_post_text):
+def process_inputs(linkedin_pdf, github_url, job_post_text, user_instructions):
     """
     Process the input files and URLs from the Gradio interface.
     
@@ -22,6 +22,7 @@ def process_inputs(linkedin_pdf, github_url, job_post_text):
         linkedin_pdf: Uploaded LinkedIn resume export PDF file
         github_url (str): GitHub profile URL  
         job_post_text (str): Job post text content
+        user_instructions (str): Additional instructions from the user
     
     Returns:
         str: Formatted output with file and URL information
@@ -78,14 +79,22 @@ def process_inputs(linkedin_pdf, github_url, job_post_text):
         result += "❌ Job post not provided\n"
         logger.info("No job post text provided")
 
+    # Process user instructions
+    if user_instructions and user_instructions.strip():
+        result += "✅ Additional instructions provided\n"
+        logger.info(f"User instructions provided ({len(user_instructions)} characters)")
+    else:
+        result += "ℹ️ No additional instructions provided\n"
+        logger.info("No additional instructions provided")
+
     logger.info("Input processing completed")
 
-    resume = write_resume(extraction_result)
+    resume = write_resume(extraction_result, user_instructions)
     
     return result
 
 
-def get_processed_data(linkedin_pdf, github_url, job_post_text):
+def get_processed_data(linkedin_pdf, github_url, job_post_text, user_instructions):
     """
     Get structured data from all inputs for further processing.
     
@@ -93,6 +102,7 @@ def get_processed_data(linkedin_pdf, github_url, job_post_text):
         linkedin_pdf: Uploaded LinkedIn resume export PDF file
         github_url (str): GitHub profile URL  
         job_post_text (str): Job post text content
+        user_instructions (str): Additional instructions from the user
     
     Returns:
         dict: Structured data containing all processed information
@@ -101,6 +111,7 @@ def get_processed_data(linkedin_pdf, github_url, job_post_text):
         "linkedin": None,
         "github": None,
         "job_post": job_post_text.strip() if job_post_text and job_post_text.strip() else None,
+        "user_instructions": user_instructions.strip() if user_instructions and user_instructions.strip() else None,
         "errors": []
     }
     

functions/writer_agent.py

@@ -2,19 +2,21 @@
 
 import json
 import logging
+import os
 from smolagents import CodeAgent
 from configuration import AGENT_MODEL, INSTRUCTIONS
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-def write_resume(content: str) -> str:
+def write_resume(content: str, user_instructions: str = None) -> str:
 
     """
     Generates a resume based on the provided content.
 
     Args:
         content (str): The content to be used for generating the resume.
+        user_instructions (str, optional): Additional instructions from the user.
 
     Returns:
         str: The generated resume.
@@ -32,10 +34,31 @@ def write_resume(content: str) -> str:
             planning_interval=5
         )
 
+        # Prepare instructions - combine default with user instructions
+        instructions = INSTRUCTIONS
+        if user_instructions and user_instructions.strip():
+            instructions += f"\n\nAdditional user instructions:\n{user_instructions.strip()}"
+            logger.info("Added user instructions to agent prompt")
+
         submitted_answer = agent.run(
-            INSTRUCTIONS + '\n' + json.dumps(content['structured_text']),
+            instructions + '\n' + json.dumps(content['structured_text']),
         )
 
         logger.info("submitted_answer: %s", submitted_answer)    
+        
+        # Create data directory if it doesn't exist
+        data_dir = 'data'
+        if not os.path.exists(data_dir):
+            os.makedirs(data_dir)
+            logger.info("Created data directory: %s", data_dir)
+        
+        # Save the resume to resume.md in the data directory
+        resume_file_path = os.path.join(data_dir, 'resume.md')
+        try:
+            with open(resume_file_path, 'w', encoding='utf-8') as f:
+                f.write(submitted_answer)
+            logger.info("Resume saved to: %s", resume_file_path)
+        except Exception as e:
+            logger.error("Failed to save resume to file: %s", e)
     
     return submitted_answer

resumate.py

@@ -67,12 +67,24 @@ with gr.Blocks() as demo:
         placeholder="Copy and paste the job post text here"
     )
 
+    gr.Markdown("""
+    ## 4. Additional instructions (optional)
+
+    Provide any additional instructions or adjustments for the resume writer agent. This could include specific formatting preferences, emphasis on certain skills, or any other customizations you'd like.
+    """)
+
+    user_instructions = gr.Textbox(
+        label="Additional Instructions",
+        placeholder="Enter any additional instructions for the resume writer (optional)",
+        lines=3
+    )
+
     submit_btn = gr.Button("Submit")
-    output = gr.Textbox(label="Output", lines=20, max_lines=50, show_copy_button=True)
+    output = gr.Textbox(label="Output", lines=5, max_lines=50, show_copy_button=True)
 
     submit_btn.click( # pylint: disable=no-member
         process_inputs,
-        inputs=[linkedin_pdf, github_profile, job_post],
+        inputs=[linkedin_pdf, github_profile, job_post, user_instructions],
         outputs=output
     )