Update app.py

app.py

@@ -1,41 +1,32 @@
 import os
+import sys
 import json
 import requests
 import gradio as gr
-from huggingface_hub import hf_hub_download
+from huggingface_hub import HfFileSystem, hf_hub_download
 from PIL import Image
 
 # Environment Variables
 HF_TOKEN = os.getenv("HF_TOKEN")
 HF_DATASET = os.getenv('HF_DATASET')
-SPACE_SUBDOMAIN = os.getenv('SPACE_SUBDOMAIN')
+SPACE_SUBDOMAIN = os.environ['SPACE_SUBDOMAIN']
 
 def get_image_with_auth(file_name):
     """Retrieve an image using Hugging Face's hub with authentication."""
     image_path = hf_hub_download(repo_id=HF_DATASET, repo_type="dataset", filename=file_name, token=HF_TOKEN)
     return Image.open(image_path)
 
-def recognize_face(image_path):
+def recognize_face(image):
     """
-    Function to send an image URL to the FastAPI backend and receive results.
+    Function to send either an image URL to the FastAPI backend and receive results.
     """
-    # Ensure the SPACE_SUBDOMAIN environment variable is set
-    if not SPACE_SUBDOMAIN:
-        raise ValueError("SPACE_SUBDOMAIN environment variable is not set")
-
-    # Construct the image URL using the full path
-    image_url = f"https://{SPACE_SUBDOMAIN}.hf.space/file={image_path}"
     
     # Set the URL to your FastAPI endpoint
-    url = 'https://dwancin-face-match.hf.space/recognize/'
-
-    # Debugging: Print image URL and API URL
-    print(f"Image URL: {image_url}")
-    print(f"API URL: {url}")
+    url = 'https://dwancin-face-match-api.hf.space/recognize/'
 
     # Prepare the payload with the image data and specify the type
     payload = {
-        "image": image_url,
+        "image": f"https://{SPACE_SUBDOMAIN}.hf.space/file={image}",
         "type": "url"
     }
 
@@ -44,19 +35,15 @@ def recognize_face(image_path):
         "Authorization": f"Bearer {HF_TOKEN}"
     }
     
-    # Debugging: Print payload and headers
-    print(f"Payload: {json.dumps(payload, indent=2)}")
-    print(f"Headers: {headers}")
-
     # Send POST request to FastAPI server with the image data and type
     response = requests.post(url, json=payload, headers=headers)
 
     # Process response
     if response.status_code == 200:
         response_data = response.json()
-        image_file_path = response_data.get('image')
-        if image_file_path:
-            image_file = get_image_with_auth(image_file_path)
+        image_path = response_data.get('image')
+        if image_path:
+            image_file = get_image_with_auth(image_path)
             formatted_json = json.dumps(response_data, indent=4)
             info = f"```json\n{formatted_json}\n```"
             print(formatted_json)
@@ -105,4 +92,4 @@ with gr.Blocks(
     submit.click(fn=recognize_face, inputs=input_image, outputs=[output_image, output_info])
 
 # Launch
-demo.launch(show_api=False)
+demo.launch(show_api=False)