Update app.py

app.py

@@ -52,11 +52,11 @@ if uploaded_file:
     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: ### Output:
-        return full_text.split("### Output:")[-1].strip()
-    else:
-        return full_text.strip()
+    # Find the last occurrence of "### Output:" and return everything after it
+    if "### Output:" in full_text:
+        return full_text.split("### Output:", 1)[-1].strip()
+    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.