Update app.py

app.py

@@ -16,7 +16,10 @@ st.title('🤖 AI Assistance For Website')
 
 # Sidebar to capture the API keys
 st.sidebar.title("😎🗝️")
+st.session_state['Website_URL']= st.sidebar.text_input("Input the Website URL you want to Query against?")
 st.session_state['HuggingFace_API_Key']= st.sidebar.text_input("What's your HuggingFace API key?",type="password")
+st.session_state['Pinecone_ENV']= st.sidebar.text_input("What's your Pinecone Environment?")
+st.session_state['Pinecone_Index']= st.sidebar.text_input("What's your Pinecone Index?")
 st.session_state['Pinecone_API_Key']= st.sidebar.text_input("What's your Pinecone API key?",type="password")
 
 load_button = st.sidebar.button("Load data to Pinecone", key="load_button")
@@ -24,7 +27,7 @@ load_button = st.sidebar.button("Load data to Pinecone", key="load_button")
 #If the bove button is clicked, pushing the data to Pinecone...
 if load_button:
     #Proceed only if API keys are provided
-    if st.session_state['HuggingFace_API_Key'] !="" and st.session_state['Pinecone_API_Key']!="" :
+    if st.session_state['Website_URL'] !="" and st.session_state['HuggingFace_API_Key'] !="" and st.session_state['Pinecone_ENV'] !="" and st.session_state['Pinecone_API_Key']!="" :
 
         #Fetch data from site
         site_data=get_website_data(constants.WEBSITE_URL)
@@ -39,12 +42,12 @@ if load_button:
         st.write("Embeddings instance creation done...")
 
         #Push data to Pinecone
-        push_to_pinecone(st.session_state['Pinecone_API_Key'],constants.PINECONE_ENVIRONMENT,constants.PINECONE_INDEX,embeddings,chunks_data)
+        push_to_pinecone(st.session_state['Pinecone_API_Key'],st.session_state['Pinecone_ENV'],st.session_state['Pinecone_Index'],embeddings,chunks_data)
         st.write("Pushing data to Pinecone done...")
 
         st.sidebar.success("Data pushed to Pinecone successfully!")
     else:
-        st.sidebar.error("Ooopssss!!! Please provide API keys.....")
+        st.sidebar.error("Ooopssss!!! Please provide API keys and webstie URL.....")
 
 #********SIDE BAR Funtionality ended*******