Upload folder using huggingface_hub

constants.py

@@ -1,2 +1,2 @@
-GEMINI_API_KEY = "AIzaSyBwk94xRhPOIkvO0E3pYhXQ7Rrk5my5IyY"
-OPENAI_API_KEY = "sk-proj-YOl2xepEsNppWm3xLshlT3BlbkFJL04qQgahGxFcFGEClnQK"
+GEMINI_API_KEY = ""
+OPENAI_API_KEY = ""

extract_images/services.py

@@ -9,7 +9,7 @@ import openai
 import requests
 import os
 
-from constants import GEMINI_API_KEY, OPENAI_API_KEY
+# from constants import GEMINI_API_KEY, OPENAI_API_KEY
 from utils import (
     draw_boxes,
     pdf_to_images,
@@ -67,7 +67,8 @@ def extract_images_pdfplumber(pdf_file):
 
 
 def extract_images_gemini(model, pdf_file):
-    genai.configure(api_key=GEMINI_API_KEY)
+    gemini_api_key = os.getenv("GEMINI_API_KEY")
+    genai.configure(api_key=gemini_api_key)
     gemini_model = genai.GenerativeModel(model)
     prompt = f"Extract the bounding boxes of all the images present in this page. Return the bounding boxes as list of lists. Do not include anyother text or symbols in the output"
 

extract_tables/services.py

@@ -9,9 +9,10 @@ import ast
 import google.generativeai as genai
 import openai
 import requests
+import os
 
 
-from constants import GEMINI_API_KEY, OPENAI_API_KEY
+# from constants import GEMINI_API_KEY, OPENAI_API_KEY
 from utils import (
     draw_boxes,
     pdf_to_images,
@@ -109,7 +110,9 @@ def extract_tables_img2table(pdf_file):
 
 
 def extract_tables_gemini(model, pdf_file):
-    genai.configure(api_key=GEMINI_API_KEY)
+    gemini_api_key = os.getenv("GEMINI_API_KEY")
+    genai.configure(api_key=gemini_api_key)
+
     gemini_model = genai.GenerativeModel(model)
     prompt = f"Extract the bounding boxes of all the tables present in this image. Return the bounding boxes as list of lists. Do not include anyother text or symbols in the output"
 
@@ -136,7 +139,8 @@ def extract_tables_gemini(model, pdf_file):
 
 
 def extract_tables_gpt(model, pdf_file):
-    openai.api_key = OPENAI_API_KEY
+    openai_api_key = os.getenv("OPENAI_API_KEY")
+    openai.api_key = openai_api_key
     image_media_type = "image/png"
 
     pdf_path = "extract_images/input_docs/uploaded_pdf.pdf"