Update app.py

app.py

@@ -498,9 +498,22 @@ def process_answer(question, full_text, retriever):
         summary = llm(prompt)  # Uses the LLM to generate a summary
         return summary
 
+    # --- Prompt Engineering ---
+    # Let's modify how we ask the model to answer
+    prompt = f"""
+    Given the following text, answer the question with a simple and direct 'Yes' or 'No' followed by a brief explanation.
+
+    Text: {full_text[:3000]}
+    
+    Question: {question}
+    Answer:
+    """
+
     # Use RetrievalQA for general queries
     qa_chain = RetrievalQA.from_chain_type(llm=llm, retriever=retriever)
-    return qa_chain.run(question)  # This is the main answer generation with retrieval
+    response = qa_chain.run(question)
+
+    return response
 
 # --- UI Layout ---
 with st.sidebar:
@@ -545,4 +558,3 @@ if uploaded_file:
         st.error("⚠️ No text could be extracted from the PDF. Try another file.")
 else:
     st.info("Upload a PDF to begin.")
-