Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 import numpy as np
-
 import os
 from PIL import Image
 import requests
@@ -8,50 +7,41 @@ from io import BytesIO
 import io
 import base64
 
-hf_token = os.environ.get("HF_TOKEN_API_DEMO") # we get it from a secret env variable, such that it's private
+hf_token = os.environ.get("HF_TOKEN_API_DEMO")  # نحصل عليه من متغير بيئة سري لضمان الخصوصية
 auth_headers = {"api_token": hf_token}
 
 def convert_mask_image_to_base64_string(mask_image):
     buffer = io.BytesIO()
-    mask_image.save(buffer, format="PNG")  # You can choose the format (e.g., "JPEG", "PNG")
-    # Encode the buffer in base64
+    mask_image.save(buffer, format="PNG")  # يمكن اختيار التنسيق (مثل "JPEG" أو "PNG")
+    # تحويل الصورة إلى base64
     image_base64_string = base64.b64encode(buffer.getvalue()).decode('utf-8')
-    return f",{image_base64_string}" # for some reason the funciton which downloads image from base64 expects prefix of "," which is redundant in the url
+    return f",{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"
-    
     payload = {
-    "file": image_base64_file,
-    "mask_file": mask_base64_file,
-    "mask_type": mask_type,
+        "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
 
-
 def predict(dict):
-
-    init_image = Image.fromarray(dict['background'][:, :, :3], 'RGB') #dict['background'].convert("RGB")#.resize((1024, 1024))
-    mask = Image.fromarray(dict['layers'][0][:,:,3], 'L') #dict['layers'].convert("RGB")#.resize((1024, 1024))
-    
+    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"
     gen_img = eraser_api_call(image_base64_file, mask_base64_file, mask_type)
-    
     return gen_img
 
-
 css = '''
 .gradio-container{max-width: 1100px !important}
 #image_upload{min-height:400px}
@@ -84,15 +74,15 @@ div#share-btn-container > div {flex-direction: row;background: black;align-items
 #prompt input{width: calc(100% - 160px);border-top-right-radius: 0px;border-bottom-right-radius: 0px;}
 #run_button {
     width: 100%;
-    height: 50px;  /* Set a fixed height for the button */
+    height: 50px;  /* تحديد ارتفاع ثابت للزر */
     display: flex;
     align-items: center;
     justify-content: center;
 }
 #output-img img, #image_upload img {
-    object-fit: contain; /* Ensure aspect ratio is preserved */
+    object-fit: contain; /* الحفاظ على نسبة العرض إلى الارتفاع */
     width: 100%;
-    height: auto; /* Let height adjust automatically */
+    height: auto; /* السماح بارتفاع تلقائي */
 }
 #prompt-container{margin-top:-18px;}
 #prompt-container .form{border-top-left-radius: 0;border-top-right-radius: 0}
@@ -121,20 +111,22 @@ with image_blocks as demo:
         ''')
     with gr.Row():
         with gr.Column():
-            image = gr.ImageEditor(sources=["upload"], layers=False, transforms=[], 
-                                   brush=gr.Brush(colors=["#000000"], color_mode="fixed"),
-                                   )
+            image = gr.ImageEditor(
+                sources=["upload"],
+                layers=False,
+                transforms=[], 
+                brush=gr.Brush(colors=["#000000"], color_mode="fixed"),  # اللون الأسود كخيار ثابت
+            )
             with gr.Row(elem_id="prompt-container", equal_height=True):
-                with gr.Column():  # Wrap the button inside a Column
+                with gr.Column():  # التفاف الزر داخل عمود
                     btn = gr.Button("Erase!", elem_id="run_button")
         
         with gr.Column():
             image_out = gr.Image(label="Output", elem_id="output-img")
 
-    # Button click will trigger the inpainting function (no prompt required)
+    # تشغيل الزر لتفعيل وظيفة التعديل
     btn.click(fn=predict, inputs=[image], outputs=[image_out], api_name='run')
     
-
     gr.HTML(
         """
             <div class="footer">
@@ -144,4 +136,4 @@ with image_blocks as demo:
         """
     )
 
-image_blocks.queue(max_size=25,api_open=False).launch(show_api=False)
+image_blocks.queue(max_size=25, api_open=False).launch(show_api=False)