Update src/app.py

src/app.py

@@ -3,12 +3,14 @@ import requests
 import os
 import re
 import io
+import contextlib
 import zipfile
 import tracker
 import rag_engine
 import doc_loader 
 from openai import OpenAI
 from datetime import datetime
+from test_integration import run_tests
 
 # --- CONFIGURATION ---
 st.set_page_config(page_title="Navy AI Toolkit", page_icon="⚓", layout="wide")
@@ -213,6 +215,26 @@ with st.sidebar:
     if st.session_state.authenticator:
         st.session_state.authenticator.logout(location='sidebar')
 
+    st.divider()
+    st.subheader("🔧 System Diagnostics")
+    
+    if st.button("Run Integration Test"):
+        with st.spinner("Running diagnostics..."):
+            # Create a buffer to capture the text that would normally be printed
+            f = io.StringIO()
+            
+            # Redirect 'print' statements to our buffer instead of the console
+            try:
+                with contextlib.redirect_stdout(f):
+                    run_tests()
+                
+                # Display the result in a code block for easy reading
+                st.success("Tests Completed")
+                st.code(f.getvalue(), language="text")
+                
+            except Exception as e:
+                st.error(f"Test Execution Failed: {e}")
+
 update_sidebar_metrics()
 
 # --- MAIN APP ---