gptpdf.py

import os
import constants
from langchain.vectorstores import FAISS
from langchain.embeddings.openai import OpenAIEmbeddings
from langchain.document_loaders import PyPDFLoader
from langchain.indexes import VectorstoreIndexCreator

os.environ['OPENAI_API_KEY'] = constants.APIKEY


loader = PyPDFLoader("databackup/mypdf.pdf")

index = VectorstoreIndexCreator().from_loaders([loader])

query = input(">> ")
resp = index.query(query)


print("------------------------------------")
print(resp)

gptpdf.py