Update app.py

app.py

@@ -6,23 +6,6 @@ import subprocess
 from booknlp.booknlp import BookNLP
 from spacy.cli import download
 
-# Ensure Spacy model is downloaded
-def ensure_spacy_model():
-    try:
-        spacy.load("en_core_web_sm")
-    except OSError:
-        download("en_core_web_sm")
-
-# Initialize Spacy model
-ensure_spacy_model()
-
-# Initialize BookNLP model
-model_params = {
-    "pipeline": "entity,quote,supersense,event,coref", 
-    "model": "big"
-}
-booknlp = BookNLP("en", model_params)
-
 # Define function to convert ebook to txt using Calibre
 def convert_to_txt(input_file):
     output_txt = os.path.splitext(input_file)[0] + ".txt"
@@ -36,6 +19,21 @@ def process_book(file):
     output_dir = "output_dir/booknlp_output/"
     book_id = os.path.splitext(os.path.basename(input_file))[0]
     
+    # Ensure Spacy model is downloaded
+    def ensure_spacy_model():
+        try:
+            spacy.load("en_core_web_sm")
+        except OSError:
+            download("en_core_web_sm")
+
+    # Initialize Spacy model and BookNLP
+    ensure_spacy_model()
+    model_params = {
+        "pipeline": "entity,quote,supersense,event,coref", 
+        "model": "big"
+    }
+    booknlp = BookNLP("en", model_params)
+    
     # Check if the file is already a .txt file
     if not input_file.endswith(".txt"):
         input_file = convert_to_txt(input_file)