Update app.py

app.py

@@ -8,10 +8,11 @@ debug_mode = True
 
 # Initialize OpenAI client
 client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))  # Replace with your actual API key
+data_extractor_url = "https://data-extractor-67qj89pa0-sonikas-projects-9936eaad.vercel.app/"
 
-def extract_data_from_product_image(image_links):
+def extract_data_from_product_image(image_links, data_extractor_url):
     #Send product label image url to data extractor
-    url = "https://data-extractor-67qj89pa0-sonikas-projects-9936eaad.vercel.app/extract"
+    url = data_extractor_url + "extract"
     data = {
         "image_links": image_links
     }
@@ -28,9 +29,9 @@ def extract_data_from_product_image(image_links):
         print(f"Error occurred: {e}")
         return {}
 
-def get_product_data_from_db(product_name):
+def get_product_data_from_db(product_name, data_extractor_url):
     #Extract data for a product by calling data extractor's API : https://data-extractor-3cn8or2tc-sonikas-projects-9936eaad.vercel.app/
-    url = "https://data-extractor-3cn8or2tc-sonikas-projects-9936eaad.vercel.app/product"
+    url = data_extractor_url + "product"
     params = {"name": product_name}
     
     try: