Update pages/4-Create your own bot (advanced).py

pages/4-Create your own bot (advanced).py

@@ -112,9 +112,17 @@ def extract_text_from_pdf(pdf_file):
     return text
 
 def generate_answer(question, context):
-    # Use the QA pipeline to get the answer
-    result = qa_pipeline(question=question, context=context)
-    return result['answer']
+    # Add context to the question
+    context_instruction = (
+        f"Your professional experience is outlined in the following resume. "
+        "Answer the question as if you are the candidate, providing details from the resume where relevant."
+    )
+
+    # Combine the question with the context instruction
+    full_question = f"{context_instruction} Question: {question}"
+
+    # Get the answer using DistilBERT pipeline
+    result = qa_pipeline(question=full_question, context=resume_text)
 
 # Streamlit app layout
 st.title("Resume-based Q&A Bot")