Update app.py

app.py

@@ -6,6 +6,7 @@ import requests
 from io import BytesIO
 import io
 import base64
+import json  # لإضافة دعم قراءة ملف config.json
 
 # الحصول على التوكن من متغير البيئة
 hf_token = os.environ.get("HF_TOKEN_API_DEMO")
@@ -13,6 +14,15 @@ if not hf_token:
     raise ValueError("HF_TOKEN_API_DEMO is not set. Please set it as an environment variable.")
 auth_headers = {"api_token": hf_token}
 
+# تحميل إعدادات النموذج من ملف config.json
+def load_config(config_path="config.json"):
+    if not os.path.exists(config_path):
+        raise FileNotFoundError(f"Config file not found at {config_path}")
+    with open(config_path, 'r') as file:
+        config = json.load(file)
+    print("Config loaded successfully:", config)
+    return config
+
 # تحويل الصورة إلى Base64
 def convert_mask_image_to_base64_string(mask_image):
     buffer = io.BytesIO()
@@ -41,9 +51,16 @@ def eraser_api_call(image_base64_file, mask_base64_file, mask_type):
     if response.status_code != 200:
         print(f"Error Response: {response.text}")  # طباعة نص الخطأ إذا حدث
         raise ValueError(f"API request failed: {response.status_code}")
-    response = response.json()
-    print("Response JSON:", response)  # طباعة الرد JSON
-    res_image = download_image(response["result_url"])
+    try:
+        response_json = response.json()
+        print("Response JSON:", response_json)  # طباعة الرد JSON
+    except ValueError:
+        raise ValueError("Invalid JSON response from API")
+    
+    if "result_url" not in response_json:
+        raise KeyError("The key 'result_url' is missing in the response JSON")
+    
+    res_image = download_image(response_json["result_url"])
     return res_image
 
 # دالة التنبؤ
@@ -69,6 +86,9 @@ def predict(dict):
     
     return gen_img
 
+# قراءة إعدادات config.json (إذا لزم الأمر)
+config = load_config()
+
 # CSS مخصص
 css = '''
 .gradio-container{max-width: 1100px !important}