fix: user api key

app.py

@@ -89,17 +89,22 @@ def load_examples(n: int = 3):
 static_examples = load_examples()
 
 # ───────── backend for user Run ─────────
-def run_mapper(prompt: str, img_url: str):
+def run_mapper(prompt: str, img_url: str, api_key: str):
+    new_client = JinaV4SimilarityMapper(client_type="web")
     if not img_url:
         raise gr.Error("Please provide an image URL.")
+    if not prompt:
+        raise gr.Error("Please provide a prompt.")
+    if not api_key:
+        raise gr.Error("Please provide a valid API key.")
     try:
         r = requests.get(img_url, stream=True, timeout=10); r.raise_for_status()
         img = Image.open(io.BytesIO(r.content)).convert("RGB")
     except Exception as e:
         raise gr.Error(f"Image load failed: {e}")
-
-    img_proc, _, _ =  MAPPER.process_image(img_url)
-    toks, maps = MAPPER.get_token_similarity_maps(prompt, img_proc)
+    new_client.model.set_api_key(api_key)
+    img_proc, _, _ =  new_client.process_image(img_url)
+    toks, maps = new_client.get_token_similarity_maps(prompt, img_proc)
     if not toks:
         raise gr.Error("Mapper returned no tokens.")
 
@@ -127,6 +132,7 @@ with gr.Blocks(css=css) as demo:
     # User input
     prompt_in = gr.Textbox(label="Prompt", placeholder="Describe what to query…")
     url_in    = gr.Textbox(label="Image URL", placeholder="https://example.com/image.jpg")
+    api_key_in = gr.Textbox(label="API Key", placeholder="Enter your Jina API key here")
     run_btn   = gr.Button("Run", elem_id="run-btn")
 
     # Output area
@@ -137,7 +143,7 @@ with gr.Blocks(css=css) as demo:
     img_out   = gr.Image(label="Image + Heatmap", visible=False)
 
     run_btn.click(run_mapper,
-        [prompt_in, url_in],
+        [prompt_in, url_in, api_key_in],
         [token_sel, maps_st, img_st, img_out, info_md, prompt_in, url_in])
 
     (token_sel.select if hasattr(token_sel,"select") else token_sel.change)(

similarity.py

@@ -24,22 +24,14 @@ class JinaEmbeddingsClient:
 
     def __init__(
         self,
-        api_key: str | None = None,
         model: str = "jina-embeddings-v4",
         return_multivector: bool = True,
         task: str = "retrieval.query",
         timeout: int = 30,
     ) -> None:
-        # Grab token from env if it wasn’t provided
-        api_key = api_key or os.getenv("JINA_TOKEN")
-        if not api_key:
-            raise ValueError(
-                "No API key supplied. Pass `api_key=` or set the env var JINA_TOKEN."
-            )
-
         self.headers = {
             "Content-Type": "application/json",
-            "Authorization": f"Bearer {api_key}",
+            "Authorization": f"Bearer Not Set",
         }
         self.base_payload = {
             "model": model,
@@ -90,6 +82,14 @@ class JinaEmbeddingsClient:
         resp.raise_for_status()
         return resp.json()
 
+    def set_api_key(self, api_key: str) -> None:
+        """
+        Set the API key for authentication.
+        """
+        if not api_key:
+            raise ValueError("API key must not be empty.")
+        self.headers["Authorization"] = f"Bearer {api_key}"
+
     @staticmethod
     def _as_tensors(data: List[Dict[str, Any]]) -> List[torch.Tensor]:
         """
@@ -118,7 +118,7 @@ class JinaV4SimilarityMapper:
         device: str = "cuda" if torch.cuda.is_available() else "cpu",
         heatmap_alpha: float = 0.6,
         num_vectors: int = 128,
-        client_type: str = "local"
+        client_type: str = "local",
     ):
         """
         Initialize the mapper with Jina Embedding v4.

test.py

@@ -7,9 +7,14 @@ import subprocess
 import requests
 import gradio as gr
 from PIL import Image
+import os 
 
 from app import MAPPER, save_run
 
+api_key = os.getenv("JINA_TOKEN")
+
+MAPPER.model.set_api_key(api_key)
+
 inputs = [
     ("A person riding a bicycle near the ocean", "https://cdn.duvine.com/wp-content/uploads/2016/04/17095703/Slides_mallorca_FOR-WEB.jpg"),
     ("Computing jobs in the US", "https://www.ayresassociates.com/wp-content/uploads/2019/02/Career-Expo-Pie-Chart-Crop.jpg"),