Update app.py

app.py

@@ -50,7 +50,13 @@ if uploaded_file:
     
     st.subheader("Extracted Text from CV/Resume")
     st.text_area("Preview:", extracted_text, height=150)
-
+    
+def get_final_output(full_text):
+    # If the model returns text with the prompt details followed by "### Output:", only keep what comes after it.
+    if "### Output:" in full_text:
+        return full_text.split("### Output:")[-1].strip()
+    else:
+        return full_text.strip()
 # Define LLM Prompt Templates
 email_template = PromptTemplate.from_template("""
 You are an AI assistant skilled in crafting personalized and engaging cold emails for research positions.
@@ -155,6 +161,7 @@ with tab1:
     reason = st.text_area("Why this professor/lab?")
     if st.button("Generate Cold Email"):
         email = email_chain.run({"recipient_name": recipient, "position_name": position, "research_interests": research_interests, "reason": reason, "resume_text": extracted_text})
+        final_email = get_final_output(email)
         st.text_area("Generated Cold Email", email, height=250)
 
 with tab2:
@@ -163,13 +170,15 @@ with tab2:
     key_skills = st.text_area("Key Skills")
     if st.button("Generate Cover Letter"):
         cover_letter = cover_letter_chain.run({"job_title": job_title, "company": company, "key_skills": key_skills, "resume_text": extracted_text})
-        st.text_area("Generated Cover Letter", cover_letter, height=250)
+        final_cover_letter = get_final_output(cover_letter)
+        st.text_area("Generated Cover Letter", final_cover_letter, height=250)
 
 with tab3:
     research_goals = st.text_area("Future Research Goals")
     if st.button("Generate Research Statement"):
         research_statement = research_statement_chain.run({"research_interests": research_interests, "goals": research_goals, "resume_text": extracted_text})
-        st.text_area("Generated Research Statement", research_statement, height=250)
+        final_rs = get_final_output(research_statement)
+        st.text_area("Generated Research Statement", final_rs, height=250)
 
 with tab4:
     program_name = st.text_input("Program Name")
@@ -177,4 +186,5 @@ with tab4:
     career_goals = st.text_area("Career Goals")
     if st.button("Generate SOP"):
         sop = sop_chain.run({"program_name": program_name, "university": university, "research_interests": research_interests, "career_goals": career_goals, "resume_text": extracted_text})
-        st.text_area("Generated SOP", sop, height=250)
+        final_sop = get_final_output(sop)
+        st.text_area("Generated SOP", final_sop, height=250)