demoPOC commited on
Commit
4b74957
·
1 Parent(s): 51b1ee3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -7
app.py CHANGED
@@ -10,7 +10,7 @@ from langchain.text_splitter import RecursiveCharacterTextSplitter
10
  from langchain.llms import OpenAI
11
  from langchain.chains import RetrievalQA
12
  from langchain.chains import VectorDBQA
13
- from langchain.document_loaders import TextLoader, WebBaseLoader
14
  from langchain.document_loaders import UnstructuredFileLoader
15
  from flask import Flask, jsonify, render_template, request
16
  from werkzeug.utils import secure_filename
@@ -88,13 +88,9 @@ def file_Upload():
88
  file.save(os.path.join(uploads_dir, secure_filename(file.filename)))
89
  loader = UnstructuredFileLoader(os.path.join(uploads_dir, secure_filename(file.filename)), mode='elements')
90
  documents.extend(loader.load())
91
-
92
- print(request.form.getlist('weburl'))
93
- for url in request.form.getlist('weburl'):
94
- print(type(url))
95
- urlLoader=WebBaseLoader(url)
96
- documents.extend(urlLoader.load())
97
 
 
 
98
 
99
  print(uploads_dir)
100
  global chain;
 
10
  from langchain.llms import OpenAI
11
  from langchain.chains import RetrievalQA
12
  from langchain.chains import VectorDBQA
13
+ from langchain.document_loaders import TextLoader, WebBaseLoader, SeleniumURLLoader
14
  from langchain.document_loaders import UnstructuredFileLoader
15
  from flask import Flask, jsonify, render_template, request
16
  from werkzeug.utils import secure_filename
 
88
  file.save(os.path.join(uploads_dir, secure_filename(file.filename)))
89
  loader = UnstructuredFileLoader(os.path.join(uploads_dir, secure_filename(file.filename)), mode='elements')
90
  documents.extend(loader.load())
 
 
 
 
 
 
91
 
92
+ urlLoader=SeleniumURLLoader(request.form.getlist('weburl'))
93
+ documents.extend(urlLoader.load())
94
 
95
  print(uploads_dir)
96
  global chain;