Update app.py

app.py

@@ -125,9 +125,14 @@ def load_document(file: NamedTemporaryFile, parser: str = "pypdf") -> List[Docum
         loader = PyPDFLoader(file.name)
         return loader.load_and_split()
     elif parser == "llamaparse":
-        documents = llama_parser.load_data(file.name)
-        # Convert LlamaParse output to langchain Document format
-        return [Document(page_content=doc.text, metadata={"source": file.name}) for doc in documents]
+        try:
+            documents = llama_parser.load_data(file.name)
+            return [Document(page_content=doc.text, metadata={"source": file.name}) for doc in documents]
+        except Exception as e:
+            print(f"Error using Llama Parse: {str(e)}")
+            print("Falling back to PyPDF parser")
+            loader = PyPDFLoader(file.name)
+            return loader.load_and_split()
     else:
         raise ValueError("Invalid parser specified. Use 'pypdf' or 'llamaparse'.")