Update app.py

app.py

@@ -2,13 +2,20 @@
 import os
 import openai
 import gradio as gr
+#
+#from datasets import load_dataset
+#token =  os.environ.get('HFToken')
+#dataset = load_dataset("rbughao/BT-DataSet",ht_token=token,token=True)
+#
 
 openai.api_key =  os.environ.get('O_APIKey')
+Data_Read =  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"])
@@ -16,7 +23,7 @@ loader = PDFReader()
 # or
 # documents = loader.load_data(file=['toolkit.pdf','pdf2','pdf3'])
 
-documents = loader.load_data(file='cms.pdf')
+documents = loader.load_data(file=ChurnData)
 index = VectorStoreIndex.from_documents(documents)
 query_engine = index.as_query_engine()