Update src/app.py

src/app.py

@@ -216,118 +216,124 @@ class ImprovedChatMemory:
 
 class ImprovedCPUChatbot:
     def __init__(self):
-        self.model_name = "distilgpt2"
-        self.model = None
-        self.tokenizer = None
-        self.pipeline = None
         self.nlp_processor = ImprovedNLPProcessor()
         self.memory = ImprovedChatMemory()
         self.is_loaded = False
         
+        # Comprehensive template responses for better coverage
         self.template_responses = {
             'experience': "To improve your experience section: Use bullet points with action verbs, quantify achievements with numbers, focus on results rather than duties, and tailor content to match job requirements.",
             'ats': "Make your resume ATS-friendly by: Using standard section headings, including relevant keywords naturally, avoiding images and complex formatting, using common fonts like Arial, and saving as PDF.",
             'skills': "Enhance your skills section by: Organizing technical and soft skills separately, matching skills to job descriptions, providing proficiency levels, and including both hard and soft skills relevant to your target role.",
             'keywords': "Add relevant keywords by: Studying job descriptions in your field, using industry-specific terms, including both acronyms and full terms, and incorporating them naturally throughout your resume.",
-            'format': "Improve resume formatting with: Clear section headings, consistent bullet points, readable fonts, appropriate white space, and a clean, professional layout that's easy to scan."
+            'format': "Improve resume formatting with: Clear section headings, consistent bullet points, readable fonts, appropriate white space, and a clean, professional layout that's easy to scan.",
+            'summary': "Write an effective summary by: Starting with your professional title, highlighting 2-3 key achievements, mentioning relevant skills, and keeping it to 3-4 lines maximum.",
+            'education': "Optimize your education section by: Listing degree, institution, and graduation year, including relevant coursework for entry-level positions, mentioning GPA if above 3.5, and adding academic achievements.",
+            'projects': "Showcase projects effectively by: Describing the problem solved, technologies used, your specific role, quantifiable results or impact, and linking to live demos or repositories when possible.",
+            'cover_letter': "Create compelling cover letters by: Addressing specific job requirements, showing knowledge of the company, highlighting relevant achievements, and ending with a strong call to action.",
+            'interview': "Prepare for interviews by: Researching the company and role, practicing STAR method responses, preparing questions to ask, and reviewing your resume thoroughly to discuss any point confidently."
         }
     
-    @st.cache_resource
-    def load_model(_self):
-        """Load the model with better configuration"""
+    def initialize(self):
+        """Initialize the chatbot - lightweight version without model loading"""
         try:
-            with st.spinner("Loading AI model..."):
-                tokenizer = AutoTokenizer.from_pretrained(_self.model_name)
-                tokenizer.pad_token = tokenizer.eos_token
-                
-                model = AutoModelForCausalLM.from_pretrained(
-                    _self.model_name,
-                    torch_dtype=torch.float32,
-                    low_cpu_mem_usage=True
-                )
-                
-                text_generator = pipeline(
-                    "text-generation",
-                    model=model,
-                    tokenizer=tokenizer,
-                    device=-1,
-                    max_new_tokens=50,
-                    do_sample=True,
-                    temperature=0.8,
-                    top_p=0.85,
-                    top_k=50,
-                    repetition_penalty=1.2,
-                    pad_token_id=tokenizer.eos_token_id,
-                    no_repeat_ngram_size=3
-                )
-                
-                return model, tokenizer, text_generator
+            # Simulate brief initialization
+            import time
+            time.sleep(0.5)  # Brief delay to show initialization
+            self.is_loaded = True
+            return True
         except Exception as e:
-            st.error(f"Failed to load model: {str(e)}")
-            return None, None, None
-    
-    def initialize(self):
-        """Initialize the chatbot"""
-        if not self.is_loaded:
-            result = self.load_model()
-            if result[0] is not None:
-                self.model, self.tokenizer, self.pipeline = result
-                self.is_loaded = True
-                return True
-            else:
-                return False
-        return True
+            st.error(f"Failed to initialize: {str(e)}")
+            return False
     
     def get_template_response(self, user_input: str) -> str:
-        """Check if we can use a template response for common questions"""
+        """Enhanced template response matching"""
         user_lower = user_input.lower()
         
-        if any(word in user_lower for word in ['experience', 'work history', 'job history']):
-            return self.template_responses['experience']
-        elif any(word in user_lower for word in ['ats', 'applicant tracking', 'ats-friendly']):
-            return self.template_responses['ats']
-        elif any(word in user_lower for word in ['skills', 'technical skills', 'abilities']):
-            return self.template_responses['skills']
-        elif any(word in user_lower for word in ['keywords', 'keyword', 'terms']):
-            return self.template_responses['keywords']
-        elif any(word in user_lower for word in ['format', 'formatting', 'layout', 'design']):
-            return self.template_responses['format']
-        elif any(phrase in user_lower for phrase in ['improve my resume', 'better resume', 'hire me', 'get hired', 'land job']):
+        # More comprehensive keyword matching
+        response_mapping = [
+            (['experience', 'work history', 'job history', 'employment', 'career'], 'experience'),
+            (['ats', 'applicant tracking', 'ats-friendly', 'system'], 'ats'),
+            (['skills', 'technical skills', 'abilities', 'competencies'], 'skills'),
+            (['keywords', 'keyword', 'terms', 'buzzwords'], 'keywords'),
+            (['format', 'formatting', 'layout', 'design', 'structure'], 'format'),
+            (['summary', 'objective', 'profile', 'about', 'intro'], 'summary'),
+            (['education', 'degree', 'university', 'college', 'school'], 'education'),
+            (['projects', 'portfolio', 'work samples', 'personal projects'], 'projects'),
+            (['cover letter', 'covering letter', 'application letter'], 'cover_letter'),
+            (['interview', 'interviews', 'interviewing', 'job interview'], 'interview')
+        ]
+        
+        for keywords, response_key in response_mapping:
+            if any(keyword in user_lower for keyword in keywords):
+                return self.template_responses[response_key]
+        
+        # General improvement queries
+        if any(phrase in user_lower for phrase in ['improve my resume', 'better resume', 'hire me', 'get hired', 'land job']):
             return "To improve your resume for HR success: Use a clear, professional format with standard headings. Tailor your content to match job descriptions. Quantify achievements with numbers. Include relevant keywords naturally. Keep it to 1-2 pages. Use bullet points with action verbs. Proofread carefully for errors."
-        elif any(word in user_lower for word in ['help', 'advice', 'tips', 'suggestions']):
-            return "Key resume tips: Match your resume to each job application. Use metrics to show your impact. Include both technical and soft skills. Write a compelling summary. Use reverse chronological order. Keep formatting clean and simple."
+        
+        if any(phrase in user_lower for phrase in ['hr', 'hiring manager', 'recruiter', 'employer']):
+            return "To appeal to HR professionals: Use standard section headings they recognize. Include relevant keywords for ATS systems. Show clear career progression. Make it scannable with bullet points. Demonstrate measurable value and impact you can bring to their organization."
+        
+        if any(word in user_lower for word in ['help', 'advice', 'tips', 'suggestions']):
+            return "Key resume best practices: Customize for each job application. Use metrics to show impact. Include both technical and soft skills. Write compelling summaries. Use reverse chronological order. Keep formatting clean and professional. Focus on achievements, not just duties."
         
         return None
     
     def generate_response(self, user_input: str, resume_context: str = "") -> str:
-        """Generate response with better quality control"""
+        """Generate response using templates and contextual advice"""
         if not self.is_loaded:
-            return "Please initialize the AI model first by clicking 'Initialize AI'."
+            return "Please initialize the AI assistant first by clicking 'Initialize AI'."
         
+        # Try template response first
         template_response = self.get_template_response(user_input)
         if template_response:
             self.memory.add_conversation(user_input, template_response)
             return template_response
         
-        general_response = self.get_comprehensive_advice(user_input)
-        self.memory.add_conversation(user_input, general_response)
-        return general_response
+        # If no template match, provide contextual advice
+        contextual_response = self.get_contextual_advice(user_input, resume_context)
+        self.memory.add_conversation(user_input, contextual_response)
+        return contextual_response
     
-    def get_comprehensive_advice(self, user_input: str) -> str:
-        """Provide comprehensive advice based on user input patterns"""
+    def get_contextual_advice(self, user_input: str, resume_context: str) -> str:
+        """Provide advice based on resume context and user input"""
         user_lower = user_input.lower()
         
-        if any(phrase in user_lower for phrase in ['improve', 'better', 'enhance', 'optimize']):
-            return """To improve your resume effectiveness: 1) Tailor it to each job by matching keywords from the job description. 2) Use quantifiable achievements (increased sales by 25%, managed team of 10). 3) Start bullet points with strong action verbs. 4) Keep it concise - ideally 1-2 pages. 5) Use a clean, professional format with consistent styling. 6) Include relevant technical and soft skills. 7) Proofread carefully for any errors."""
+        # Analyze resume context if available
+        if resume_context:
+            word_count = len(resume_context.split())
+            has_contact = '@' in resume_context or any(char.isdigit() for char in resume_context[:200])
+            has_bullets = any(marker in resume_context for marker in ['•', '*', '-', '→'])
+            
+            context_advice = []
+            
+            if word_count < 200:
+                context_advice.append("Your resume appears brief - consider expanding with more detailed achievements and responsibilities.")
+            elif word_count > 800:
+                context_advice.append("Your resume might be too lengthy - focus on the most relevant and impactful information.")
+            
+            if not has_contact:
+                context_advice.append("Ensure your contact information (email and phone) is clearly visible at the top.")
+            
+            if not has_bullets:
+                context_advice.append("Use bullet points to improve readability and make your achievements easier to scan.")
+            
+            if context_advice:
+                return "Based on your resume: " + " ".join(context_advice)
         
-        elif any(phrase in user_lower for phrase in ['hr', 'hire', 'hiring', 'recruiter', 'employer']):
-            return """To make your resume appealing to HR and hiring managers: 1) Use standard section headings they expect (Experience, Education, Skills). 2) Include relevant keywords to pass ATS screening. 3) Show clear career progression and achievements. 4) Make it easy to scan with bullet points and white space. 5) Demonstrate value you can bring to their organization. 6) Include measurable results and impacts."""
+        # Default comprehensive advice based on query type
+        if any(phrase in user_lower for phrase in ['improve', 'better', 'enhance', 'optimize']):
+            return """To improve your resume effectiveness: 1) Tailor content to match specific job descriptions. 2) Use quantifiable achievements with numbers and percentages. 3) Start bullet points with strong action verbs like 'Led', 'Implemented', 'Increased'. 4) Keep it concise but comprehensive (1-2 pages). 5) Use clean, professional formatting with consistent styling. 6) Include both technical and soft skills relevant to your field. 7) Proofread thoroughly for grammar and spelling errors."""
         
         elif any(phrase in user_lower for phrase in ['job', 'career', 'position', 'role', 'work']):
-            return """For job search success: 1) Customize your resume for each application. 2) Research the company and role requirements. 3) Highlight relevant experience and skills prominently. 4) Use industry-specific terminology. 5) Show how your background aligns with their needs. 6) Include both technical competencies and soft skills."""
+            return """For successful job applications: 1) Research each company and role thoroughly. 2) Customize your resume for every application. 3) Highlight experiences most relevant to the specific position. 4) Use industry-specific terminology and keywords. 5) Show clear alignment between your background and their requirements. 6) Demonstrate both technical competencies and interpersonal skills. 7) Follow up professionally after applications."""
+        
+        elif any(word in user_lower for word in ['why', 'what', 'how', 'when', 'where']):
+            return """Resume fundamentals: Focus on achievements over duties, use specific numbers when possible, maintain consistent formatting, tailor content for each application, include relevant keywords naturally, keep information current and accurate, and ensure easy readability with clear section headings."""
         
         else:
-            return """Key resume best practices: Use a professional format with clear headings. Lead with your strongest qualifications. Include relevant keywords naturally. Quantify achievements with specific numbers. Keep descriptions concise but impactful. Ensure error-free writing and consistent formatting. Focus on what value you bring to employers."""
+            return """Key resume principles: Use professional formatting with clear headings. Lead with your strongest qualifications. Include relevant keywords naturally throughout. Quantify achievements with specific metrics. Keep descriptions concise but impactful. Ensure error-free writing and consistent style. Focus on the value you can bring to potential employers."""
 
 @st.cache_resource
 def download_nltk_data():