Spaces:
Running
Running
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 |
-
|
|
|
|
| 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:
|