Update app.py

app.py

@@ -105,21 +105,24 @@ def summarize(pdf_file, custom_prompt, custom_chunk, chunk_size, chunk_overlap):
         # Set the token for hub_llm within the function
         hub_llm.model_kwargs["huggingfacehub_api_token"] = huggingfacehub_api_token
     
-    try:
-        loader = PyPDFLoader(pdf_file.name)
-        if custom_chunk:
-            text_splitter = RecursiveCharacterTextSplitter(chunk_size=chunk_size, chunk_overlap=chunk_overlap)
-            docs = loader.load_and_split(text_splitter=text_splitter)
-        else:
-            docs = loader.load_and_split()
-
-        PROMPT = PromptTemplate(template=custom_prompt, input_variables=['text'])
-        chain = load_summarize_chain(hub_llm, chain_type='map_reduce', combine_prompt=PROMPT)
-
-        # Introduce a delay before calling the API
-        time.sleep(1)
-        summary = chain.run(docs)
-        return summary
+        try:
+            loader = PyPDFLoader(pdf_file.name)
+            if custom_chunk:
+                text_splitter = RecursiveCharacterTextSplitter(chunk_size=chunk_size, chunk_overlap=chunk_overlap)
+                docs = loader.load_and_split(text_splitter=text_splitter)
+            else:
+                docs = loader.load_and_split()
+
+            PROMPT = PromptTemplate(template=custom_prompt, input_variables=['text'])
+            chain = load_summarize_chain(hub_llm, chain_type='map_reduce', combine_prompt=PROMPT)
+
+            # Introduce a delay before calling the API
+            time.sleep(1)
+            summary = chain.run(docs)
+            return summary
+        except Exception as e:
+            return f"An error occurred: {e}"
+    
     except Exception as e:
         return f"An error occurred: {e}"