Update pages/Omnieye.py

pages/Omnieye.py

@@ -1,18 +1,49 @@
 import streamlit as st
 import re
+from utils.file_utils import keyword_search
+from utils.summarizer import summarize_text
 
 st.title("👁️ Omnieye")
-st.write("Keyword/file analyzer module")
+st.write("Keyword/file analyzer module with AI summarization")
+
+# Initialize counters if not set
+if "uploaded_files" not in st.session_state:
+    st.session_state.uploaded_files = []
+if "errors" not in st.session_state:
+    st.session_state.errors = []
 
 uploaded = st.file_uploader("Upload a text/log file", type=["txt", "log"])
 if uploaded:
-    content = uploaded.read().decode("utf-8", errors="ignore")
+    try:
+        content = uploaded.read().decode("utf-8", errors="ignore")
+
+        # Track uploaded file
+        st.session_state.uploaded_files.append(uploaded.name)
+
+        # Keyword search
+        query = st.text_input("🔍 Search for keyword")
+        matches = []
+        if query:
+            matches = keyword_search(content, query)
+            st.write(f"Found {len(matches)} matches")
+            st.code("\n".join(matches[:50]))
+
+        # File preview
+        st.subheader("📋 File Preview")
+        st.code("\n".join(content.splitlines()[:30]))
+
+        # AI Summarization
+        st.subheader("🧠 AI Summary")
+        summary = summarize_text(content)
+        st.write(summary)
 
-    query = st.text_input("🔍 Search for keyword")
-    if query:
-        matches = [line for line in content.splitlines() if re.search(query, line, re.IGNORECASE)]
-        st.write(f"Found {len(matches)} matches")
-        st.code("\n".join(matches[:50]))
+        # Save results to session state (for Chatbot context)
+        st.session_state.omnieye_output = {
+            "file_preview": "\n".join(content.splitlines()[:30]),
+            "matches": matches[:50],
+            "summary": summary,
+        }
 
-    st.subheader("📋 File Preview")
-    st.code("\n".join(content.splitlines()[:30]))
+    except Exception as e:
+        st.error(f"⚠️ Error processing file: {e}")
+        st.session_state.errors.append(str(e))