vsubu1 commited on
Commit
59fe3f8
·
1 Parent(s): 9deb690

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import streamlit as st
2
  from utils import *
3
- import Constants
4
 
5
  # Creating Session State Variable
6
  if 'HuggingFace_API_Key' not in st.session_state:
@@ -27,7 +27,7 @@ if load_button:
27
  if st.session_state['HuggingFace_API_Key'] !="" and st.session_state['Pinecone_API_Key']!="" :
28
 
29
  #Fetch data from site
30
- site_data=get_website_data(constants.WEBSITE_URL)
31
  st.write("Data pull done...")
32
 
33
  #Split data into chunks
@@ -39,7 +39,7 @@ if load_button:
39
  st.write("Embeddings instance creation done...")
40
 
41
  #Push data to Pinecone
42
- push_to_pinecone(st.session_state['Pinecone_API_Key'],constants.PINECONE_ENVIRONMENT,constants.PINECONE_INDEX,embeddings,chunks_data)
43
  st.write("Pushing data to Pinecone done...")
44
 
45
  st.sidebar.success("Data pushed to Pinecone successfully!")
@@ -64,7 +64,7 @@ if submit:
64
  st.write("Embeddings instance creation done...")
65
 
66
  #Pull index data from Pinecone
67
- index=pull_from_pinecone(st.session_state['Pinecone_API_Key'],constants.PINECONE_ENVIRONMENT,constants.PINECONE_INDEX,embeddings)
68
  st.write("Pinecone index retrieval done...")
69
 
70
  #Fetch relavant documents from Pinecone index
 
1
  import streamlit as st
2
  from utils import *
3
+ import constants as cs
4
 
5
  # Creating Session State Variable
6
  if 'HuggingFace_API_Key' not in st.session_state:
 
27
  if st.session_state['HuggingFace_API_Key'] !="" and st.session_state['Pinecone_API_Key']!="" :
28
 
29
  #Fetch data from site
30
+ site_data=get_website_data(cs.WEBSITE_URL)
31
  st.write("Data pull done...")
32
 
33
  #Split data into chunks
 
39
  st.write("Embeddings instance creation done...")
40
 
41
  #Push data to Pinecone
42
+ push_to_pinecone(st.session_state['Pinecone_API_Key'],cs.PINECONE_ENVIRONMENT,cs.PINECONE_INDEX,embeddings,chunks_data)
43
  st.write("Pushing data to Pinecone done...")
44
 
45
  st.sidebar.success("Data pushed to Pinecone successfully!")
 
64
  st.write("Embeddings instance creation done...")
65
 
66
  #Pull index data from Pinecone
67
+ index=pull_from_pinecone(st.session_state['Pinecone_API_Key'],cs.PINECONE_ENVIRONMENT,cs.PINECONE_INDEX,embeddings)
68
  st.write("Pinecone index retrieval done...")
69
 
70
  #Fetch relavant documents from Pinecone index