Test

main.py

@@ -84,7 +84,7 @@ class LLMClient:
 
 
 # ==========================================
-# 2. METADATEN SCHREIBSCHEMA
+# 2. METADATEN SCHREIBSCHEMA (ERWEITERT FÜR VORLAUFTEXTE)
 # ==========================================
 class InvoiceMetadata(BaseModel):
     invoice_number: str = Field(description="Die Rechnungsnummer nach 'Nummer:' (z.B. R26-219)")
@@ -97,10 +97,18 @@ class InvoiceMetadata(BaseModel):
     recipient_city: str = Field(description="Stadt des Kunden")
     recipient_email: str = Field(description="E-Mail-Adresse des Kunden (LEER lassen, wenn nicht explizit im Text!)")
     vat_rate_percent: float = Field(description="Der primäre Umsatzsteuersatz in Prozent (z.B. 19.0)")
+    
+    # Neue strukturierte Felder für jegliche Daten VOR Position 1
+    delivery_note_id: str = Field(default="", description="Gefundene Lieferscheinnummer (z.B. LS26-035) VOR Position 1. Sonst leer lassen.")
+    order_id: str = Field(default="", description="Gefundene Nummer der Auftragsbestätigung oder Bestellung (z.B. AB26-080) VOR Position 1. Sonst leer lassen.")
+    contract_id: str = Field(default="", description="Gefundene Nummer von Vergabeangebot, Vertrag oder Ausschreibung (z.B. AN26-130) VOR Position 1. Sonst leer lassen.")
+    service_start_date: str = Field(default="", description="Startdatum eines explizit genannten Einsatz- oder Leistungszeitraums VOR Position 1 im Format YYYY-MM-DD. Sonst leer.")
+    service_end_date: str = Field(default="", description="Enddatum eines explizit genannten Einsatz- oder Leistungszeitraums VOR Position 1 im Format YYYY-MM-DD. Sonst leer.")
+    invoice_note: str = Field(default="", description="Sämtlicher unstrukturierter Fließtext, Beschreibungen, Abnahmekriterien (Acceptance Criteria) oder Projektklauseln, die sich VOR Position 1 befinden. Vollständig mit Zeilenumbrüchen sichern.")
 
 
 # ==========================================
-# 3. MATHEMATISCHE FUNKTIONEN
+# 3. MATHEMATISCHE & PARSING-FUNKTIONEN
 # ==========================================
 def kaufmaennisches_runden(betrag, nachkommastellen=2):
     if not betrag:
@@ -156,8 +164,11 @@ def extract_invoice_items_pure_python(text):
                     break
                 if any(prev_lower.startswith(w) for w in ["zwischensumme", "endbetrag", "gesamtbetrag", "umsatzsteuer"]):
                     break
+                
+                # REPARATUR: Wenn eine reine Positionsnummer (z.B. "1") erreicht wird, MUSS abgebrochen werden,
+                # um nicht in den globalen Kopf- und Vorlauftext darüber hineinzulaufen!
                 if prev_line.isdigit() and len(prev_line) <= 2:
-                    continue
+                    break
                     
                 desc_lines.insert(0, prev_line)
                 
@@ -180,6 +191,8 @@ def map_unece_unit_code(unit_str):
     return "C62"
 
 def parse_to_strict_xml_date(date_str):
+    if not date_str or str(date_str).strip() == "":
+        return ""
     clean_str = str(date_str).strip()
     for fmt in ("%Y-%m-%d", "%d.%m.%Y", "%d-%m-%Y"):
         try: return datetime.strptime(clean_str, fmt).strftime("%Y-%m-%d")
@@ -191,14 +204,13 @@ def generate_compliant_xrechnung(data):
     CBC = "urn:oasis:names:specification:ubl:schema:xsd:CommonBasicComponents-2"
     CAC = "urn:oasis:names:specification:ubl:schema:xsd:CommonAggregateComponents-2"
     
-    # Namespaces explizit registrieren
     ET.register_namespace('ubl', INVOICE_NS)
     ET.register_namespace('cbc', CBC)
     ET.register_namespace('cac', CAC)
     
     root = ET.Element(f"{{{INVOICE_NS}}}Invoice")
     
-    # Offizielle Profil-IDs für XRechnung 3.0 nach Kosit-Vorgabe
+    # Kopfdaten Definitionen nach Kosit-Vorgabe (XRechnung 3.0)
     ET.SubElement(root, f"{{{CBC}}}CustomizationID").text = "urn:cen.eu:en16931:2017#compliant#urn:xeinkauf.de:kosit:xrechnung_3.0"
     ET.SubElement(root, f"{{{CBC}}}ProfileID").text = "urn:fdc:peppol.eu:2017:poacc:billing:01:1.0"
     
@@ -213,9 +225,35 @@ def generate_compliant_xrechnung(data):
     ET.SubElement(root, f"{{{CBC}}}DueDate").text = computed_due
     
     ET.SubElement(root, f"{{{CBC}}}InvoiceTypeCode").text = "380"
+    
+    # Globales Freitextfeld BT-22 (Rechnungserläuterung) für unstrukturierte Vorlauftexte / Vodafone-Klauseln
+    if data.get("invoice_note") and str(data["invoice_note"]).strip() != "":
+        ET.SubElement(root, f"{{{CBC}}}Note").text = str(data["invoice_note"]).strip()
+        
     ET.SubElement(root, f"{{{CBC}}}DocumentCurrencyCode").text = "EUR"
     ET.SubElement(root, f"{{{CBC}}}BuyerReference").text = str(data["buyer_reference"])
     
+    # Globaler Leistungszeitraum (BG-14)
+    if data.get("service_start_date") and data.get("service_end_date"):
+        period_elem = ET.SubElement(root, f"{{{CAC}}}InvoicePeriod")
+        ET.SubElement(period_elem, f"{{{CBC}}}StartDate").text = parse_to_strict_xml_date(data["service_start_date"])
+        ET.SubElement(period_elem, f"{{{CBC}}}EndDate").text = parse_to_strict_xml_date(data["service_end_date"])
+        
+    # Auftragsbestätigung / Bestellungs-Referenz (BT-13)
+    if data.get("order_id") and str(data["order_id"]).strip() != "":
+        order_ref = ET.SubElement(root, f"{{{CAC}}}OrderReference")
+        ET.SubElement(order_ref, f"{{{CBC}}}ID").text = str(data["order_id"]).strip()
+        
+    # Lieferschein-Referenz (BT-16)
+    if data.get("delivery_note_id") and str(data["delivery_note_id"]).strip() != "":
+        despatch_ref = ET.SubElement(root, f"{{{CAC}}}DespatchDocumentReference")
+        ET.SubElement(despatch_ref, f"{{{CBC}}}ID").text = str(data["delivery_note_id"]).strip()
+        
+    # Vertrags- / Ausschreibungsreferenz (BT-12)
+    if data.get("contract_id") and str(data["contract_id"]).strip() != "":
+        contract_ref = ET.SubElement(root, f"{{{CAC}}}ContractDocumentReference")
+        ET.SubElement(contract_ref, f"{{{CBC}}}ID").text = str(data["contract_id"]).strip()
+
     vat_rate = float(data['vat_rate_percent'])
     vat_category = "S" if vat_rate > 0.0 else "E"
 
@@ -241,7 +279,6 @@ def generate_compliant_xrechnung(data):
     legal_supp = ET.SubElement(party_supp, f"{{{CAC}}}PartyLegalEntity")
     ET.SubElement(legal_supp, f"{{{CBC}}}RegistrationName").text = "scitech it solutions GmbH"
     
-    # KORREKTUR: Saubere, valide Erzeugung des Contact-Blocks mit cbc-Namespaces
     contact_supp = ET.SubElement(party_supp, f"{{{CAC}}}Contact")
     ET.SubElement(contact_supp, f"{{{CBC}}}Name").text = "Geschäftsführung"
     ET.SubElement(contact_supp, f"{{{CBC}}}Telephone").text = "+49 3302 4940537"
@@ -303,7 +340,6 @@ def generate_compliant_xrechnung(data):
         item_elem = ET.SubElement(line, f"{{{CAC}}}Item")
         ET.SubElement(item_elem, f"{{{CBC}}}Name").text = str(item["description"])
         
-        # Saubere, UBL-konforme Schachtelung der Steuerkategorie innerhalb des Items
         item_tax = ET.SubElement(item_elem, f"{{{CAC}}}ClassifiedTaxCategory")
         ET.SubElement(item_tax, f"{{{CBC}}}ID").text = vat_category
         ET.SubElement(item_tax, f"{{{CBC}}}Percent").text = f"{vat_rate:.2f}"
@@ -357,7 +393,6 @@ with col2:
         if st.button("Daten extrahieren", width="stretch"):
             with st.spinner("Extrahiere Daten über vLLM-API..."):
                 
-                # STUFE 1: Stammdaten via API mit Schema-Zwang
                 schema_dict = InvoiceMetadata.model_json_schema()
                 
                 sys_prompt = (
@@ -373,7 +408,16 @@ with col2:
                     "- recipient_zip\n"
                     "- recipient_city\n"
                     "- recipient_email\n"
-                    "- vat_rate_percent (als Zahl, z.B. 19.0)\n\n"
+                    "- vat_rate_percent (als Zahl, z.B. 19.0)\n"
+                    "- delivery_note_id (Lieferscheinnummer vor Position 1, falls vorhanden)\n"
+                    "- order_id (Auftragsbestätigung/Bestellnummer vor Position 1, falls vorhanden)\n"
+                    "- contract_id (Vergabeangebot/Vertragsreferenz vor Position 1, falls vorhanden)\n"
+                    "- service_start_date (Leistungszeitraum Start vor Position 1, Format YYYY-MM-DD)\n"
+                    "- service_end_date (Leistungszeitraum Ende vor Position 1, Format YYYY-MM-DD)\n"
+                    "- invoice_note (Sämtlicher unstrukturierter Freitext, Projektbeschreibungen oder Abnahmekriterien vor Position 1)\n\n"
+                    "WICHTIGE REGEL: Analysiere den Text direkt VOR der ersten Tabellenposition '1'. "
+                    "Sortiere alle dort vorlaufenden Zeilen vollständig in die dafür vorgesehenen ID-, Datums- und Freitext-Felder ein. "
+                    "Schreibe diese vorlaufenden Metadaten NIEMALS in die Beschreibung der ersten Position hinein!\n\n"
                     "Antworte ausschließlich mit dem reinen JSON-Objekt. Schreibe keinen anderen Text."
                 )
                 prompt_meta = f"Rechnungstext:\n{pdf_text}"
@@ -405,17 +449,40 @@ with col2:
                 if final_data.get("recipient_email") == "office@scitech-gmbh.de":
                     final_data["recipient_email"] = ""
                 
-                # STUFE 2: Zeilenbasiertes Python-Parsing für die Tabelle
+                # Zeilenbasiertes Python-Parsing für die Tabelle (stoppt nun korrekt dank 'break' bei Positionsnummern)
                 final_data["items"] = extract_invoice_items_pure_python(pdf_text)
                 st.session_state.extracted_data = final_data
 
     if st.session_state.extracted_data:
         data = st.session_state.extracted_data
+        
+        # Standard Rechnungsdaten
         inv_num = st.text_input("Rechnungsnummer", value=data.get("invoice_number", ""))
         inv_date = st.text_input("Rechnungsdatum (YYYY-MM-DD)", value=data.get("invoice_date", ""))
         inv_delivery = st.text_input("Lieferdatum (YYYY-MM-DD)", value=data.get("delivery_date", ""))
         buyer_ref = st.text_input("Ihr Auftrag / Referenz", value=data.get("buyer_reference", ""))
         
+        # NEU: UI-Eingabefelder für die Daten vor Position 1
+        st.markdown("---")
+        st.markdown("🔗 **Globale Referenzen & Vorlauftexte (Vor Position 1):**")
+        c_ref1, c_ref2, c_ref3 = st.columns(3)
+        with c_ref1:
+            dn_id = st.text_input("Lieferscheinnummer (BT-16)", value=data.get("delivery_note_id", ""))
+        with c_ref2:
+            o_id = st.text_input("Auftragsbestätigung / Bestellung (BT-13)", value=data.get("order_id", ""))
+        with c_ref3:
+            c_id = st.text_input("Vertrag / Vergabeangebot (BT-12)", value=data.get("contract_id", ""))
+            
+        c_p1, c_p2 = st.columns(2)
+        with c_p1:
+            s_start = st.text_input("Leistungszeitraum Start (YYYY-MM-DD)", value=data.get("service_start_date", ""))
+        with c_p2:
+            s_end = st.text_input("Leistungszeitraum Ende (YYYY-MM-DD)", value=data.get("service_end_date", ""))
+            
+        i_note = st.text_area("Globale Rechnungserläuterung / Freitext (BT-22)", value=data.get("invoice_note", ""), height=150)
+        st.markdown("---")
+        
+        # Adressdaten
         c_s, c_r = st.columns(2)
         with c_s:
             st.markdown("**Absender (Verkäufer):**")
@@ -459,11 +526,19 @@ with col2:
                 "buyer_reference": buyer_ref, "recipient_name": r_name, "recipient_email": r_email,
                 "recipient_street": r_street, "recipient_zip": r_zip, "recipient_city": r_city,
                 "items": edited_items, "total_net_amount": computed_net_amount, "total_vat_amount": computed_vat_amount,
-                "vat_rate_percent": vat_rate, "total_gross_amount": computed_gross_amount, "iban": "DE06120700240456207000"
+                "vat_rate_percent": vat_rate, "total_gross_amount": computed_gross_amount, "iban": "DE06120700240456207000",
+                # Neue Felder für den XML-Builder anhängen
+                "delivery_note_id": dn_id,
+                "order_id": o_id,
+                "contract_id": c_id,
+                "service_start_date": s_start,
+                "service_end_date": s_end,
+                "invoice_note": i_note
             }
             st.download_button(label="📥 Download XML", data=generate_compliant_xrechnung(final_invoice_dict), file_name=f"XRechnung_{inv_num}.xml", mime="text/xml")
+
 # ==========================================
-# DYNAMISCHER FOOTER (RECHTLICHE HINWEISE & SYSTEM-INFO)
+# DYNAMISCHER FOOTER
 # ==========================================
 st.markdown("---")
 st.caption(