Update app.py

app.py

@@ -12,6 +12,7 @@ from llama_index.embeddings.gemini import GeminiEmbedding
 from llama_index.core.memory import ChatMemoryBuffer
 from llama_index.readers.web import FireCrawlWebReader
 from llama_index.core import SummaryIndex
+from streamlit_analytics2 import streamlit_analytics
 
 # Setup functions
 def embed_setup():
@@ -84,51 +85,53 @@ if 'chat_history' not in st.query_params:
 if 'last_response' not in st.query_params:
     st.query_params['last_response'] = None
 
-# URL input for document ingestion
-url = st.text_input("Enter URL to crawl and ingest documents:")
 
-# Combined Ingest and Setup button
-if st.button("Ingest and Setup"):
-    if url:
-        with st.spinner("Crawling, ingesting documents, and setting up query engine..."):
-            st.query_params['documents'] = ingest_documents(url)
-            embed_setup()
-            client = qdrant_setup()
-            llm = llm_setup()
-            vector_store = QdrantVectorStore(client=client, collection_name=os.getenv("COLLECTION_NAME"))
-            index = VectorStoreIndex.from_documents(st.query_params['documents'], vector_store=vector_store)
-            st.query_params['chat_engine'] = query_index(index)
-        st.success(f"Documents ingested from {url} and query engine setup completed successfully!")
-    else:
-        st.error("Please enter a URL")
-
-# Query input
-query = st.text_input("Enter your query:")
-
-# Search button
-if st.button("Search"):
-    if st.query_params['chat_engine'] is None:
-        st.error("Please complete the setup first")
-    elif query:
-        with st.spinner("Searching..."):
-            response = st.query_params['chat_engine'].chat(query)
-        
-        # Add the query and response to chat history
-        st.query_params['chat_history'].append(("User", query))
-        st.query_params['chat_history'].append(("Assistant", str(response.response)))
-        
-        # Display the most recent response prominently
-        st.subheader("Assistant's Response:")
-        st.write(response.response)
-    else:
-        st.error("Please enter a query")
-
-# Sidebar for chat history
-st.sidebar.title("Chat History")
-for role, message in st.query_params['chat_history']:
-    st.sidebar.text(f"{role}: {message}")
-
-# Clear chat history button in sidebar
-if st.sidebar.button("Clear Chat History"):
-    st.query_params['chat_history'] = []
-    st.sidebar.success("Chat history cleared!")
+with streamlit_analytics.track():
+    # URL input for document ingestion
+    url = st.text_input("Enter URL to crawl and ingest documents:")
+    
+    # Combined Ingest and Setup button
+    if st.button("Ingest and Setup"):
+        if url:
+            with st.spinner("Crawling, ingesting documents, and setting up query engine..."):
+                st.query_params['documents'] = ingest_documents(url)
+                embed_setup()
+                client = qdrant_setup()
+                llm = llm_setup()
+                vector_store = QdrantVectorStore(client=client, collection_name=os.getenv("COLLECTION_NAME"))
+                index = VectorStoreIndex.from_documents(st.query_params['documents'], vector_store=vector_store)
+                st.query_params['chat_engine'] = query_index(index)
+            st.success(f"Documents ingested from {url} and query engine setup completed successfully!")
+        else:
+            st.error("Please enter a URL")
+    
+    # Query input
+    query = st.text_input("Enter your query:")
+    
+    # Search button
+    if st.button("Search"):
+        if st.query_params['chat_engine'] is None:
+            st.error("Please complete the setup first")
+        elif query:
+            with st.spinner("Searching..."):
+                response = st.query_params['chat_engine'].chat(query)
+            
+            # Add the query and response to chat history
+            st.query_params['chat_history'].append(("User", query))
+            st.query_params['chat_history'].append(("Assistant", str(response.response)))
+            
+            # Display the most recent response prominently
+            st.subheader("Assistant's Response:")
+            st.write(response.response)
+        else:
+            st.error("Please enter a query")
+    
+    # Sidebar for chat history
+    st.sidebar.title("Chat History")
+    for role, message in st.query_params['chat_history']:
+        st.sidebar.text(f"{role}: {message}")
+    
+    # Clear chat history button in sidebar
+    if st.sidebar.button("Clear Chat History"):
+        st.query_params['chat_history'] = []
+        st.sidebar.success("Chat history cleared!")