Update PDF_Reader.py

PDF_Reader.py

@@ -1,17 +1,28 @@
 import os
 from together import Together
+import PyPDF2
 
 client = Together(api_key=os.environ.get("TOGETHER_API_KEY"))
 
-def process_text(question, text):
+def process_text(question, pdf_paths):
+    texts = [get_pdf_content(pdf) for pdf in pdf_paths]
+    combined_text = "\n".join(texts)
     response = client.chat.completions.create(
         model="meta-llama/Llama-3-8b-chat-hf",
         messages=[{"role": "user", "content": question}],
-        context=text
+        context=combined_text
     )
     return response.choices[0].message.content
 
+def get_pdf_content(pdf_path):
+    with open(pdf_path, "rb") as file:
+        reader = PyPDF2.PdfReader(file)
+        text = ""
+        for page in reader.pages:
+            text += page.extract_text()
+    return text
+
 if __name__ == "__main__":
-    text = "New York is known for its vibrant cultural scene, museums, and theaters."
-    question = "What can I learn about New York from this text?"
-    print(process_text(question, text))
+    pdf_paths = ["doc1.pdf", "doc2.pdf"]  # Adjust paths as necessary
+    question = "What can I learn about New York from these texts?"
+    print(process_text(question, pdf_paths))