🚑 Encode to base64

argilla-io · Dec 4, 2024 · b0dcb76 · b0dcb76
1 parent 9cb2dde
commit b0dcb76
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/argilla-frontend/components/features/annotation/progress/share/useShareViewModel.ts b/argilla-frontend/components/features/annotation/progress/share/useShareViewModel.ts
@@ -26,7 +26,7 @@ export const useShareViewModel = () => {
 
     const url = new URL(`${window.location.origin}/share`);
     const params = new URLSearchParams("");
-    params.set("share_image", encodeURI(createImageLink()));
+    params.set("share_image_base64", btoa(createImageLink()));
     params.set("dataset_name", dataset.name);
     params.set("dataset_id", dataset.id);
 

diff --git a/argilla-server/src/argilla_server/_app.py b/argilla-server/src/argilla_server/_app.py
@@ -12,6 +12,7 @@
 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
+import base64
 import contextlib
 import glob
 import inspect
@@ -62,11 +63,12 @@ def configure_extra_pages(app: FastAPI):
     @app.get("/share", include_in_schema=False)
     async def share_page(
         request: Request,
-        share_image: str = Query(),
+        share_image_base64: str = Query(),
         dataset_name: str = Query(),
         dataset_id: str = Query(),
     ):
         url = request.url
+        share_image = base64.b64decode(share_image_base64)
 
         share_page = f"""
         <!DOCTYPE html>