Update app.py

app.py

@@ -188,14 +188,19 @@ def main():
             st.error(f"Error loading cases: {str(e)}")
 
     # Tab 2: Document Analysis
+   # In the Document Analysis tab section
     elif tab == "📄 Document Analysis":
         st.title("📄 Document Analysis")
         try:
-            notebook = LegalNotebookInterface(case_manager, vector_store, doc_processor)
-            notebook.render()
+            cases = case_manager.get_all_cases()
+            if not cases:
+                st.info("Please create a case and add documents first.")
+            else:
+                notebook = LegalNotebookInterface(case_manager, vector_store, doc_processor)
+                notebook.render()
         except Exception as e:
             st.error(f"Error initializing document analysis: {str(e)}")
-
+        
     # Tab 3: Legal Assistant
     elif tab == "🤖 Legal Assistant":
         st.title("🤖 AI Legal Assistant")