Update app.py

app.py

@@ -1,32 +1,6 @@
 import streamlit as st
 import os
 from pinecone import Pinecone
-# Set up the Streamlit app
-st.set_page_config(page_title="Search Engine", layout="wide")
-
-# Set up the Streamlit app title and search bar
-st.title("Search Engine")
-index_name = st.text_input("Enter a database name:", "")
-key = st.text_input("Enter a key:", "")
-namespace = st.text_input("Enter a table name:", "")
-# initialize connection to pinecone (get API key at app.pinecone.io)
-api_key = os.environ.get('PINECONE_API_KEY') or key
-
-# configure client
-pc = Pinecone(api_key=api_key)
-
-from pinecone import ServerlessSpec
-
-cloud = os.environ.get('PINECONE_CLOUD') or 'aws'
-region = os.environ.get('PINECONE_REGION') or 'us-east-1'
-
-spec = ServerlessSpec(cloud=cloud, region=region)
-
-
-# connect to index
-index = pc.Index(index_name)
-
-
 from sentence_transformers import SentenceTransformer
 # import torch
 
@@ -34,7 +8,37 @@ from sentence_transformers import SentenceTransformer
 
 model = SentenceTransformer('intfloat/e5-small')
 
+# Set up the Streamlit app
+st.set_page_config(page_title="Search Engine", layout="wide")
+
+# Set up the Streamlit app title and search bar
+st.title("Search Engine")
+st.button("Connect to Search Engine Database", type="primary")
+if st.button():
+    index_name = st.text_input("Enter a database name:", "")
+    key = st.text_input("Enter a key:", "")
+    namespace = st.text_input("Enter a table name:", "")
+    # initialize connection to pinecone (get API key at app.pinecone.io)
+    api_key = os.environ.get('PINECONE_API_KEY') or key
+    
+    # configure client
+    pc = Pinecone(api_key=api_key)
+    
+    from pinecone import ServerlessSpec
+    
+    cloud = os.environ.get('PINECONE_CLOUD') or 'aws'
+    region = os.environ.get('PINECONE_REGION') or 'us-east-1'
+    
+    spec = ServerlessSpec(cloud=cloud, region=region)
+    
+    
+    # connect to index
+    index = pc.Index(index_name)
+    st.write('Successfully connected to your Search Engine DB!')
+else:
+    st.write("Goodbye")
 
+    
 query = st.text_input("Enter a search query:", "")
 
 # If the user has entered a search query, search the Pinecone index with the query