Update app.py

app.py

@@ -139,9 +139,13 @@ def PDFChunkerWithSeparator(filepath, separator):
     # print the number of pages in pdf file
     print(len(reader.pages))
     content = ""
-    for page in reader.pages:
-        content += page.extract_text()
-
+    if filepath.endswith(".pdf"):
+        for page in reader.pages:
+            content += page.extract_text()
+    elif filepath.endswith(".txt"):
+        with open(filepath) as f:
+            content = f.readlines()
+            f.close()
     splitted_content_list = content.split(separator)
 
     doclist = []
@@ -175,8 +179,9 @@ def loadKB(fileprovided, urlProvided, uploads_dir, request):
             separator = "</Q>"
             documents.extend(PDFChunkerWithSeparator(os.path.join(uploads_dir, secure_filename(file.filename)),separator))
     else:
-        loader = TextLoader('Jio.txt')
-        documents.extend(loader.load())
+        #loader = TextLoader('Jio.txt')
+        #documents.extend(loader.load())
+        documents.extend(PDFChunkerWithSeparator('JTest.txt',separator))
 
     if urlProvided:
         weburl = request.form.getlist('weburl')