Update app.py

app.py

@@ -2,20 +2,24 @@ from transformers import TrOCRProcessor, VisionEncoderDecoderModel
 from PIL import Image
 import pdf2image
 import gradio as gr
+import base64
+import io
 import torch
 
-# ✅ Używamy mniejszej wersji modelu dla szybszego działania
+# ✅ Używamy lżejszego modelu dla większej wydajności
 MODEL_NAME = "microsoft/trocr-base-stage1"
-
-# ✅ Ładujemy model i procesor
 processor = TrOCRProcessor.from_pretrained(MODEL_NAME)
 model = VisionEncoderDecoderModel.from_pretrained(MODEL_NAME)
 
-# ✅ Funkcja do przetwarzania faktur PDF
-def process_invoice(pdf_file):
+# ✅ Funkcja przetwarzania faktury (odbiera Base64 PDF)
+def process_invoice(pdf_base64):
     try:
+        # ✅ Dekodowanie Base64 na plik PDF
+        pdf_bytes = base64.b64decode(pdf_base64)
+        pdf_stream = io.BytesIO(pdf_bytes)
+
         # ✅ Konwersja PDF do obrazu
-        images = pdf2image.convert_from_bytes(pdf_file.read())
+        images = pdf2image.convert_from_bytes(pdf_stream.read())
 
         # ✅ Przetwarzanie pierwszej strony faktury
         image = images[0].convert("RGB")
@@ -29,13 +33,13 @@ def process_invoice(pdf_file):
     except Exception as e:
         return f"Błąd przetwarzania: {str(e)}"
 
-# ✅ Poprawne użycie `gr.File(type="binary")`
+# ✅ API Gradio (teraz akceptuje Base64 jako wejście)
 interface = gr.Interface(
     fn=process_invoice,
-    inputs=gr.File(type="binary"),
+    inputs=gr.Textbox(label="PDF w Base64"),
     outputs="text",
     title="AI Invoice Processor",
-    description="Wgraj fakturę PDF, a AI wyodrębni dane tekstowe.",
+    description="Prześlij fakturę jako Base64, a AI wyodrębni dane tekstowe.",
 )
 
 # ✅ Uruchomienie aplikacji