Update app.py

app.py

@@ -8,13 +8,14 @@ import gradio as gr
 #dataset = load_dataset("rbughao/BT-DataSet",ht_token=token,token=True)
 #
 
-
 openai.api_key =  os.environ.get('O_APIKey')
+DataRead =  os.environ.get('Data_Reader')
+ChurnData =  os.environ.get('Churn_Data')
 
-#using a folder
+#read data
 from llama_index import VectorStoreIndex, SimpleDirectoryReader, SummaryIndex, download_loader
-PDFReader = download_loader("PDFReader")
-loader = PDFReader()
+DataReader = download_loader(Data_Read)
+loader = DataReader()
 
 #loading options
 # loader = SimpleDirectoryReader(input_files = ["pdf1","pdf2"])
@@ -22,7 +23,7 @@ loader = PDFReader()
 # or
 # documents = loader.load_data(file=['toolkit.pdf','pdf2','pdf3'])
 
-documents = loader.load_data(file='toolkit.pdf')
+documents = loader.load_data(file=ChurnData)
 index = VectorStoreIndex.from_documents(documents)
 query_engine = index.as_query_engine()