Update utils.py

utils.py

@@ -3,7 +3,7 @@ from typing import TYPE_CHECKING, Any, Callable, Dict, List, Tuple, Type
 import logging
 import json
 import os
-import datetime
+from datetime import datetime
 import hashlib
 import csv
 import requests
@@ -58,6 +58,14 @@ from sklearn.feature_extraction.text import TfidfVectorizer
 from sklearn.metrics.pairwise import cosine_similarity
 import numpy as np
 
+from reportlab.lib.pagesizes import inch, A4
+from reportlab.platypus import SimpleDocTemplate, Frame, Spacer
+from reportlab.lib import colors
+from reportlab.lib.units import mm
+from reportlab.platypus import Paragraph, SimpleDocTemplate, Frame, Image, Table
+from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
+from reportlab.lib.units import cm
+
 
 logging.basicConfig(
     level=logging.INFO,
@@ -734,6 +742,74 @@ def create_picture(history,  prompt):
     return image
 
 
+########################################
+# Ausgabe in PDF des Chathistory
+########################################
+#callback Methode, die auf jeder PDF Seite das Datum oben hinschreibt
+def on_each_page(canvas, doc):
+    page_width, page_height = A4  # Oder das von Ihnen verwendete Seitenformat, z.B. A4
+    canvas.saveState()
+    canvas.setFont('Times-Roman', 10)
+    # Formatieren Sie das Datum nach Ihrem Wunsch
+    current_date = datetime.now().strftime("%Y-%m-%d")
+    print(current_date)
+    # Positionieren Sie das Datum oben rechts auf der Seite
+    canvas.drawRightString(page_width - 72, page_height - 28, current_date)
+    canvas.restoreState()
+
+#PDF Inhalte zusammenstellen und PDF unter dem angegebene Pfad file_path_download ablegen
+def erstellePdf(file_path_download, ueberschrift, dic_history):
+    # Initialisiere eine leere Listseinstellung ("flowables"), die später gefüllt wird
+    elements = []
+    # Definiere ein neues Papierformat mit A4 Maßen
+    paper_size = A4
+
+    # Erstellen Sie ein neues StyleSheet-Objekt
+    styles = getSampleStyleSheet()
+    # Neuen Style hinzufügen
+    new_style = ParagraphStyle('NewStyle', fontName='Helvetica', fontSize=12)
+    styles.add(new_style)
+    #style für Trennlinie
+    line_style = ParagraphStyle('LineStyle', fontSize=4, leading=6, borderPadding=0,
+                             spaceBefore=0, spaceAfter=0, textColor='black')
+    #Feststehende Überschriften erzeugen
+    # Chat-Überschrift
+    title = Paragraph(ueberschrift, styles['Title'])
+    headline_nutzer = Paragraph('Nutzer:', styles['Heading3'])
+    headline_assi = Paragraph('Assistent:', styles['Heading3'])
+
+    #Pdf Abschnittsweise zusammenstellen
+    elements.append(title)
+    for nutzer, assi in dic_history.items():
+        elements.append(headline_nutzer)
+        p = Paragraph(nutzer, styles['NewStyle'])
+        elements.append(p)
+        # Einen Abstand hinzufügen (optional)
+        elements.append(Spacer(1, 2*mm))
+        elements.append(headline_assi)
+        p = Paragraph(assi, styles['NewStyle'])
+        elements.append(p)
+        # Einen Abstand hinzufügen (optional)
+        elements.append(Spacer(1, 8*mm))
+        #Trennlinie
+        elements.append(Paragraph('_' * 100, line_style)) 
+        # Einen Abstand hinzufügen (optional)
+        elements.append(Spacer(1, 8*mm))
+
+        #Für später, um bilder einzufügen
+        # Fügen Sie andere Flowables wie Bilder oder Tabellen hinzu
+        #image = Image('path/to/your/image.png', width=10*cm, height=5*cm)
+        #elements.append(image)
+        #table = Table([['Cell 1', 'Cell 2'], ['Cell 3', 'Cell 4']])
+        #elements.append(table)
+
+
+    # Generiere das PDF-Dokument
+    doc = CustomDocTemplate(file_path_download, pagesize=paper_size)
+    #on_each_page ist eine callback Methode, die auf jeder neuen PDF Seite ausgeführt wird
+    doc.onPage = on_each_page
+    doc.build(elements)
+    
 
 ###################################################
 #zur Zeit nicht im Gebrauch
@@ -771,4 +847,20 @@ def is_stop_word_or_prefix(s: str, stop_words: list) -> bool:
                 return True
     return False
 
+##########################################
+# Klasse, die die SimpleDocTemplate überschreibt (für PDF Generierung)
+# war nötig, da SimpleDocTemplate die on_each_page nicht ausgeführt hat - warum auch immer ...
+##########################################
+class CustomDocTemplate(SimpleDocTemplate):
+    def handle_pageBegin(self):
+        # Sorgt dafür, dass die Standard-Page-Begin-Logik ausgeführt wird
+        self._handle_pageBegin()
+        # Jetzt können Sie das Canvas-Objekt über self.canv sicher verwenden
+        self.canv.saveState()
+        self.canv.setFont('Helvetica', 10)
+        current_date = datetime.now().strftime("%Y-%m-%d")
+        # Passen Sie hier die Positionierung an Ihre Bedürfnisse an
+        self.canv.drawRightString(550, 800, current_date)  # Position anpassen
+        self.canv.restoreState()
+