Update app.py

app.py

@@ -2,6 +2,7 @@ import spaces
 import gradio as gr
 from PyPDF2 import PdfReader
 from io import BytesIO
+from fpdf import FPDF  # Importiere FPDF, um PDF-Dateien zu erstellen
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 import re
@@ -60,17 +61,34 @@ def generate_flashcards(text):
     except Exception as e:
         return f"Fehler bei der Generierung der Karteikarten: {str(e)}"
 
+def create_pdf(flashcards):
+    pdf = FPDF()
+    pdf.add_page()
+    pdf.set_font("Arial", size=12)
+    
+    for question, answer in flashcards:
+        pdf.multi_cell(0, 10, f"Frage: {question}")
+        pdf.multi_cell(0, 10, f"Antwort: {answer}")
+        pdf.ln(10)  # Adds space between flashcards
+    
+    pdf_file_path = "/mnt/data/flashcards_output.pdf"
+    pdf.output(pdf_file_path)
+    return pdf_file_path
+
 @spaces.GPU(duration=60)
 def process_pdf(pdf_file):
     if pdf_file is None:
-        return []
+        return [], None
     
     text = extract_text_from_pdf(pdf_file)
     if text.startswith("Fehler beim Lesen der PDF"):
-        return []
+        return [], None
     
     flashcards = generate_flashcards(text)
-    return flashcards if isinstance(flashcards, list) else []
+    if isinstance(flashcards, list):
+        pdf_path = create_pdf(flashcards)
+        return flashcards, pdf_path
+    return [], None
 
 def update_flashcard(flashcards, index, current_side):
     if not flashcards or index >= len(flashcards):
@@ -108,10 +126,12 @@ with gr.Blocks() as iface:
         question_box = gr.Textbox(label="Frage", interactive=False)
         answer_box = gr.Textbox(label="Antwort", interactive=False, visible=False)
     
+    pdf_download_button = gr.File(label="Download Flashcards PDF")
+
     generate_button.click(
         process_pdf,
         inputs=[pdf_input],
-        outputs=[flashcards_state]
+        outputs=[flashcards_state, pdf_download_button]
     ).then(
         lambda cards: update_flashcard(cards, 0, "question") + (0, "question"),
         inputs=[flashcards_state],