Upload folder using huggingface_hub

extract_images/services.py

@@ -7,6 +7,7 @@ import google.generativeai as genai
 from PIL import Image, ImageDraw
 import openai
 import requests
+import os
 
 from constants import GEMINI_API_KEY, OPENAI_API_KEY
 from utils import (
@@ -41,6 +42,7 @@ def extract_images_pdfplumber(pdf_file):
         f.write(pdf_file)
 
     images = []
+    output_dir = "extract_tables/table_outputs"
     pdf_obj = pdfplumber.open(pdf_path)
     for page_idx, page in enumerate(pdf_obj.pages):
         page_bbox = []
@@ -55,8 +57,8 @@ def extract_images_pdfplumber(pdf_file):
             page_bbox.append(image_bbox)
             cropped_page = page.crop(image_bbox)
             image_obj = cropped_page.to_image(resolution=400)
-            image_path = (
-                f"extract_images/image_outputs/image-{page_idx}-{image_idx}.png"
+            image_path = os.path.join(
+                output_dir, f"image-{page_idx + 1}-{image_idx}.png"
             )
             image_obj.save(image_path)
             image = Image.open(image_path)