Update app.py

app.py

@@ -53,7 +53,7 @@ if uploaded_file:
     
 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:
+    if "### Output:" in full_text: ### Output:
         return full_text.split("### Output:")[-1].strip()
     else:
         return full_text.strip()
@@ -162,7 +162,7 @@ with tab1:
     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)
+        st.text_area("Generated Cold Email", final_email, height=250)
 
 with tab2:
     job_title = st.text_input("Job Title")