Updated app.py

app.py

@@ -16,43 +16,43 @@ def register_vocabulary_in_main():
         logger.info("Successfully registered classes in __main__")
     except Exception as e:
         logger.warning(f"Could not register classes in __main__: {e}")
-    def setup_nltk():
-        """Set up NLTK data directory and ensure punkt tokenizer is available"""
-        logger.info("Setting up NLTK...")
-        
-        # Create potential NLTK data directories with proper permissions
-        nltk_dirs = [
-            os.path.expanduser('~/.nltk_data'),
-            './nltk_data',
-            '/usr/local/share/nltk_data'
-        ]
-        
+def setup_nltk():
+    """Set up NLTK data directory and ensure punkt tokenizer is available"""
+    logger.info("Setting up NLTK...")
+    
+    # Create potential NLTK data directories with proper permissions
+    nltk_dirs = [
+        os.path.expanduser('~/.nltk_data'),
+        './nltk_data',
+        '/usr/local/share/nltk_data'
+    ]
+    
+    for directory in nltk_dirs:
+        try:
+            os.makedirs(directory, exist_ok=True)
+            logger.info(f"Created NLTK data directory: {directory}")
+        except Exception as e:
+            logger.warning(f"Could not create NLTK directory {directory}: {e}")
+    
+    # Try to find punkt tokenizer
+    try:
+        nltk.data.find('tokenizers/punkt')
+        logger.info("NLTK punkt tokenizer found!")
+        return
+    except LookupError:
+        # Not found, try to download to different locations
         for directory in nltk_dirs:
             try:
-                os.makedirs(directory, exist_ok=True)
-                logger.info(f"Created NLTK data directory: {directory}")
+                logger.info(f"Attempting to download punkt tokenizer to {directory}")
+                nltk.download('punkt', download_dir=directory)
+                logger.info(f"Successfully downloaded punkt tokenizer to {directory}")
+                return
             except Exception as e:
-                logger.warning(f"Could not create NLTK directory {directory}: {e}")
+                logger.warning(f"Failed to download punkt to {directory}: {e}")
         
-        # Try to find punkt tokenizer
-        try:
-            nltk.data.find('tokenizers/punkt')
-            logger.info("NLTK punkt tokenizer found!")
-            return
-        except LookupError:
-            # Not found, try to download to different locations
-            for directory in nltk_dirs:
-                try:
-                    logger.info(f"Attempting to download punkt tokenizer to {directory}")
-                    nltk.download('punkt', download_dir=directory)
-                    logger.info(f"Successfully downloaded punkt tokenizer to {directory}")
-                    return
-                except Exception as e:
-                    logger.warning(f"Failed to download punkt to {directory}: {e}")
-            
-            # If we get here, we couldn't download punkt anywhere
-            logger.error("Could not download NLTK punkt tokenizer to any location")
-            logger.error("The application may not function correctly")
+        # If we get here, we couldn't download punkt anywhere
+        logger.error("Could not download NLTK punkt tokenizer to any location")
+        logger.error("The application may not function correctly")
         """
 Main application entry point for Image Captioning API
 """