Update app.py

app.py

@@ -8,37 +8,42 @@ import io
 import base64
 
 hf_token = os.environ.get("HF_TOKEN_API_DEMO")  # نحصل عليه من متغير بيئة سري لضمان الخصوصية
-auth_headers = {"api_token": hf_token}
+auth_headers = {"Authorization": f"Bearer {hf_token}"}  # تحديث رأس المصادقة
 
 def convert_mask_image_to_base64_string(mask_image):
     buffer = io.BytesIO()
     mask_image.save(buffer, format="PNG")  # يمكن اختيار التنسيق (مثل "JPEG" أو "PNG")
     # تحويل الصورة إلى base64
     image_base64_string = base64.b64encode(buffer.getvalue()).decode('utf-8')
-    return f",{image_base64_string}"  # يتم إضافة "," لتكون متوافقة مع الوظيفة التي تستخرج الصور
+    return image_base64_string  # لا حاجة لإضافة "," هنا
 
 def download_image(url):
     response = requests.get(url)
     return Image.open(BytesIO(response.content)).convert("RGB")
 
 def eraser_api_call(image_base64_file, mask_base64_file, mask_type):
-    url = "http://engine.prod.bria-api.com/v1/eraser"
+    # تحديث الرابط ليعمل مع Gen-Fill API
+    url = "https://api.bria.ai/v1/gen-fill"
     payload = {
         "file": image_base64_file,
         "mask_file": mask_base64_file,
         "mask_type": mask_type,
     }
     response = requests.post(url, json=payload, headers=auth_headers)
-    response = response.json()
-    res_image = download_image(response["result_url"])
-    return res_image
+    
+    if response.status_code == 200:
+        response = response.json()
+        res_image = download_image(response["result_url"])  # تحميل الصورة الناتجة
+        return res_image
+    else:
+        raise Exception(f"API Error: {response.status_code} - {response.text}")
 
 def predict(dict):
     init_image = Image.fromarray(dict['background'][:, :, :3], 'RGB')
     mask = Image.fromarray(dict['layers'][0][:, :, 3], 'L')
     image_base64_file = convert_mask_image_to_base64_string(init_image)
     mask_base64_file = convert_mask_image_to_base64_string(mask)
-    mask_type = "manual"
+    mask_type = "manual"  # حدد نوع القناع
     gen_img = eraser_api_call(image_base64_file, mask_base64_file, mask_type)
     return gen_img