mirror of
https://github.com/Mintplex-Labs/anything-llm.git
synced 2024-11-19 20:50:09 +01:00
80 lines
2.4 KiB
Python
80 lines
2.4 KiB
Python
import os
|
|
from InquirerPy import inquirer
|
|
from scripts.youtube import youtube
|
|
from scripts.link import link, links, crawler
|
|
from scripts.substack import substack
|
|
from scripts.medium import medium
|
|
from scripts.gitbook import gitbook
|
|
from scripts.sitemap import sitemap
|
|
|
|
def main():
|
|
if os.name == 'nt':
|
|
methods = {
|
|
'1': 'YouTube Channel',
|
|
'2': 'Article or Blog Link',
|
|
'3': 'Substack',
|
|
'4': 'Medium',
|
|
'5': 'Gitbook',
|
|
'6': 'Sitemap',
|
|
}
|
|
print("There are options for data collection to make this easier for you.\nType the number of the method you wish to execute.")
|
|
print("1. YouTube Channel\n2. Article or Blog Link (Single)\n3. Substack\n4. Medium\n\n[In development]:\nTwitter\n\n")
|
|
selection = input("Your selection: ")
|
|
method = methods.get(str(selection))
|
|
else:
|
|
method = inquirer.select(
|
|
message="What kind of data would you like to add to convert into long-term memory?",
|
|
choices=[
|
|
{"name": "YouTube Channel", "value": "YouTube Channel"},
|
|
{"name": "Substack", "value": "Substack"},
|
|
{"name": "Medium", "value": "Medium"},
|
|
{"name": "Article or Blog Link(s)", "value": "Article or Blog Link(s)"},
|
|
{"name": "Gitbook", "value": "Gitbook"},
|
|
{"name": "Twitter", "value": "Twitter", "disabled": "Needs PR"},
|
|
{"name": "Sitemap", "value": "Sitemap"},
|
|
{"name": "Abort", "value": "Abort"},
|
|
],
|
|
).execute()
|
|
|
|
if 'Article or Blog Link' in method:
|
|
method = inquirer.select(
|
|
message="Do you want to scrape a single article/blog/url or many at once?",
|
|
choices=[
|
|
{"name": "Single URL", "value": "Single URL"},
|
|
{"name": "Multiple URLs", "value": "Multiple URLs"},
|
|
{"name": "URL Crawler", "value": "URL Crawler"},
|
|
{"name": "Abort", "value": "Abort"},
|
|
],
|
|
).execute()
|
|
if method == 'Single URL':
|
|
link()
|
|
exit(0)
|
|
if method == 'Multiple URLs':
|
|
links()
|
|
exit(0)
|
|
if method == 'URL Crawler':
|
|
crawler()
|
|
exit(0)
|
|
|
|
if method == 'Abort': exit(0)
|
|
if method == 'YouTube Channel':
|
|
youtube()
|
|
exit(0)
|
|
if method == 'Substack':
|
|
substack()
|
|
exit(0)
|
|
if method == 'Medium':
|
|
medium()
|
|
exit(0)
|
|
if method == 'Gitbook':
|
|
gitbook()
|
|
exit(0)
|
|
if method == 'Sitemap':
|
|
sitemap()
|
|
exit(0)
|
|
|
|
print("Selection was not valid.")
|
|
exit(1)
|
|
|
|
if __name__ == "__main__":
|
|
main() |