userIdc2024 commited on
Commit
1c35399
·
verified ·
1 Parent(s): 892a31c

Update generator_function/image_function.py

Browse files
generator_function/image_function.py CHANGED
@@ -6,6 +6,8 @@ from urllib.parse import urlparse
6
  from functools import lru_cache
7
  import os, base64, logging
8
  from openai import OpenAI
 
 
9
 
10
  from database.operations import start_job, finish_job
11
 
@@ -157,7 +159,8 @@ def _process_one(args: Tuple[str, str, str, int]) -> Dict[str, Any]:
157
  b = _fetch(src)
158
  if not b:
159
  out["error"] = "Fetch failed"; return out
160
- r2 = _upload_to_r2(b)
 
161
  if r2:
162
  out["r2_url"] = r2; out["success"] = True
163
  else:
 
6
  from functools import lru_cache
7
  import os, base64, logging
8
  from openai import OpenAI
9
+ from helpers_function.helper_meta_data import meta_data_helper_function
10
+
11
 
12
  from database.operations import start_job, finish_job
13
 
 
159
  b = _fetch(src)
160
  if not b:
161
  out["error"] = "Fetch failed"; return out
162
+ image_with_metadata = meta_data_helper_function(b)
163
+ r2 = _upload_to_r2(image_with_metadata)
164
  if r2:
165
  out["r2_url"] = r2; out["success"] = True
166
  else: