Update app.py

app.py

@@ -1,15 +1,5 @@
 import streamlit as st
 import re
-from langchain_groq import ChatGroq
-from langchain import hub
-from langchain_chroma import Chroma
-from langchain_community.document_loaders import WebBaseLoader
-from langchain_core.output_parsers import StrOutputParser
-from langchain_core.runnables import RunnablePassthrough
-from langchain_text_splitters import RecursiveCharacterTextSplitter
-from sentence_transformers import SentenceTransformer
-import bs4
-import torch
 import os
 
 # Sidebar Style with Multicolored Background
@@ -22,6 +12,20 @@ sidebar_bg_style = """
 """
 st.markdown(sidebar_bg_style, unsafe_allow_html=True)
 
+# Main Content Style with Multicolored Background
+main_bg_style = """
+    <style>
+        .main .block-container {
+            background: linear-gradient(135deg, #ff9a9e, #fad0c4, #fbc2eb, #a18cd1);
+            padding: 2rem;
+        }
+        .css-18e3th9 {
+            background: linear-gradient(135deg, #ff9a9e, #fad0c4, #fbc2eb, #a18cd1);
+        }
+    </style>
+"""
+st.markdown(main_bg_style, unsafe_allow_html=True)
+
 # Sidebar: Input for URL and API keys
 st.sidebar.title("Settings")
 
@@ -35,37 +39,26 @@ if url_input:
     else:
         st.sidebar.markdown('<p style="color:red; font-weight:bold;">Invalid URL, please enter a valid one</p>', unsafe_allow_html=True)
 
+# Option to use pre-provided API keys
+use_preprovided_keys = st.sidebar.checkbox("Use pre-provided API keys")
+
 # Input fields for API keys with placeholders and helper text
-api_key_1 = st.sidebar.text_input("Enter LangChain API Key", type="password", placeholder="Enter your LangChain API Key", help="Please enter a valid LangChain API key here")
-api_key_2 = st.sidebar.text_input("Enter Groq API Key", type="password", placeholder="Enter your Groq API Key", help="Please enter your Groq API key here")
+if not use_preprovided_keys:
+    api_key_1 = st.sidebar.text_input("Enter LangChain API Key", type="password", placeholder="Enter your LangChain API Key", help="Please enter a valid LangChain API key here")
+    api_key_2 = st.sidebar.text_input("Enter Groq API Key", type="password", placeholder="Enter your Groq API Key", help="Please enter your Groq API key here")
+else:
+    api_key_1 = "your-preprovided-langchain-api-key"  # Replace with your actual pre-provided key
+    api_key_2 = "your-preprovided-groq-api-key"  # Replace with your actual pre-provided key
+    st.sidebar.markdown('<p style="color:blue; font-weight:bold;">Using pre-provided API keys</p>', unsafe_allow_html=True)
 
 # Submit button for API keys with a success/warning message
 if st.sidebar.button("Submit API Keys"):
-    if api_key_1 and api_key_2:
+    if use_preprovided_keys or (api_key_1 and api_key_2):
         os.environ["LANGCHAIN_API_KEY"] = api_key_1
         os.environ["GROQ_API_KEY"] = api_key_2
-        st.sidebar.markdown('<p style="color:green; font-weight:bold;">Both API keys are entered</p>', unsafe_allow_html=True)
+        st.sidebar.markdown('<p style="color:green; font-weight:bold;">API keys are set</p>', unsafe_allow_html=True)
     else:
-        st.sidebar.markdown('<p style="color:red; font-weight:bold;">Please fill in both API keys</p>', unsafe_allow_html=True)
-
-# Main Section with Multicolored Background and Chatbot Title
-main_bg_style = """
-    <style>
-        .main-content {
-            background: linear-gradient(135deg, #ff9a9e, #fad0c4, #fbc2eb, #a18cd1);
-            padding: 20px;
-            border-radius: 10px;
-            color: #333; /* Ensure text is visible against the background */
-        }
-        .stTextInput>div>div>input {
-            border: 1px solid #ccc; /* Style input borders */
-        }
-    </style>
-"""
-st.markdown(main_bg_style, unsafe_allow_html=True)
-
-# Wrapper for main content to apply background style
-st.markdown('<div class="main-content">', unsafe_allow_html=True)
+        st.sidebar.markdown('<p style="color:red; font-weight:bold;">Please fill in both API keys or select the option to use pre-provided keys</p>', unsafe_allow_html=True)
 
 # Title of the chatbot
 st.markdown('<h1 style="color:#4CAF50; font-weight:bold;">🤖 Chatbot with URL-based Document Retrieval</h1>', unsafe_allow_html=True)
@@ -77,11 +70,9 @@ query = st.text_input("Ask a question based on the blog post", placeholder="Type
 if 'chat_history' not in st.session_state:
     st.session_state['chat_history'] = []
 
-# Warning message for empty query
+# Submit button for chat
 if st.button("Submit Query"):
-    if not query:
-        st.markdown('<p style="color:red; font-weight:bold;">Please enter a question before submitting.</p>', unsafe_allow_html=True)
-    elif url_input:
+    if query and url_input:
         # Blog loading logic based on user input URL
         loader = WebBaseLoader(
             web_paths=(url_input,),  # Use the user-input URL
@@ -140,7 +131,4 @@ if st.button("Submit Query"):
 # Display chat history
 for q, r in st.session_state['chat_history']:
     st.write(f"**User:** {q}")
-    st.write(f"**Bot:** {r}")
-
-# Close the main content wrapper
-st.markdown('</div>', unsafe_allow_html=True)
+    st.write(f"**Bot:** {r}")