Update app.py

app.py

@@ -43,7 +43,6 @@ def process_pdf_to_epub(pdf_file, title, author):
             anchor_text = get_anchor_text(pdf_path, page_num, pdf_engine="pdfreport", target_length=4000)
             prompt = build_finetuning_prompt(anchor_text)
 
-            # Format prompt
             messages = [
                 {
                     "role": "user",
@@ -74,11 +73,14 @@ def process_pdf_to_epub(pdf_file, title, author):
             prompt_length = inputs["input_ids"].shape[1]
             new_tokens = output[:, prompt_length:]
 
-        try:
-            decoded_list = processor.tokenizer.batch_decode(new_tokens, skip_special_tokens=True)
-            decoded = decoded_list[0].strip() if decoded_list else "[No output generated]"
-        except Exception as decode_error:
-            decoded = f"[Decoding error on page {page_num}: {str(decode_error)}]"
+        except Exception as processing_error:
+            decoded = f"[Processing error on page {page_num}: {str(processing_error)}]"
+        else:
+            try:
+                decoded_list = processor.tokenizer.batch_decode(new_tokens, skip_special_tokens=True)
+                decoded = decoded_list[0].strip() if decoded_list else "[No output generated]"
+            except Exception as decode_error:
+                decoded = f"[Decoding error on page {page_num}: {str(decode_error)}]"
 
         # Create chapter
         chapter = epub.EpubHtml(title=f"Page {page_num}", file_name=f"page_{page_num}.xhtml", lang="en")