Update app.py

app.py

@@ -20,7 +20,7 @@ st.set_page_config(
 )
 
 def init_huggingface_directories():
-    """Initialize directory structure for Hugging Face Spaces silently."""
+    """Initialize directory structure for Hugging Face Spaces."""
     if os.environ.get('SPACE_ID'):
         base_dir = Path("/data")
     else:
@@ -32,39 +32,42 @@ def init_huggingface_directories():
         'vectors': base_dir / "vectors",
         'nltk_data': base_dir / "nltk_data",
         'temp': base_dir / "temp",
-        'logs': base_dir / "logs"
+        'logs': base_dir / "logs",
+        'prompts': base_dir / "prompts",
+        'ontology': base_dir / "ontology"
     }
     
-    for dir_path in directories.values():
+    # Create directories
+    for dir_name, dir_path in directories.items():
         dir_path.mkdir(parents=True, exist_ok=True)
+        st.sidebar.write(f"✓ {dir_name} directory ready")
         
     return directories
 
-@st.cache_resource(show_spinner=False)
-def initialize_nltk(nltk_data_dir):
-    """Initialize NLTK resources silently."""
-    nltk.data.path.append(str(nltk_data_dir))
-    resources = ['punkt', 'averaged_perceptron_tagger', 'maxent_ne_chunker', 'words', 'stopwords']
-    for resource in resources:
-        try:
-            nltk.download(resource, download_dir=str(nltk_data_dir), quiet=True)
-        except Exception:
-            pass
-    return True
-
 @st.cache_resource(show_spinner=False)
 def init_components():
     """Initialize all components silently."""
     try:
+        # Set up directories
         directories = init_huggingface_directories()
-        initialize_nltk(directories['nltk_data'])
 
+        # Initialize NLTK
+        nltk.data.path.append(str(directories['nltk_data']))
+        resources = ['punkt', 'averaged_perceptron_tagger', 'maxent_ne_chunker', 'words', 'stopwords']
+        for resource in resources:
+            try:
+                nltk.download(resource, download_dir=str(directories['nltk_data']), quiet=True)
+            except Exception:
+                pass
+
+        # Initialize spaCy
         try:
             nlp = spacy.load("en_core_web_sm")
         except OSError:
             os.system("python -m spacy download en_core_web_sm > /dev/null 2>&1")
             nlp = spacy.load("en_core_web_sm")
 
+        # Initialize components with proper paths
         case_manager = CaseManager(base_path=str(directories['cases']))
         vector_store = VectorStore(storage_path=str(directories['vectors']))
         doc_processor = DocumentProcessor(base_path=str(directories['data']))
@@ -72,7 +75,7 @@ def init_components():
         return case_manager, vector_store, doc_processor
 
     except Exception as e:
-        st.error("Failed to initialize components")
+        st.error(f"Failed to initialize components: {str(e)}")
         raise
 
 def main():
@@ -113,10 +116,6 @@ def main():
             </style>
             """, unsafe_allow_html=True)
 
-        # Header
-        st.markdown("<h1 class='main-header'>SuoMoto.AI</h1>", unsafe_allow_html=True)
-        st.markdown("<p class='main-tagline'>Empowering Legal Intelligence: Automate, Analyze, Act.</p>", unsafe_allow_html=True)
-
         # Initialize components
         case_manager, vector_store, doc_processor = init_components()
 
@@ -124,6 +123,10 @@ def main():
         if 'user_authenticated' not in st.session_state:
             st.session_state.user_authenticated = False
             st.session_state.current_user = None
+        
+        # Header
+        st.markdown("<h1 class='main-header'>SuoMoto.AI</h1>", unsafe_allow_html=True)
+        st.markdown("<p class='main-tagline'>Empowering Legal Intelligence: Automate, Analyze, Act.</p>", unsafe_allow_html=True)
 
         # Sidebar navigation with admin access
         if st.session_state.user_authenticated: