Update app.py

app.py

@@ -22,15 +22,12 @@ MODELS = {
         "key_env": "DEEPSEEK_API_KEY",
         "response_format": None,
     },
-    "Llama 4 Mavericks": {
-        "api_url": "https://openrouter.ai/api/v1/chat/completions",
-        "model": "meta-llama/llama-4-maverick:free",
-        "key_env": "OPENROUTER_API_KEY",
-        "response_format": {"type": "json_object"},
-        "extra_headers": {
-            "HTTP-Referer": "https://huggingface.co",
-            "X-Title": "Invoice Extractor",
-        },
+    "OpenAI GPT-4.1": {
+        "api_url": "https://api.openai.com/v1/chat/completions",
+        "model": "gpt-4-1106-preview",
+        "key_env": "OPENAI_API_KEY",
+        "response_format": None,  # OpenAI doesn't require explicit json_object response_format for extraction
+        "extra_headers": {},
     },
     "Mistral Small": {
         "api_url": "https://openrouter.ai/api/v1/chat/completions",
@@ -146,24 +143,16 @@ def extract_invoice_info(model_choice, text):
     if not data:
         return None
 
-    # normalize + supplier fallback
-    if model_choice in ("Llama 4 Mavericks","Mistral Small"):
-        hdr = data.setdefault("invoice_header", {})
-        for k in ("invoice_number","invoice_date","po_number","invoice_value","supplier_name","customer_name"):
-            hdr.setdefault(k, None)
-        if not hdr.get("supplier_name"):
-            hdr["supplier_name"] = fallback_supplier(text)
-        items = data.setdefault("line_items", [])
-        for itm in items:
-            for k in ("item_number","description","quantity","unit_price","total_price"):
-                itm.setdefault(k, None)
-    else:
-        for k in ("invoice_number","invoice_date","po_number","invoice_value"):
-            data.setdefault(k, None)
-        items = data.setdefault("line_items", [])
-        for itm in items:
-            for k in ("description","quantity","unit_price","total_price"):
-                itm.setdefault(k, None)
+    # normalize fields (no special fallback needed for GPT-4.1, just like Mistral)
+    hdr = data.setdefault("invoice_header", {})
+    for k in ("invoice_number","invoice_date","po_number","invoice_value","supplier_name","customer_name"):
+        hdr.setdefault(k, None)
+    if not hdr.get("supplier_name"):
+        hdr["supplier_name"] = fallback_supplier(text)
+    items = data.setdefault("line_items", [])
+    for itm in items:
+        for k in ("item_number","description","quantity","unit_price","total_price"):
+            itm.setdefault(k, None)
 
     return data
 
@@ -190,18 +179,12 @@ with tab2:
         info = extract_invoice_info(mdl, txt)
         if info:
             st.success("Extraction Complete")
-            if mdl in ("Llama 4 Mavericks","Mistral Small"):
-                h=info["invoice_header"]
-                c1,c2,c3 = st.columns(3)
-                c1.metric("Invoice #", h["invoice_number"]); c1.metric("Supplier", h["supplier_name"])
-                c2.metric("Date", h["invoice_date"]);    c2.metric("Customer", h["customer_name"])
-                c3.metric("PO #", h["po_number"]);        c3.metric("Total", h["invoice_value"])
-                st.subheader("Line Items"); st.table(info["line_items"])
-            else:
-                c1,c2 = st.columns(2)
-                c1.metric("Invoice #", info["invoice_number"]); c1.metric("PO #", info["po_number"])
-                c2.metric("Date", info["invoice_date"]);        c2.metric("Value", info["invoice_value"])
-                st.subheader("Line Items"); st.table(info["line_items"])
+            h=info["invoice_header"]
+            c1,c2,c3 = st.columns(3)
+            c1.metric("Invoice #", h["invoice_number"]); c1.metric("Supplier", h["supplier_name"])
+            c2.metric("Date", h["invoice_date"]);    c2.metric("Customer", h["customer_name"])
+            c3.metric("PO #", h["po_number"]);        c3.metric("Total", h["invoice_value"])
+            st.subheader("Line Items"); st.table(info["line_items"])
 
     if "last_api" in st.session_state:
         with st.expander("Debug"):