User specified username, repo, and file type selector. Need to deal with access token

github_st.py

@@ -11,10 +11,8 @@ load_dotenv()
 
 #get the GITHUB_ACCESS_TOKEN from the .env file
 GITHUB_ACCESS_TOKEN = os.getenv("GITHUB_ACCESS_TOKEN")
-USER = "heaversm"
-REPO = "gdrive-docker"
 GITHUB_BASE_URL = "https://github.com/"
-FILE_TYPES_TO_LOAD = (".py", ".ts")
+
 
 @st.cache_resource
 def get_hugging_face_model():
@@ -32,6 +30,16 @@ def get_similar_files(query, db, embeddings):
 
 
 st.title("Find Similar Code")
+
+#streamlit text input for USER
+USER = st.text_input("Enter the Github User", value = "heaversm")
+#streamlit text input for REPO
+REPO = st.text_input("Enter the Github Repository", value = "gdrive-docker")
+
+#streamlit file type selector
+FILE_TYPES_TO_LOAD = st.multiselect("Select File Types", [".py", ".ts",".js",".css",".html"], default = [".py"])
+
+
 text_input = st.text_area("Enter a Code Example", value =
 """
 def create_app():
@@ -43,6 +51,8 @@ def create_app():
 )
 
 button = st.button("Find Similar Code")
+
+
 if button:
   loader = GithubFileLoader(
     #repo is USER/REPO
@@ -50,7 +60,7 @@ if button:
     access_token=GITHUB_ACCESS_TOKEN,
     github_api_url="https://api.github.com",
     file_filter=lambda file_path: file_path.endswith(
-        FILE_TYPES_TO_LOAD
+      tuple(FILE_TYPES_TO_LOAD)
     )
   )
   documents = loader.load()