28 lines
708 B
Python
28 lines
708 B
Python
|
import sys
|
||
|
|
import os
|
||
|
|
|
||
|
|
from images2text import main as images_to_text
|
||
|
|
from text2csvdeck import create_csv_deck
|
||
|
|
|
||
|
|
CSV_DECK_NAME = "output_deck.csv"
|
||
|
|
APKG_NAME = "output.apkg"
|
||
|
|
|
||
|
|
|
||
|
|
def pipeline(directory_path):
|
||
|
|
# 1. Convert images in the directory to a text file
|
||
|
|
text_file_name = images_to_text(directory_path)
|
||
|
|
|
||
|
|
# 2. Convert the text file to a CSV deck using ChatGPT
|
||
|
|
create_csv_deck(text_file_name)
|
||
|
|
|
||
|
|
# 3. Convert the CSV deck to an Anki package
|
||
|
|
os.system(f"python csv2ankicards.py {CSV_DECK_NAME} {APKG_NAME}")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
if len(sys.argv) != 2:
|
||
|
|
print("Usage: python pipeline.py <directory_path_containing_images>")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
pipeline(sys.argv[1])
|