Update app.py

app.py

@@ -3,7 +3,7 @@ import requests
 import time
 import os
 
-# CONFIG — Edit if you want to move to environment variables instead of hard-coding
+# CONFIG
 UNSTRACT_API_KEY = os.getenv("UNSTRACT_API_KEY", "pktmL5lfqlVv7IWW_MYhdXRl399GA1n8vaLktHefxVY")
 BASE_URL = "https://llmwhisperer-api.us-central.unstract.com/api/v2"
 
@@ -12,10 +12,16 @@ def upload_pdf_to_unstract(pdf_file):
     headers = {
         "unstract-key": UNSTRACT_API_KEY,
     }
+    # Always reset file pointer and use "application/pdf"
+    pdf_file.seek(0)
+    file_bytes = pdf_file.read()
     files = {
-        "file": (pdf_file.name, pdf_file, "application/pdf"),
+        "file": (
+            pdf_file.name if pdf_file.name.lower().endswith(".pdf") else "invoice.pdf",
+            file_bytes,
+            "application/pdf"
+        ),
     }
-    # 'mode': 'form' is not required, default is 'native_text'
     with st.spinner("Uploading and starting OCR..."):
         resp = requests.post(url, headers=headers, files=files)
     if resp.status_code not in (200, 202):