Skip to content

Added an option for the user to specify a repo folder by providing a … #1

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Added an option for the user to specify a repo folder by providing a … #1

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
21 changes: 20 additions & 1 deletion app.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@
STORE_DIR = "store"
CLONE_NEW_REPO = "Clone new repo"
LOAD_EXISTING_INDEX = "Load existing index"
SPECIFIC_FOLDER = "Specify a specific folder path"

def ask_for_repo():
repo_url = input("Enter repo url: ")
Expand Down Expand Up @@ -100,6 +101,22 @@ def existing_repo_flow():

return index

def folder_path_flow():
# Ask user to enter the folder path
folder_path = input("Enter the folder path of the repo: ")

# Check if the folder exists
if not os.path.exists(folder_path):
print("The specified folder does not exist. Exiting...")
return None

# Load documents & create the index
print("Creating index...")
documents = SimpleDirectoryReader(input_dir=folder_path, exclude_hidden=True).load_data()
index = GPTSimpleVectorIndex(documents)

return index

def main():
# Declare index
index = None
Expand All @@ -122,7 +139,7 @@ def main():
# Ask if user wants to clone new repo or load existing index
answer = inquirer.prompt(
[inquirer.List("type", message="Would you like to clone a new repo or load an existing index?", choices=[
CLONE_NEW_REPO, LOAD_EXISTING_INDEX
CLONE_NEW_REPO, LOAD_EXISTING_INDEX, SPECIFIC_FOLDER
])]
)["type"]

Expand All @@ -133,6 +150,8 @@ def main():
elif answer == LOAD_EXISTING_INDEX:
# Existing index
index = existing_repo_flow()
elif answer == SPECIFIC_FOLDER:
index = folder_path_flow()
else:
print("Invalid input, exiting...")
return
Expand Down