Update QnA.py

QnA.py

@@ -9,19 +9,19 @@ def get_pdf_content(pdf_path):
         reader = PyPDF2.PdfReader(file)
         text = ""
         for page in reader.pages:
-            text += page.extract_text()
+            text += page.extract_text() if page.extract_text() else ""
     return text
 
 def get_answer(question, pdf_path):
     pdf_content = get_pdf_content(pdf_path)
+    full_question = f"{question}\n\n{pdf_content}"  # Append PDF content to the question
     response = client.chat.completions.create(
         model="meta-llama/Llama-3-8b-chat-hf",
-        messages=[{"role": "user", "content": question}],
-        context=pdf_content
+        messages=[{"role": "user", "content": full_question}],
     )
     return response.choices[0].message.content
 
 if __name__ == "__main__":
     pdf_path = "sample.pdf"  # Adjust path as necessary
     question = "What are some fun things to do in New York?"
-    print(get_answer(question, pdf_path))
+    print(get_answer(question, pdf_path))