ES FUNKTIONIERT

lib.py

@@ -5,6 +5,8 @@ import psycopg2
 import pymongo
 import pytesseract
 import ollama
+import io
+from PIL import Image
 from aiohttp.web_response import json_response
 
 import ollama_conn
@@ -47,6 +49,21 @@ def extract_image_content(path):
     img = cv2.imread(path)
     return pytesseract.image_to_string(img)
 
+def extract_text_and_pictures(pdf_datei):
+    pytesseract.pytesseract.tesseract_cmd = r'C:\Program Files\Tesseract-OCR\tesseract.exe'
+    doc = pymupdf.open(pdf_datei)
+    text = ""
+    for seite in doc:
+        text += seite.get_text()
+        for bild in seite.get_images(full=True):
+            xref = bild[0]
+            basis_bild = doc.extract_image(xref)
+            bild_daten = basis_bild["image"]
+            bild_objekt = Image.open(io.BytesIO(bild_daten))
+            bild_text = pytesseract.image_to_string(bild_objekt, lang="deu")
+            text += " " + bild_text
+    return text
+
 def prompt_embedding(prompt):
     embedding = ollama.embeddings(model="mxbai-embed-large", prompt=prompt)