Update app.py

app.py

@@ -15,24 +15,92 @@ load_dotenv()
 # Hardcoded Groq API key (NOT RECOMMENDED for production)
 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;
+        backdrop-filter: blur(10px);
+        -webkit-backdrop-filter: blur(10px);
+        border: 1px solid rgba(255, 255, 255, 0.2);
+        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;
+        border: 1px solid rgba(255, 255, 255, 0.3);
+        border-radius: 10px;
+        padding: 10px;
+        box-shadow: 0 4px 12px rgba(0, 0, 0, 0.1);
+    }
+    /* Buttons */
+    .stButton > button {
+        background: linear-gradient(45deg, #6b48ff, #00ddeb);
+        color: #ffffff;
+        border: none;
+        border-radius: 10px;
+        padding: 10px 20px;
+        font-weight: bold;
+        box-shadow: 0 4px 12px rgba(0, 0, 0, 0.2);
+        transition: transform 0.2s;
+    }
+    .stButton > button:hover {
+        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;
+    }
+    </style>
+""", unsafe_allow_html=True)
+
 # Set Streamlit app title
-st.title("News Research Tool 📈")
-st.sidebar.title("News Article URLs")
+st.title("WebChatter 💬")
 
 # Initialize session state for FAISS index
 if "index_created" not in st.session_state:
     st.session_state.index_created = False
 
-# Get URLs from user input
-urls = []
-for i in range(3):
-    url = st.sidebar.text_input(f"URL {i+1}")
-    if url:
-        urls.append(url)
-
-# Button to process URLs
-process_url_clicked = st.sidebar.button("Process URLs")
-faiss_index_path = "faiss_index"
+# Sidebar for URL input
+with st.sidebar:
+    st.header("Enter Web URL")
+    url = st.text_input("URL", placeholder="e.g., https://www.bbc.com/news/science-environment-67299122")
+    urls = [url] if url else []
+    process_url_clicked = st.button("Process URL")
 
 # Placeholder for main content
 main_placeholder = st.empty()
@@ -51,16 +119,20 @@ def load_faiss_index(path, embeddings):
 
 if process_url_clicked:
     if not urls:
-        main_placeholder.error("Please provide at least one valid URL.")
+        main_placeholder.error("Please provide a valid URL.")
     else:
         try:
             main_placeholder.text("Data Loading...Started...✅✅✅")
-            loader = WebBaseLoader(urls)
+            loader = WebBaseLoader(
+                web_paths=urls,
+                bs_kwargs={"parse_only": ["title", "p", "h1", "h2", "h3"]},
+                requests_kwargs={"headers": {"User-Agent": "Mozilla/5.0"}}
+            )
             data = loader.load()
 
             # Check loaded data
             if not data or all(len(doc.page_content.strip()) == 0 for doc in data):
-                main_placeholder.error("No content loaded from URLs. Try different URLs (e.g., https://www.bbc.com/news/science-environment-67299122).")
+                main_placeholder.error("No content loaded from URL. Try a different URL (e.g., https://www.bbc.com/news/science-environment-67299122).")
                 st.stop()
 
             main_placeholder.text("Text Splitter...Started...✅✅✅")
@@ -72,7 +144,7 @@ if process_url_clicked:
 
             # Check document count
             if not docs:
-                main_placeholder.error("No document chunks created. Try different URLs.")
+                main_placeholder.error("No document chunks created. Try a different URL.")
                 st.stop()
             main_placeholder.text(f"Split into {len(docs)} document chunks.")
 
@@ -86,12 +158,17 @@ if process_url_clicked:
             time.sleep(2)
             main_placeholder.empty()
         except Exception as e:
-            main_placeholder.error(f"Error processing URLs: {str(e)}")
+            main_placeholder.error(f"Error processing URL: {str(e)}")
+
+# Query input with Ask button
+with main_placeholder.container():
+    st.header("Ask a Question")
+    query = st.text_input("Question", placeholder="e.g., What is the article about?")
+    ask_clicked = st.button("Ask")
 
-query = main_placeholder.text_input("Question: ")
-if query:
+if ask_clicked and query:
     if not st.session_state.index_created or not os.path.exists(faiss_index_path):
-        main_placeholder.error("No FAISS index found. Please process URLs first.")
+        main_placeholder.error("No FAISS index found. Please process a URL first.")
     else:
         with st.spinner("Processing your question..."):
             try:
@@ -101,7 +178,7 @@ if query:
                 result = chain({"question": query}, return_only_outputs=True)
 
                 if not result.get("answer"):
-                    main_placeholder.warning("No answer generated. Try a different question or URLs.")
+                    main_placeholder.warning("No answer generated. Try a different question or URL.")
                     st.stop()
 
                 st.header("Answer")