增加了 test_login 单元测试

Anki_flashcards_creator.py

@@ -0,0 +1,76 @@
+# import required packages
+import PyPDF2
+import openai
+import os
+
+# Initialize OpenAI API with your key
+openai.api_key = 'YOUR-API-KEY'
+
+ROOT_DIRECTORY = os.path.dirname(os.path.realpath(__file__))
+
+# Read PDF
+def read_pdf(file_path):
+
+    with open(file_path, 'rb') as file:
+        reader = PyPDF2.PdfReader(file)
+        text = " ".join([page.extract_text() for page in reader.pages])
+
+    return text
+
+# dividing text into smaller chunks:
+def divide_text(text, section_size):
+    sections = []
+    start = 0
+    end = section_size
+    while start < len(text):
+        section = text[start:end]
+        sections.append(section)
+        start = end
+        end += section_size
+    return sections
+
+# Create Anki cards
+def create_anki_cards(pdf_text,):
+    # Limit the number of prompts to avoid excessive API usage
+
+    SECTION_SIZE = 1000
+    divided_sections = divide_text(pdf_text, SECTION_SIZE)
+    # text = divided_sections[0]
+    generated_flashcards = ' '
+    for i, text in enumerate(divided_sections):
+
+        ## You might need to change the Prompt to get consistent format.
+        messages = [
+            {"role": "system", "content": "You are a helpful assistant."},
+            {"role": "user", "content": f"Create anki flashcards with the provided text using a format: question;answer next line question;answer etc. Keep question and the corresponding answer on the same line {text}"}
+            ]
+
+        response = openai.ChatCompletion.create(
+                model="gpt-3.5-turbo",
+                messages=messages, 
+                temperature =0.3,
+                max_tokens=2048
+            )
+
+        response_from_api = response['choices'][0]['message']['content']#.strip()
+        generated_flashcards += response_from_api
+
+        if i==0:
+            break
+
+    # # Save the cards to a text file
+    with open("flashcards.txt", "w") as f:
+        f.write(generated_flashcards)
+
+
+# Main script execution
+if __name__ == "__main__":
+
+    pdf_text = read_pdf(f'{ROOT_DIRECTORY}/SOURCE_DOCUMENTS/constitution.pdf')
+
+    create_anki_cards(pdf_text)
+
+
+
+
+