Update app.py

app.py

@@ -1,50 +1,41 @@
 import streamlit as st
-from enhanced_text_humanizer import EnhancedTextHumanizer
-import time
-import nltk
-import os
-import nltk
-import os
-
 import nltk
 import os
+from enhanced_text_humanizer import EnhancedTextHumanizer
+import time
+# Set NLTK data path
+nltk.data.path.append('./nltk_data')
 
-
-
-
-import nltk
-import os
-from nltk.tokenize.punkt import PunktLanguageVars
-
-# Create and set NLTK data directory
-NLTK_DATA = os.path.join(os.getcwd(), 'nltk_data')
-os.makedirs(NLTK_DATA, exist_ok=True)
-nltk.data.path.append(NLTK_DATA)
-
-# Download essential resources
-resources = [
-    ('punkt', 'tokenizers/punkt'),
-    ('averaged_perceptron_tagger', 'taggers/averaged_perceptron_tagger'),
-    ('wordnet', 'corpora/wordnet'),
-]
-
-# Download each resource
-for resource, path in resources:
+# Download required NLTK resources at startup
+@st.cache_resource
+def download_nltk_resources():
     try:
-        nltk.data.find(path)
-    except LookupError:
-        nltk.download(resource, download_dir=NLTK_DATA)
-
-# Initialize PunktLanguageVars
-punkt_vars = PunktLanguageVars()
-
-
+        # Create NLTK data directory if it doesn't exist
+        os.makedirs('nltk_data', exist_ok=True)
+        
+        # Download required resources
+        nltk.download('punkt', download_dir='./nltk_data')
+        nltk.download('punkt_tab', download_dir='./nltk_data')
+        nltk.download('averaged_perceptron_tagger', download_dir='./nltk_data')
+        nltk.download('wordnet', download_dir='./nltk_data')
+        nltk.download('omw-1.4', download_dir='./nltk_data')
+        
+        return True
+    except Exception as e:
+        st.error(f"Error downloading NLTK resources: {e}")
+        return False
 
 def initialize_humanizer():
+    # First ensure NLTK resources are downloaded
+    resources_downloaded = download_nltk_resources()
+    
     with st.spinner('Loading language models... This may take a moment.'):
         humanizer = EnhancedTextHumanizer()
     return humanizer
 
+
+
+
 def main():
     st.set_page_config(
         page_title="Text Humanizer App",