Update app.py

app.py

@@ -19,4 +19,23 @@ def extract_seller(pdf_file):
         entities = extractor(chunk)
 
         for entity in entities:
-            if "ORG" in entity["entity_group"]:  #
+            if "ORG" in entity["entity_group"]:  # Szukamy nazw organizacji
+                seller_name = entity["word"]
+                break  # Pobieramy pierwszą wykrytą firmę jako sprzedawcę
+
+        if seller_name:  # Jeśli znaleziono sprzedawcę, przerywamy pętlę
+            break
+
+    return {"Sprzedawca": seller_name if seller_name else "Nie znaleziono"}
+
+# Interfejs użytkownika w Hugging Face Spaces
+iface = gr.Interface(
+    fn=extract_seller,
+    inputs=gr.File(label="Wybierz plik PDF"),
+    outputs="json",
+    title="Ekstrakcja Sprzedawcy z Faktury",
+    description="Prześlij plik PDF, aby wydobyć nazwę sprzedawcy."
+)
+
+if __name__ == "__main__":
+    iface.launch()