Update app.py

app.py

@@ -12,7 +12,6 @@ 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
-import streamlit_analytics
 
 # Setup functions
 def embed_setup():
@@ -85,52 +84,51 @@ if 'chat_history' not in st.query_params:
 if 'last_response' not in st.query_params:
     st.query_params['last_response'] = None
 
-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!")
+# 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!")