Update pages/Example.py

pages/Example.py

@@ -2,26 +2,40 @@ import streamlit as st
 from utils.backend import run_llm
 
 st.title("🧪 Example Page with Chatbot")
-st.write("This is a demo page showing how any `/pages/*.py` file becomes a sidebar entry.")
+st.write("This demo chatbot also ingests Omnieye + Omnilog outputs.")
 
 # Initialize session state
 if "example_chat" not in st.session_state:
     st.session_state.example_chat = []
 
+# Collect context from Omnieye + Omnilog
+system_context = []
+if "omnieye_output" in st.session_state:
+    system_context.append(f"Omnieye preview:\n{st.session_state.omnieye_output.get('file_preview', '')}")
+    if st.session_state.omnieye_output.get("matches"):
+        system_context.append("Keyword matches:\n" + "\n".join(st.session_state.omnieye_output["matches"]))
+if "omnilog_output" in st.session_state:
+    system_context.append(f"Omnilog preview:\n{st.session_state.omnilog_output.get('normalized_preview', '')}")
+    if st.session_state.omnilog_output.get("matches"):
+        system_context.append("Log matches:\n" + "\n".join(st.session_state.omnilog_output["matches"]))
+
 # Display chat history
 for msg in st.session_state.example_chat:
     with st.chat_message(msg["role"]):
         st.markdown(msg["content"])
 
 # Chat input
-if prompt := st.chat_input("Say something to the Example Chatbot..."):
+if prompt := st.chat_input("Ask the Example Chatbot about files or logs..."):
     # Add user message
     st.session_state.example_chat.append({"role": "user", "content": prompt})
     with st.chat_message("user"):
         st.markdown(prompt)
 
+    # Build context-aware input
+    ai_input = "\n\n".join(system_context + [prompt])
+
     # Generate AI response
     with st.chat_message("assistant"):
-        ai_reply = run_llm(prompt)
+        ai_reply = run_llm(ai_input)
         st.markdown(ai_reply)
         st.session_state.example_chat.append({"role": "assistant", "content": ai_reply})