Skip to content

Update Anki_flashcards_creator.py #2

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

Draft
0xj0n1 wants to merge 1 commit into
base: main
Choose a base branch
from
Draft

Update Anki_flashcards_creator.py #2

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
148 changes: 114 additions & 34 deletions Anki_flashcards_creator.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,21 +1,58 @@
# import required packages
import PyPDF2
import openai
import os
from openai import OpenAI
import sys

# SICHERE Methode für API-Schlüssel
# Option 1: Umgebungsvariable verwenden (empfohlen)
api_key = os.environ.get("OPENAI_API_KEY")

# Option 2: Aus einer separaten, nicht unter Versionskontrolle stehenden Datei laden
if not api_key:
try:
with open(".env", "r") as f:
for line in f:
if line.strip().startswith("OPENAI_API_KEY="):
api_key = line.strip().split("=", 1)[1].strip()
break
except Exception as e:
print(f"Fehler beim Lesen der .env-Datei: {e}")

# Prüfen ob API-Schlüssel vorhanden ist
if not api_key:
print("Fehler: Kein OpenAI API-Schlüssel gefunden.")
print("Bitte setzen Sie die Umgebungsvariable OPENAI_API_KEY oder erstellen Sie eine .env-Datei.")
sys.exit(1)

# OpenAI Client initialisieren
client = OpenAI(api_key=api_key)

# Initialize OpenAI API with your key
openai.api_key = 'YOUR-API-KEY'

ROOT_DIRECTORY = os.path.dirname(os.path.realpath(__file__))
SOURCE_DIR = os.path.join(ROOT_DIRECTORY, "SOURCE_DOCUMENTS")
OUTPUT_DIR = ROOT_DIRECTORY

# Sicherstellen, dass das Quellverzeichnis existiert
if not os.path.exists(SOURCE_DIR):
os.makedirs(SOURCE_DIR)
print(f"Verzeichnis erstellt: {SOURCE_DIR}")
print("Bitte legen Sie Ihre PDF-Dateien in diesem Verzeichnis ab.")

# 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
try:
with open(file_path, 'rb') as file:
reader = PyPDF2.PdfReader(file)
text = " ".join([page.extract_text() for page in reader.pages])
return text
except FileNotFoundError:
print(f"Fehler: Die Datei {file_path} wurde nicht gefunden.")
print(f"Bitte legen Sie die PDF-Datei im Verzeichnis {SOURCE_DIR} ab.")
return None
except Exception as e:
print(f"Fehler beim Lesen der PDF-Datei: {e}")
return None

# dividing text into smaller chunks:
def divide_text(text, section_size):
Expand All @@ -30,45 +67,88 @@ def divide_text(text, section_size):
return sections

# Create Anki cards
def create_anki_cards(pdf_text,):
# Limit the number of prompts to avoid excessive API usage

def create_anki_cards(pdf_text):
if not pdf_text:
return False

SECTION_SIZE = 1000
divided_sections = divide_text(pdf_text, SECTION_SIZE)
# text = divided_sections[0]
generated_flashcards = ' '
for i, text in enumerate(divided_sections):
generated_flashcards = ''

print(f"Verarbeite {len(divided_sections)} Textabschnitte...")

## 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}"}
try:
for i, text in enumerate(divided_sections):
print(f"Verarbeite Abschnitt {i+1}/{len(divided_sections)}...")

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(
# Neue OpenAI API-Syntax verwenden
response = client.chat.completions.create(
model="gpt-3.5-turbo",
messages=messages,
temperature =0.3,
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)

response_from_api = response.choices[0].message.content
generated_flashcards += response_from_api + "\n\n"

# Diese Zeile entfernt, damit alle Abschnitte verarbeitet werden
# if i==0:
# break

# Speichern mit vollständigem Pfad
output_path = os.path.join(OUTPUT_DIR, "flashcards.txt")
with open(output_path, "w", encoding="utf-8") as f:
f.write(generated_flashcards)

print(f"Erfolgreich! Flashcards wurden in {output_path} gespeichert.")
return True

except Exception as e:
print(f"Fehler bei der Erstellung der Flashcards: {e}")
return False

# Main script execution
if __name__ == "__main__":

pdf_text = read_pdf(f'{ROOT_DIRECTORY}/SOURCE_DOCUMENTS/constitution.pdf')
# PDF-Datei
pdf_file = "Global Business - Unit 2.pdf"
pdf_path = os.path.join(SOURCE_DIR, pdf_file)

# Prüfen, ob die Datei existiert
if not os.path.exists(pdf_path):
print(f"Die Datei {pdf_file} wurde nicht gefunden in {SOURCE_DIR}.")
print("Verfügbare PDF-Dateien:")
pdf_files = [f for f in os.listdir(SOURCE_DIR) if f.lower().endswith('.pdf')]

if pdf_files:
for i, file in enumerate(pdf_files):
print(f"{i+1}. {file}")
selection = input("Wählen Sie eine Datei aus (Nummer eingeben) oder drücken Sie Enter, um zu beenden: ")

if selection.isdigit() and 1 <= int(selection) <= len(pdf_files):
pdf_file = pdf_files[int(selection)-1]
pdf_path = os.path.join(SOURCE_DIR, pdf_file)
else:
print("Keine gültige Auswahl. Programm wird beendet.")
sys.exit(1)
else:
print(f"Keine PDF-Dateien in {SOURCE_DIR} gefunden.")
sys.exit(1)

# PDF lesen und Flashcards erstellen
print(f"Lese PDF: {pdf_file}...")
pdf_text = read_pdf(pdf_path)

create_anki_cards(pdf_text)
if pdf_text:
print("PDF erfolgreich gelesen. Erstelle Flashcards...")
create_anki_cards(pdf_text)
else:
print("Programm wird beendet, da die PDF-Datei nicht gelesen werden konnte.")



Expand Down