Upload folder using huggingface_hub

app.py

@@ -926,9 +926,9 @@ class BillRequest(BaseModel):
 
 class BillResponse(BaseModel):
     is_success: bool
-    error: Optional[str] = None
-    data: Dict[str, Any]
     token_usage: Dict[str, int]
+    data: Dict[str, Any]
+    
 
 @app.post("/extract-bill-data", response_model=BillResponse)
 async def extract_bill_data(payload: BillRequest):
@@ -944,9 +944,9 @@ async def extract_bill_data(payload: BillRequest):
         except Exception as e:
             return BillResponse(
                 is_success=False,
-                error=f"Local file read failed: {e}",
-                data={"pagewise_line_items": [], "total_item_count": 0},
                 token_usage={"total_tokens": 0, "input_tokens": 0, "output_tokens": 0},
+                data={"pagewise_line_items": [], "total_item_count": 0},
+                
             )
     else:
         try:
@@ -955,25 +955,25 @@ async def extract_bill_data(payload: BillRequest):
             if resp.status_code != 200:
                 return BillResponse(
                     is_success=False,
-                    error=f"Download failed (status={resp.status_code})",
-                    data={"pagewise_line_items": [], "total_item_count": 0},
                     token_usage={"total_tokens": 0, "input_tokens": 0, "output_tokens": 0},
+                    data={"pagewise_line_items": [], "total_item_count": 0},
+                    
                 )
             file_bytes = resp.content
         except Exception as e:
             return BillResponse(
                 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},
+                data={"pagewise_line_items": [], "total_item_count": 0},
+                
             )
     
     if not file_bytes:
         return BillResponse(
             is_success=False,
-            error="No file bytes",
-            data={"pagewise_line_items": [], "total_item_count": 0},
             token_usage={"total_tokens": 0, "input_tokens": 0, "output_tokens": 0},
+            data={"pagewise_line_items": [], "total_item_count": 0},
+            
         )
     
     logger.info(f"Processing with engine: {OCR_ENGINE}")
@@ -991,11 +991,12 @@ async def extract_bill_data(payload: BillRequest):
     
     return BillResponse(
         is_success=True,
+        token_usage={"total_tokens": 0, "input_tokens": 0, "output_tokens": 0},
         data={
             "pagewise_line_items": pages_dict,
             "total_item_count": total_items,
         },
-        token_usage={"total_tokens": 0, "input_tokens": 0, "output_tokens": 0},
+        
     )
 
 @app.get("/")