Update app.py

app.py

@@ -18,13 +18,11 @@ GROQ_API_KEY = "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"
 # Custom CSS for glassy, colorful, and shadowy UI
 st.markdown("""
     <style>
-    /* Global styles */
     body {
         background: linear-gradient(135deg, #1e3c72, #2a5298);
         color: #ffffff;
         font-family: 'Arial', sans-serif;
     }
-    /* Glassy container for sidebar and main content */
     .stSidebar, .main .block-container {
         background: rgba(255, 255, 255, 0.1);
         border-radius: 15px;
@@ -34,7 +32,6 @@ st.markdown("""
         box-shadow: 0 8px 32px rgba(0, 0, 0, 0.2);
         padding: 20px;
     }
-    /* Input fields */
     .stTextInput > div > input {
         background: rgba(255, 255, 255, 0.15);
         color: #ffffff;
@@ -43,7 +40,6 @@ st.markdown("""
         padding: 10px;
         box-shadow: 0 4px 12px rgba(0, 0, 0, 0.1);
     }
-    /* Buttons */
     .stButton > button {
         background: linear-gradient(45deg, #6b48ff, #00ddeb);
         color: #ffffff;
@@ -58,30 +54,25 @@ st.markdown("""
         transform: translateY(-2px);
         box-shadow: 0 6px 16px rgba(0, 0, 0, 0.3);
     }
-    /* Headers */
     h1, h2, h3 {
         color: #ffffff;
         text-shadow: 0 2px 4px rgba(0, 0, 0, 0.2);
     }
-    /* Status messages */
     .stText {
         color: #e0e0e0;
         font-weight: bold;
     }
-    /* Error messages */
     .stAlert {
         background: rgba(255, 50, 50, 0.2);
         border: 1px solid rgba(255, 50, 50, 0.5);
         border-radius: 10px;
         color: #ffcccc;
     }
-    /* Warning messages */
     .stAlert[role="alert"] > div {
         background: rgba(255, 200, 0, 0.2);
         border: 1px solid rgba(255, 200, 0, 0.5);
         color: #fff5cc;
     }
-    /* Spinner */
     .stSpinner > div {
         color: #00ddeb;
     }
@@ -91,11 +82,9 @@ st.markdown("""
 # Set Streamlit app title
 st.title("WebChatter 💬")
 
-# Initialize session state for FAISS index and processing status
+# Initialize session state
 if "index_created" not in st.session_state:
     st.session_state.index_created = False
-if "processing_status" not in st.session_state:
-    st.session_state.processing_status = ""
 
 # Sidebar for URL input
 with st.sidebar:
@@ -124,54 +113,43 @@ if process_url_clicked:
         if not url.strip():
             st.error("Please provide a valid URL.")
         else:
-            try:
-                st.session_state.processing_status = "Data Loading...Started...✅✅✅"
-                st.text(st.session_state.processing_status)
-                loader = WebBaseLoader(
-                    web_path=url.strip(),
-                    bs_kwargs={"parse_only": ["title", "p", "h1", "h2", "h3"]},
-                    requests_kwargs={"headers": {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36"}}
-                )
-                data = loader.load()
-
-                # Check loaded data
-                if not data or all(len(doc.page_content.strip()) == 0 for doc in data):
-                    st.error("No content loaded from URL. Try a different URL (e.g., https://www.bbc.com/news/science-environment-67299122).")
-                    st.session_state.processing_status = ""
-                    st.stop()
-
-                st.session_state.processing_status = "Text Splitter...Started...✅✅✅"
-                st.text(st.session_state.processing_status)
-                text_splitter = RecursiveCharacterTextSplitter(
-                    separators=['\n\n', '\n', '.', ','],
-                    chunk_size=1000
-                )
-                docs = text_splitter.split_documents(data)
-
-                # Check document count
-                if not docs:
-                    st.error("No document chunks created. Try a different URL.")
-                    st.session_state.processing_status = ""
-                    st.stop()
-                st.session_state.processing_status = f"Split into {len(docs)} document chunks."
-                st.text(st.session_state.processing_status)
-
-                st.session_state.processing_status = "Embedding Vector Started Building...✅✅✅"
-                st.text(st.session_state.processing_status)
-                embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
-                vectorstore_openai = FAISS.from_documents(docs, embeddings)
-
-                faiss_index_path = "faiss_index"
-                save_faiss_index(vectorstore_openai, faiss_index_path)
-                st.session_state.index_created = True
-                st.session_state.processing_status = "FAISS index saved successfully! ✅✅✅"
-                st.text(st.session_state.processing_status)
-                time.sleep(2)
-                st.session_state.processing_status = ""
-                st.experimental_rerun()  # Refresh to clear status messages
-            except Exception as e:
-                st.error(f"Error processing URL: {str(e)}")
-                st.session_state.processing_status = ""
+            with st.spinner("Processing URL..."):
+                try:
+                    st.text("Data Loading...Started...✅✅✅")
+                    loader = WebBaseLoader(
+                        web_path=url.strip(),
+                        bs_kwargs={"parse_only": ["title", "p", "h1", "h2", "h3"]},
+                        requests_kwargs={"headers": {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36"}}
+                    )
+                    data = loader.load()
+
+                    if not data or all(len(doc.page_content.strip()) == 0 for doc in data):
+                        st.error("No content loaded from URL. Try a different URL (e.g., https://www.bbc.com/news/science-environment-67299122).")
+                        st.stop()
+
+                    st.text("Text Splitter...Started...✅✅✅")
+                    text_splitter = RecursiveCharacterTextSplitter(
+                        separators=['\n\n', '\n', '.', ','],
+                        chunk_size=1000
+                    )
+                    docs = text_splitter.split_documents(data)
+
+                    if not docs:
+                        st.error("No document chunks created. Try a different URL.")
+                        st.stop()
+                    st.text(f"Split into {len(docs)} document chunks.")
+
+                    st.text("Embedding Vector Started Building...✅✅✅")
+                    embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
+                    vectorstore_openai = FAISS.from_documents(docs, embeddings)
+
+                    faiss_index_path = "faiss_index"
+                    save_faiss_index(vectorstore_openai, faiss_index_path)
+                    st.session_state.index_created = True
+                    st.text("FAISS index saved successfully! ✅✅✅")
+                    time.sleep(2)
+                except Exception as e:
+                    st.error(f"Error processing URL: {str(e)}")
 
 # Query input with Ask button
 with main_container: