Update app.py

app.py

@@ -18,7 +18,7 @@ llm = HuggingFaceHub(repo_id="mistralai/Mistral-7B-Instruct-v0.3", model_kwargs=
 st.set_page_config(page_title="DocuMentorAI", layout="wide", page_icon="📄")
 st.title("📄 DocuMentorAI")
 
-# Improved CSS with dedicated output styling
+# Improved CSS
 st.markdown("""
 <style>
     .output-container {
@@ -26,14 +26,13 @@ st.markdown("""
         padding: 20px;
         border-radius: 10px;
         margin-top: 20px;
-    }
-    .generated-content {
-        font-size: 16px;
-        line-height: 1.6;
         white-space: pre-wrap;
     }
-    .download-button {
-        margin-top: 10px;
+    .stTextArea textarea {
+        font-size: 16px !important;
+    }
+    .stButton button {
+        width: 100%;
     }
 </style>
 """, unsafe_allow_html=True)
@@ -63,7 +62,7 @@ def extract_text(uploaded_file):
 
 def extract_professor_details(text):
     professor_pattern = r"(Dr\.|Professor|Prof\.?)\s+([A-Z][a-z]+\s[A-Z][a-z]+)"
-    university_pattern = r"(University|Institute|College|School of [A-Za-z]+)"
+    university_pattern = r"(University|Institute|College|School of [A-Z][A-Za-z\s]+)"
     
     professor_match = re.search(professor_pattern, text)
     university_match = re.search(university_pattern, text)
@@ -71,14 +70,87 @@ def extract_professor_details(text):
     return (professor_match.group(0) if professor_match else "Not Found",
             university_match.group(0) if university_match else "Not Found")
 
-# Sidebar for Input Collection
-with st.sidebar:
-    st.subheader("📝 Input Details")
-    job_opening_text = st.text_area("Job/Research Opening Details", height=150)
-    cv_resume_file = st.file_uploader("Upload CV/Resume", type=["pdf", "png", "jpg", "jpeg"])
-    cv_resume_text = extract_text(cv_resume_file)
+def clean_email_output(email_text):
+    """Clean and format email content"""
+    start_idx = email_text.find("Dear")
+    if start_idx == -1:
+        start_idx = 0
+        
+    end_markers = ["Best regards,", "Sincerely,", "Yours sincerely,", "Kind regards,"]
+    end_idx = len(email_text)
+    for marker in end_markers:
+        idx = email_text.find(marker)
+        if idx != -1:
+            end_idx = email_text.find("\n\n", idx) if email_text.find("\n\n", idx) != -1 else len(email_text)
+            break
+    
+    email_content = email_text[start_idx:end_idx].strip()
+    
+    if "Phone:" in email_text or "Email:" in email_text:
+        contact_info = "\n\n" + "\n".join([
+            line for line in email_text[end_idx:].split("\n")
+            if any(info in line for info in ["Phone:", "Email:"])
+        ]).strip()
+        email_content += contact_info
+    
+    return email_content
+
+def clean_cover_letter_output(letter_text):
+    """Clean and format cover letter content"""
+    start_markers = ["Dear", "To Whom", "Hiring"]
+    start_idx = len(letter_text)
+    for marker in start_markers:
+        idx = letter_text.find(marker)
+        if idx != -1:
+            start_idx = min(start_idx, idx)
+    
+    end_markers = ["Sincerely,", "Best regards,", "Yours truly,", "Regards,"]
+    end_idx = len(letter_text)
+    for marker in end_markers:
+        idx = letter_text.find(marker)
+        if idx != -1:
+            end_idx = letter_text.find("\n\n", idx) if letter_text.find("\n\n", idx) != -1 else len(letter_text)
+            break
+    
+    return letter_text[start_idx:end_idx].strip()
+
+def clean_research_statement_output(statement_text):
+    """Clean and format research statement content"""
+    # Remove common headers
+    headers = ["Research Statement", "Statement of Research", "Research Interests"]
+    cleaned_text = statement_text
+    for header in headers:
+        if cleaned_text.startswith(header):
+            cleaned_text = cleaned_text[len(header):].lstrip(":\n")
+    
+    # Remove any trailing references or bibliography sections
+    end_markers = ["References", "Bibliography", "Citations"]
+    for marker in end_markers:
+        idx = cleaned_text.find(marker)
+        if idx != -1:
+            cleaned_text = cleaned_text[:idx].strip()
+    
+    return cleaned_text.strip()
 
-# Initialize session state for generated content
+def clean_sop_output(sop_text):
+    """Clean and format Statement of Purpose content"""
+    # Remove common headers
+    headers = ["Statement of Purpose", "Personal Statement", "Academic Statement"]
+    cleaned_text = sop_text
+    for header in headers:
+        if cleaned_text.startswith(header):
+            cleaned_text = cleaned_text[len(header):].lstrip(":\n")
+    
+    # Remove any trailing sections
+    end_markers = ["Thank you", "References", "Additional Information"]
+    for marker in end_markers:
+        idx = cleaned_text.find(marker)
+        if idx != -1:
+            cleaned_text = cleaned_text[:idx].strip()
+    
+    return cleaned_text.strip()
+
+# Initialize session state
 if 'generated_content' not in st.session_state:
     st.session_state.generated_content = {
         'email': None,
@@ -90,43 +162,68 @@ if 'generated_content' not in st.session_state:
 # Template Definitions
 templates = {
     'email': PromptTemplate.from_template("""
-Write a professional cold email for a research position.
-Output only the email content without any additional text or formatting.
+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:
 Professor: {professor_name}
 University: {university_name}
 Research Interests: {research_interests}
 Why This Lab: {reason}
-CV Highlights: {resume_text}
 """),
+    
     'cover_letter': PromptTemplate.from_template("""
-Write a compelling job application cover letter.
-Output only the letter content without any additional text or formatting.
+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.
+
+Details to use:
 Job Title: {job_title}
 Company: {company}
 Key Skills: {key_skills}
-CV Highlights: {resume_text}
+Relevant Experience: {resume_text}
 """),
+    
     'research_statement': PromptTemplate.from_template("""
-Write a research statement for Ph.D. applications.
-Output only the statement content without any additional text or formatting.
-Research Background: {research_background}
-Key Research Projects: {key_projects}
+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.
+
+Content to include:
+Background: {research_background}
+Key Projects: {key_projects}
 Future Goals: {future_goals}
 """),
+    
     'sop': PromptTemplate.from_template("""
-Write a compelling Statement of Purpose (SOP).
-Output only the SOP content without any additional text or formatting.
+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.
+
+Details to use:
 Motivation: {motivation}
 Academic Background: {academic_background}
-Research & Projects: {research_experiences}
+Research Experience: {research_experiences}
 Career Goals: {career_goals}
-Why This Program: {why_this_program}
+Program Fit: {why_this_program}
 """)
 }
 
 # Create LangChain instances
 chains = {key: LLMChain(llm=llm, prompt=template) for key, template in templates.items()}
 
+# Sidebar for Input Collection
+with st.sidebar:
+    st.subheader("📝 Input Details")
+    job_opening_text = st.text_area("Job/Research Opening Details", height=150)
+    cv_resume_file = st.file_uploader("Upload CV/Resume", type=["pdf", "png", "jpg", "jpeg"])
+    cv_resume_text = extract_text(cv_resume_file)
+
 # Tab Layout
 tab1, tab2, tab3, tab4 = st.tabs(["Cold Email", "Cover Letter", "Research Statement", "SOP"])
 
@@ -140,13 +237,14 @@ with tab1:
         if job_opening_text and cv_resume_text:
             with st.spinner("Generating..."):
                 try:
-                    st.session_state.generated_content['email'] = chains['email'].run({
+                    generated_email = chains['email'].run({
                         "professor_name": professor_name,
                         "university_name": university_name,
                         "research_interests": research_interests,
                         "reason": reason,
                         "resume_text": cv_resume_text
                     })
+                    st.session_state.generated_content['email'] = clean_email_output(generated_email)
                 except Exception as e:
                     st.error(f"Generation error: {e}")
         else:
@@ -155,7 +253,7 @@ with tab1:
     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'], 
+        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)
 
@@ -169,12 +267,13 @@ with tab2:
         if job_opening_text and cv_resume_text:
             with st.spinner("Generating..."):
                 try:
-                    st.session_state.generated_content['cover_letter'] = chains['cover_letter'].run({
+                    generated_letter = chains['cover_letter'].run({
                         "job_title": job_title,
                         "company": company_name,
                         "key_skills": key_skills,
                         "resume_text": cv_resume_text
                     })
+                    st.session_state.generated_content['cover_letter'] = clean_cover_letter_output(generated_letter)
                 except Exception as e:
                     st.error(f"Generation error: {e}")
         else:
@@ -196,11 +295,12 @@ with tab3:
     if st.button("Generate Research Statement", key="research_stmt_btn"):
         with st.spinner("Generating..."):
             try:
-                st.session_state.generated_content['research_statement'] = chains['research_statement'].run({
+                generated_statement = chains['research_statement'].run({
                     "research_background": research_background,
                     "key_projects": key_projects,
                     "future_goals": future_goals
                 })
+                st.session_state.generated_content['research_statement'] = clean_research_statement_output(generated_statement)
             except Exception as e:
                 st.error(f"Generation error: {e}")
 
@@ -222,13 +322,14 @@ with tab4:
     if st.button("Generate SOP", key="sop_btn"):
         with st.spinner("Generating..."):
             try:
-                st.session_state.generated_content['sop'] = chains['sop'].run({
+                generated_sop = chains['sop'].run({
                     "motivation": motivation,
                     "academic_background": academic_background,
                     "research_experiences": research_experiences,
                     "career_goals": career_goals,
                     "why_this_program": why_this_program
                 })
+                st.session_state.generated_content['sop'] = clean_sop_output(generated_sop)
             except Exception as e:
                 st.error(f"Generation error: {e}")