Update app.py

app.py

@@ -12,43 +12,36 @@ logger = logging.getLogger(__name__)
 # ==============================================================================
 #  CLIENT-SIDE API CALL LOGIC
 # ==============================================================================
-
-def call_api(server_endpoint_url: str, image: Image.Image) -> dict:
+def call_api(server_base_url: str, image: Image.Image) -> dict:
     """
     Takes a PIL image, converts it to base64, sends it to the specified server API,
     and returns the decrypted JSON data.
     """
-    # --- Input Validation ---
-    if not server_endpoint_url or "/run/mcp_decode" not in server_endpoint_url:
-        raise gr.Error("Please provide the FULL API endpoint URL, including the /run/mcp_decode path.")
+    if not server_base_url or "hf.space" not in server_base_url:
+        raise gr.Error("Please provide a valid Hugging Face Space URL for the server API.")
     if image is None:
         raise gr.Error("Please upload an encrypted image.")
 
-    api_endpoint = server_endpoint_url.strip()
+    # --- CORRECTED API PATH ---
+    api_endpoint = f"{server_base_url.strip('/')}/run/keylock-auth-decoder"
     logger.info(f"Client attempting to call API at: {api_endpoint}")
 
     try:
-        # 1. Convert the PIL Image to a base64 string
         with io.BytesIO() as buffer:
             image.save(buffer, format="PNG")
             image_base64 = base64.b64encode(buffer.getvalue()).decode('utf-8')
 
-        # 2. Prepare the JSON payload for the Gradio API
         payload = {"data": [image_base64]}
-
-        # 3. Make the POST request to the server API
         headers = {"Content-Type": "application/json"}
         response = requests.post(api_endpoint, headers=headers, json=payload, timeout=30)
         
-        # 4. Process the response
         response_json = response.json()
         if response.status_code == 200:
-            if "data" in response_json and len(response_json["data"]) > 0:
+            if "data" in response_json:
                 return response_json["data"][0]
             elif "error" in response_json:
                 raise gr.Error(f"API returned an error: {response_json['error']}")
         else:
-            # Handle non-200 responses with more detail
             error_detail = response_json.get("error", "Unknown error.")
             raise gr.Error(f"API Error (Status {response.status_code}): {error_detail}")
 
@@ -62,20 +55,17 @@ def call_api(server_endpoint_url: str, image: Image.Image) -> dict:
 # ==============================================================================
 #  GRADIO INTERFACE
 # ==============================================================================
-
 with gr.Blocks(theme=gr.themes.Soft(), title="KeyLock API Client") as demo:
     gr.Markdown("# 🔑 KeyLock API Client")
-    gr.Markdown(
-        "This UI calls the **Secure Decoder API**. It sends an encrypted image to the server for decryption."
-    )
+    gr.Markdown("This UI calls the **Secure Decoder API**. It sends an encrypted image to the server for decryption.")
 
     with gr.Row():
         with gr.Column(scale=1):
-            gr.Markdown("### 1. Configure Server Endpoint")
+            gr.Markdown("### 1. Configure Server URL")
             server_url_input = gr.Textbox(
-                label="Full Decoder API Endpoint URL",
-                placeholder="https://your-server-space.hf.space/run/mcp_decode",
-                info="You must include the /run/mcp_decode path."
+                label="Decoder API Server Base URL",
+                placeholder="https://your-name-keylock-auth.hf.space",
+                info="This is the direct URL of your server space, without any path."
             )
 
             gr.Markdown("### 2. Upload Encrypted Image")