Changes

app.py

@@ -19,7 +19,7 @@ def main():
   with col2:
     # Radio buttons for analysis type
     analysis_type = st.radio("Choose analysis type:", 
-                                ("Code review", "Code refinement","Documentation", "Unit tests"))
+                                ("Code review", "Code refinement","Documentation", "Resume Writer"))
 
   # Submit button
   if st.button("Submit"):
@@ -40,6 +40,10 @@ def main():
       res=generate_documentation(code, prompt)
       #st.text_area(label="Result",value=res, height=300)
       st.markdown(res)
+    elif analysis_type == "Resume Writer":
+      res=resume_writer(code, prompt)
+      #st.text_area(label="Result",value=res, height=300)
+      st.markdown(res)
   
     st.success(f"Code analysis for {analysis_type} submitted successfully!")  
 

gemini.py

@@ -23,13 +23,13 @@ class GeminiModel:
     """
 
     def __init__(self,
-                 model: Optional[str] = 'gemini-pro',
+                 model_name: Optional[str] = 'gemini-pro',
                  ):
         
         
         load_dotenv()
         genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
-        self.model = genai.GenerativeModel(model)
+        self.model = genai.GenerativeModel(model_name) # type: ignore
         
 
     def execute(self, prompt: str) -> str:

handler.py

@@ -55,6 +55,23 @@ def generate_documentation(code,c_prompt):
 
     return res
 
+def resume_writer(code,c_prompt):
+    
+    if code is None or len(code) < 5 or code.isspace():
+        raise InvalidCodeException("No code provided")
+
+    if c_prompt is not None and len(c_prompt) > 30:
+        prompt = custom_review_prompt(code.strip(), c_prompt.strip())
+    else:
+        prompt = resume_prompt(code.strip())
+
+    model = gemini.GeminiModel()
+    try:
+        res = model.execute(prompt)
+    except Exception as e:
+        raise CodeReviewException(str(e))
+
+    return res
 class InvalidCodeException(Exception):
     pass
 

prompts.py

@@ -170,13 +170,140 @@ A refined and optimized version of the code.
 def default_doc_prompt(code):
 
     
-    prompt= f""""For the following python code:\n{code}\n"
-            "Generate proper formal documentation." 
-            "The documentation should describe what the code does and explain the functionality."
-            "Include all details which should go in formal documentation"
-    """ 
+    prompt= f""""**Role:** Technical Writer
+
+**Task:** Generate formal documentation for the provided Python code.
+
+**Context:** The provided Python code snippet requires comprehensive formal documentation. This documentation should clearly explain the code's functionality and purpose, including all relevant details expected in formal documentation. 
+
+**Instructions:**
+* Take a deep breath and thoroughly analyze the provided Python code. 
+* Understand the code's logic, flow, and intended functionality.
+* Generate formal documentation that is clear, concise, and informative.
+* Explain the code's purpose, functionality, and any relevant details.
+* Use proper formatting, structure, and language appropriate for formal documentation.
+* Ensure the documentation is comprehensive and addresses all aspects of the code.
+
+**Guidelines:**
+* Follow the guidelines for writing formal documentation.
+* Use clear and concise language.
+* Avoid ambiguity and ensure the documentation is easy to understand.
+* Use proper grammar and punctuation.
+* Structure the documentation logically and use headings and subheadings for clarity.
+* Include relevant comments and explanations within the code itself.
+
+**Examples:**
+* Provide examples of how the code can be used.
+* Include code snippets to illustrate specific functionalities.
+* Use diagrams or flowcharts to visualize complex logic.
+
+**Steps:**
+1. Analyze the Python code and understand its functionality.
+2. Gather all relevant information about the code, including its purpose, inputs, outputs, and any dependencies.
+3. Structure the documentation logically, using headings and subheadings to organize the information.
+4. Write clear and concise explanations of the code's functionality.
+5. Include relevant examples and code snippets to illustrate the code's usage.
+6. Proofread and edit the documentation carefully to ensure accuracy and clarity.
+
+**Goal:** Generate comprehensive and accurate formal documentation for the provided Python code in one iteration. 
+
+**Additional Notes:**
+* Consider using a documentation generator tool to assist with the formatting and structure of the documentation.
+* Review the documentation with a technical expert to ensure its accuracy and completeness. 
+* If further clarification is needed, feel free to ask for additional information or context regarding the code. 
+   
+**Code:** {code}
+      """ 
+
+    promptt = f"""**Instruction**: As an adept programmer, meticulously dissect the given code snippet and meticulously document it in accordance with the provided Output Format. If you encounter any functionalities or concepts you're uncertain about, provide the best possible descriptions and indicate areas necessitating further investigation. Avoid conjectures and ensure explanations are substantiated. If certain sections of the **Output Format** aren't applicable, simply exclude them from the output.
+
+**Output Format**:
+##Dependencies:
+1. Frameworks/Libraries:
+List any external libraries, frameworks, or modules required by the code along with a succinct description for each.
+2. Cloud Services:
+Enumerate all cloud services invoked within the code, providing a concise description for each service's purpose or functionality in one line.
+
+## Functions used:
+Present a clear and organized table showcasing the name of each function, along with its corresponding input/arguments, expected output, and the function's purpose. The table should be formatted neatly for easy readability. All "|" should fall on same line. Use the following column headings:
+| Function Name | Input/Arguments | Output | Purpose |
+
+##Steps Conducted:
+Translate the developer's actions for each function into a detailed and easily readable format. Additionally, include a flow diagram to illustrate the code's flow step by step.
+
+##Suggestions for Improvement:
+Offer recommendations for enhancing the code, potential optimizations, or logic enhancements. Provide a code snippet for each suggestion, avoiding unnecessary assumptions.
+
+Task: Showcase your proficiency by creating a comprehensive documentation set for the provided code snippet. Adhere strictly to the outlined Output Format.
+**Code**: {code}
+
+    """
+    return prompt
+
+def resume_prompt(details, guidelines=None):
+    prompt=f"""**Role:** Resume Writer
+
+**Task:** Generate a well-structured, impactful resume brief based on the provided candidate's resume.
+
+**Context:** You are a professional resume writer with extensive experience in crafting compelling resumes that effectively showcase candidates' skills and experience. You have access to the candidate's resume and are familiar with industry best practices for resume writing.
+
+**Instructions:**
+
+* Take a deep breath and thoroughly analyze the candidate's resume.
+* Identify the most relevant and impactful information that aligns with the target job description.
+* Use a formal and impactful tone while highlighting the candidate's key strengths and accomplishments.
+* Structure the resume brief in a clear and concise manner, using bullet points and headings for easy readability.
+* Ensure that the resume brief is tailored to the specific job the candidate is applying for.
+
+**Guidelines:**
+
+* Follow industry best practices for resume writing.
+* Use clear and concise language.
+* Avoid jargon and technical terms that the hiring manager may not understand.
+* Proofread carefully for any errors in grammar or spelling.
+
+**Example:**
+
+**Candidate:** John Doe
+
+**Target Job:** Software Engineer
+
+**Resume Brief:**
+
+* **Summary:** Highly motivated and results-oriented software engineer with 5+ years of experience in developing and maintaining complex software applications. Proven ability to work independently and as part of a team.
+* **Technical Skills:** Proficient in Java, Python, C++, and SQL. Experience with Agile methodologies and cloud computing platforms.
+* **Experience:**
+    * Developed a new customer relationship management system that resulted in a 20% increase in sales.
+    * Led a team of engineers in the development of a new mobile app that has been downloaded over 1 million times.
+* **Education:** Bachelor of Science in Computer Science from the University of California, Berkeley.
+
+**Steps:**
+
+1. Review the candidate's resume and identify their key skills and experience.
+2. Determine the target job description and identify the most relevant information from the resume.
+3. Write a concise and impactful summary of the candidate's qualifications.
+4. Highlight the candidate's technical skills and experience.
+5. Provide specific examples of the candidate's accomplishments.
+6. Proofread the resume brief carefully.
+
+**Goal:**
+
+To generate a well-structured, impactful resume brief that effectively showcases the candidate's qualifications and increases their chances of landing an interview.
+
+**Additional Notes:**
+
+* Consider including a call to action at the end of the resume brief, such as "Please contact me to discuss my qualifications in more detail."
+* You may also want to include a link to the candidate's full resume.
+* If you have any questions about the candidate's qualifications, be sure to reach out to them for clarification.
+
+**By following these guidelines and steps, you can generate an effective prompt that will help the model generate a well-structured, impactful resume brief in one go.**
+
+ Details: {details}
+"""
     return prompt
 
+
+
 def validation_prompt(code, guidelines=None):
     prompt=f"""Check if the below input contains code snippets or not. Return only Yes or NO and nothing else.
     Input: {code}