wholewhale commited on
Commit
30a372e
1 Parent(s): e9619d2

error handling

Browse files
Files changed (1) hide show
  1. app.py +20 -12
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
- loader = OnlinePDFLoader(pdf_doc.name)
22
- documents = loader.load()
23
- text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=100)
24
- texts = text_splitter.split_documents(documents)
25
- embeddings = OpenAIEmbeddings()
26
- global db
27
- db = Chroma.from_documents(texts, embeddings)
28
- retriever = db.as_retriever()
29
- global qa
30
- qa = ConversationalRetrievalChain.from_llm(
31
- llm=OpenAI(temperature=0.2, model_name="gpt-3.5-turbo", max_tokens=-1, n=2),
 
 
 
 
 
 
32
  retriever=retriever,
33
  return_source_documents=False)
34
- return "Ready"
 
 
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