Update app.py

app.py

@@ -26,7 +26,7 @@ MODELS = {
         "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
+        "response_format": None,
         "extra_headers": {},
     },
     "Mistral Small": {
@@ -143,7 +143,16 @@ def extract_invoice_info(model_choice, text):
     if not data:
         return None
 
-    # normalize fields (no special fallback needed for GPT-4.1, just like Mistral)
+    # DeepSeek models: flat format
+    if model_choice.startswith("DeepSeek"):
+        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)
+        return data
+    # Other models (OpenAI GPT-4.1, Mistral): nested format
     hdr = data.setdefault("invoice_header", {})
     for k in ("invoice_number","invoice_date","po_number","invoice_value","supplier_name","customer_name"):
         hdr.setdefault(k, None)
@@ -179,12 +188,25 @@ with tab2:
         info = extract_invoice_info(mdl, txt)
         if info:
             st.success("Extraction Complete")
-            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 mdl.startswith("DeepSeek"):
+                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"])
+            else:
+                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"):