Update app.py

app.py

@@ -59,6 +59,53 @@ def extract_text(uploaded_file):
     if not uploaded_file:
         return ""
     return extract_text_from_pdf(uploaded_file) if uploaded_file.type == "application/pdf" else extract_text_from_image(uploaded_file)
+def parse_resume(resume_text):
+    """Extract key information from resume text"""
+    parsed_info = {
+        'education': [],
+        'skills': [],
+        'experience': [],
+        'projects': [],
+        'publications': []
+    }
+    
+    # Find education details
+    edu_markers = ['Education:', 'EDUCATION', 'Academic Background']
+    exp_markers = ['Experience:', 'EXPERIENCE', 'Work History', 'Employment']
+    skill_markers = ['Skills:', 'SKILLS', 'Technical Skills', 'Technologies']
+    proj_markers = ['Projects:', 'PROJECTS', 'Key Projects']
+    pub_markers = ['Publications:', 'PUBLICATIONS', 'Research Papers']
+    
+    # Helper function to extract section content
+    def extract_section(text, start_markers, end_markers):
+        content = []
+        for start in start_markers:
+            start_idx = text.find(start)
+            if start_idx != -1:
+                section_start = start_idx + len(start)
+                section_end = len(text)
+                
+                # Find the next section marker
+                for end in end_markers:
+                    next_section = text.find(end, section_start)
+                    if next_section != -1:
+                        section_end = min(section_end, next_section)
+                
+                section_content = text[section_start:section_end].strip()
+                content.append(section_content)
+        
+        return '\n'.join(content)
+    
+    # Extract sections
+    all_markers = edu_markers + exp_markers + skill_markers + proj_markers + pub_markers
+    
+    parsed_info['education'] = extract_section(resume_text, edu_markers, all_markers)
+    parsed_info['experience'] = extract_section(resume_text, exp_markers, all_markers)
+    parsed_info['skills'] = extract_section(resume_text, skill_markers, all_markers)
+    parsed_info['projects'] = extract_section(resume_text, proj_markers, all_markers)
+    parsed_info['publications'] = extract_section(resume_text, pub_markers, all_markers)
+    
+    return parsed_info
 
 def extract_professor_details(text):
     professor_pattern = r"(Dr\.|Professor|Prof\.?)\s+([A-Z][a-z]+\s[A-Z][a-z]+)"
@@ -164,56 +211,98 @@ templates = {
     'email': PromptTemplate.from_template("""
 Write ONLY a formal cold email for a research position.
 Start with 'Dear Professor' and end with a signature.
-Include only 2-3 most relevant highlights from the CV.
-Do not include any other text or formatting.
 
-Details to use:
+Use these specific details from the CV:
+Education: {education}
+Relevant Experience: {experience}
+Key Skills: {skills}
+Notable Projects: {projects}
+Publications: {publications}
+
+Additional Context:
 Professor: {professor_name}
 University: {university_name}
 Research Interests: {research_interests}
 Why This Lab: {reason}
+
+Guidelines:
+1. Keep the email concise (max 400 words)
+2. Focus on the most relevant experience and skills that match the lab's research
+3. Mention 1-2 specific projects or publications that align with the lab's work
+4. Include a clear statement of interest and why you're a good fit
+5. End with your contact information
 """),
     
     'cover_letter': PromptTemplate.from_template("""
 Write ONLY a professional cover letter.
-Start with a formal greeting and end with a signature.
-Focus on relevant experience and skills.
-Do not include any headers or additional formatting.
+Use these specific details from the CV:
+Education: {education}
+Relevant Experience: {experience}
+Technical Skills: {skills}
+Notable Projects: {projects}
+Publications: {publications}
 
-Details to use:
+Position Details:
 Job Title: {job_title}
 Company: {company}
-Key Skills: {key_skills}
-Relevant Experience: {resume_text}
+Required Skills: {key_skills}
+
+Guidelines:
+1. Start with a formal greeting
+2. Focus on experiences and skills that directly match the job requirements
+3. Provide specific examples from your projects and work history
+4. Demonstrate how your background makes you an ideal candidate
+5. End with a professional closing
 """),
     
     'research_statement': PromptTemplate.from_template("""
-Write ONLY a research statement.
-Focus on research background, achievements, and future goals.
-Maintain a professional and academic tone.
-Do not include any headers or section titles.
+Write ONLY a research statement focused on your academic journey and research goals.
+Use these specific details from your background:
+Education: {education}
+Research Experience: {experience}
+Technical Skills: {skills}
+Research Projects: {projects}
+Publications: {publications}
 
-Content to include:
-Background: {research_background}
+Additional Context:
+Research Background: {research_background}
 Key Projects: {key_projects}
 Future Goals: {future_goals}
+
+Guidelines:
+1. Describe your research journey and motivation
+2. Highlight key research achievements and findings
+3. Connect past work to future research goals
+4. Demonstrate technical expertise and methodological knowledge
+5. End with your vision for future contributions to the field
 """),
     
     'sop': PromptTemplate.from_template("""
-Write ONLY a Statement of Purpose.
-Focus on academic journey, research interests, and career goals.
-Maintain a personal yet professional tone.
-Do not include any headers or section titles.
+Write ONLY a Statement of Purpose (SOP).
+Use these specific details from your background:
+Education: {education}
+Research Experience: {experience}
+Technical Skills: {skills}
+Notable Projects: {projects}
+Publications: {publications}
 
-Details to use:
+Additional Context:
 Motivation: {motivation}
-Academic Background: {academic_background}
-Research Experience: {research_experiences}
-Career Goals: {career_goals}
-Program Fit: {why_this_program}
+Academic Goals: {academic_background}
+Research Interests: {research_experiences}
+Career Objectives: {career_goals}
+Program Interest: {why_this_program}
+
+Guidelines:
+1. Tell a coherent story about your academic journey
+2. Connect your background to your future goals
+3. Demonstrate why you're prepared for graduate study
+4. Show alignment between your interests and the program
+5. Make a compelling case for why you should be admitted
 """)
 }
 
+
 # Create LangChain instances
 chains = {key: LLMChain(llm=llm, prompt=template) for key, template in templates.items()}
 
@@ -237,12 +326,20 @@ with tab1:
         if job_opening_text and cv_resume_text:
             with st.spinner("Generating..."):
                 try:
+                    # Parse resume information
+                    resume_info = parse_resume(cv_resume_text)
+                    
+                    # Generate email with parsed information
                     generated_email = chains['email'].run({
                         "professor_name": professor_name,
                         "university_name": university_name,
                         "research_interests": research_interests,
                         "reason": reason,
-                        "resume_text": cv_resume_text
+                        "education": resume_info['education'],
+                        "experience": resume_info['experience'],
+                        "skills": resume_info['skills'],
+                        "projects": resume_info['projects'],
+                        "publications": resume_info['publications']
                     })
                     st.session_state.generated_content['email'] = clean_email_output(generated_email)
                 except Exception as e:
@@ -250,13 +347,6 @@ with tab1:
         else:
             st.error("Please provide all required inputs")
 
-    if st.session_state.generated_content['email']:
-        st.markdown('<div class="output-container">', unsafe_allow_html=True)
-        st.markdown(st.session_state.generated_content['email'])
-        st.download_button("Download Email", st.session_state.generated_content['email'],
-                         file_name="cold_email.txt", key="email_download")
-        st.markdown('</div>', unsafe_allow_html=True)
-
 # Cover Letter Tab
 with tab2:
     job_title = st.text_input("Job Title")
@@ -267,11 +357,15 @@ with tab2:
         if job_opening_text and cv_resume_text:
             with st.spinner("Generating..."):
                 try:
+                    resume_info = parse_resume(cv_resume_text)
                     generated_letter = chains['cover_letter'].run({
                         "job_title": job_title,
                         "company": company_name,
                         "key_skills": key_skills,
-                        "resume_text": cv_resume_text
+                        "reason": reason,
+                        "skills": resume_info['skills'],
+                        "education": resume_info['education'],
+                        "experience": resume_info['experience']
                     })
                     st.session_state.generated_content['cover_letter'] = clean_cover_letter_output(generated_letter)
                 except Exception as e:
@@ -295,8 +389,15 @@ with tab3:
     if st.button("Generate Research Statement", key="research_stmt_btn"):
         with st.spinner("Generating..."):
             try:
+                resume_info = parse_resume(cv_resume_text)
                 generated_statement = chains['research_statement'].run({
-                    "research_background": research_background,
+                    "reason": reason,
+                    "education": resume_info['education'],
+                    "experience": resume_info['experience'],
+                    "skills": resume_info['skills'],
+                    "projects": resume_info['projects'],
+                    "publications": resume_info['publications']
+                    "research_background": resume_info['publications'],
                     "key_projects": key_projects,
                     "future_goals": future_goals
                 })
@@ -322,12 +423,17 @@ with tab4:
     if st.button("Generate SOP", key="sop_btn"):
         with st.spinner("Generating..."):
             try:
+                resume_info = parse_resume(cv_resume_text)
                 generated_sop = chains['sop'].run({
                     "motivation": motivation,
-                    "academic_background": academic_background,
-                    "research_experiences": research_experiences,
+                    "academic_background": resume_info['education'],
+                    "research_experiences": resume_info['publications'],
                     "career_goals": career_goals,
-                    "why_this_program": why_this_program
+                    "why_this_program": why_this_program,
+                    "experience": resume_info['experience'],
+                    "skills": resume_info['skills'],
+                    "projects": resume_info['projects']
+
                 })
                 st.session_state.generated_content['sop'] = clean_sop_output(generated_sop)
             except Exception as e: