Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,26 +1,21 @@
 import os
-os.environ["STREAMLIT_BROWSER_GATHER_USAGE_STATS"] = "false"
-
-import streamlit as st
-import spacy
-import benepar
-from nltk import Tree
+import subprocess
 import nltk
+import spacy
 
-# Setup NLTK and benepar
-
-    
+# Setup NLTK
 nltk_data_path = "/tmp/nltk_data"
 nltk.download('punkt', download_dir=nltk_data_path)
+nltk.download('benepar_en3', download_dir=nltk_data_path)
 nltk.data.path.append(nltk_data_path)
 
-benepar.download('benepar_en3')
-
+# Load spaCy model with fallback
 try:
     nlp = spacy.load("en_core_web_sm")
 except:
     subprocess.run(["python", "-m", "spacy", "download", "en_core_web_sm"])
     nlp = spacy.load("en_core_web_sm")
+    
 if "benepar" not in nlp.pipe_names:
     nlp.add_pipe("benepar", config={"model": "benepar_en3"})