Upload folder using huggingface_hub

app.py

@@ -599,49 +599,49 @@ async def extract_bill_data(payload: BillRequest):
     doc_url = payload.document
     file_bytes = None
 
-# ---------------------------
-# 1. LOCAL FILE SUPPORT
-# ---------------------------
-if doc_url.startswith("file://"):
-    local_path = doc_url.replace("file://", "")
-    try:
-        with open(local_path, "rb") as f:
-            file_bytes = f.read()
-    except Exception as e:
-        return {
-            "is_success": False,
-            "error": f"Local file read error: {e}",
-            "data": {"pagewise_line_items": [], "total_item_count": 0},
-            "token_usage": {"total_tokens": 0, "input_tokens": 0, "output_tokens": 0}
-        }
-
-# ---------------------------
-# 2. REMOTE URL DOWNLOAD
-# ---------------------------
-else:
-    try:
-        headers = {"User-Agent": "Mozilla/5.0"}
-        resp = requests.get(doc_url, headers=headers, timeout=30)
-        if resp.status_code != 200:
-            raise RuntimeError(f"Download failed status={resp.status_code}")
-        file_bytes = resp.content
-    except Exception as e:
+    # ---------------------------
+    # 1. LOCAL FILE SUPPORT
+    # ---------------------------
+    if doc_url.startswith("file://"):
+        local_path = doc_url.replace("file://", "")
+        try:
+            with open(local_path, "rb") as f:
+                file_bytes = f.read()
+        except Exception as e:
+            return {
+                "is_success": False,
+                "error": f"Local file read error: {e}",
+                "data": {"pagewise_line_items": [], "total_item_count": 0},
+                "token_usage": {"total_tokens": 0, "input_tokens": 0, "output_tokens": 0}
+            }
+
+    # ---------------------------
+    # 2. REMOTE URL DOWNLOAD
+    # ---------------------------
+    else:
+        try:
+            headers = {"User-Agent": "Mozilla/5.0"}
+            resp = requests.get(doc_url, headers=headers, timeout=30)
+            if resp.status_code != 200:
+                raise RuntimeError(f"Download failed status={resp.status_code}")
+            file_bytes = resp.content
+        except Exception as e:
+            return {
+                "is_success": False,
+                "error": f"HTTP error: {e}",
+                "data": {"pagewise_line_items": [], "total_item_count": 0},
+                "token_usage": {"total_tokens": 0, "input_tokens": 0, "output_tokens": 0}
+            }
+
+    # Final safety check
+    if not file_bytes:
         return {
             "is_success": False,
-            "error": f"HTTP error: {e}",
+            "error": "No file bytes found.",
             "data": {"pagewise_line_items": [], "total_item_count": 0},
             "token_usage": {"total_tokens": 0, "input_tokens": 0, "output_tokens": 0}
         }
 
-# Safety check
-if not file_bytes:
-    return {
-        "is_success": False,
-        "error": "No file bytes found.",
-        "data": {"pagewise_line_items": [], "total_item_count": 0},
-        "token_usage": {"total_tokens": 0, "input_tokens": 0, "output_tokens": 0}
-    }
-
     images = []
     clean_url = doc_url.split("?", 1)[0].lower()
     try: