Commit
·
30a372e
1
Parent(s):
e9619d2
error handling
Browse files
app.py
CHANGED
@@ -18,20 +18,28 @@ def loading_pdf():
|
|
18 |
return "Working on the upload. Also, pondering the usefulness of sporks..."
|
19 |
|
20 |
def pdf_changes(pdf_doc):
|
21 |
-
|
22 |
-
|
23 |
-
|
24 |
-
|
25 |
-
|
26 |
-
|
27 |
-
|
28 |
-
|
29 |
-
|
30 |
-
|
31 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
32 |
retriever=retriever,
|
33 |
return_source_documents=False)
|
34 |
-
|
|
|
|
|
35 |
|
36 |
def clear_data():
|
37 |
global qa, db
|
|
|
18 |
return "Working on the upload. Also, pondering the usefulness of sporks..."
|
19 |
|
20 |
def pdf_changes(pdf_doc):
|
21 |
+
try:
|
22 |
+
if pdf_doc is None:
|
23 |
+
return "No PDF uploaded."
|
24 |
+
|
25 |
+
loader = OnlinePDFLoader(pdf_doc.name)
|
26 |
+
documents = loader.load()
|
27 |
+
loader = OnlinePDFLoader(pdf_doc.name)
|
28 |
+
documents = loader.load()
|
29 |
+
text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=100)
|
30 |
+
texts = text_splitter.split_documents(documents)
|
31 |
+
embeddings = OpenAIEmbeddings()
|
32 |
+
global db
|
33 |
+
db = Chroma.from_documents(texts, embeddings)
|
34 |
+
retriever = db.as_retriever()
|
35 |
+
global qa
|
36 |
+
qa = ConversationalRetrievalChain.from_llm(
|
37 |
+
llm=OpenAI(temperature=0.2, model_name="gpt-3.5-turbo-16k", max_tokens=-1, n=2),
|
38 |
retriever=retriever,
|
39 |
return_source_documents=False)
|
40 |
+
return "Ready"
|
41 |
+
except Exception as e:
|
42 |
+
return f"Error loading PDF: {e}"
|
43 |
|
44 |
def clear_data():
|
45 |
global qa, db
|