Update app.py

app.py

@@ -4,6 +4,7 @@ from diffusers import StableDiffusionInstructPix2PixPipeline
 import gradio as gr
 from PIL import Image
 import random
+from langdetect import detect, LangDetectException
 
 # Load the InstructPix2Pix model
 model_id = "timbrooks/instruct-pix2pix"
@@ -29,7 +30,7 @@ def change_style():
     seed = torch.manual_seed(torch.randint(0, 10000, (1,)).item())
     return f"تم تغيير النمط. المعرف الجديد: {seed}"
 
-# Dictionary to map Arabic colors to English
+# Dictionary to map Arabic to English colors
 arabic_to_english_colors = {
     "أبيض": "White",
     "أسود": "Black",
@@ -48,16 +49,31 @@ arabic_to_english_colors = {
     "بيج": "Beige"
 }
 
-# Function to translate Arabic color to English and change the wall color
-def change_color(image, color):
-    # Translate Arabic color to English using the dictionary
-    color_in_english = arabic_to_english_colors.get(color, None)
-    
-    # If color not found in the dictionary, return an error message
-    if not color_in_english:
-        return f"اللون '{color}' غير موجود في القائمة. يرجى إدخال لون صحيح."
+# List of English colors
+english_colors = [
+    "White", "Black", "Blue", "Green", "Red", "Yellow", 
+    "Gray", "Orange", "Purple", "Pink", "Brown", 
+    "Navy", "Coral", "Teal", "Beige"
+]
 
-    # Construct the furniture prompt in English
+# Function to detect whether the text is in Arabic using langdetect
+def is_arabic(text):
+    try:
+        return detect(text) == 'ar'  # 'ar' is the ISO 639-1 code for Arabic
+    except LangDetectException:
+        return False  # If detection fails, assume it's not Arabic
+
+# Function to translate Arabic color to English and change the wall color
+def change_color(image, arabic_color, english_color):
+    # If a color is chosen from the Arabic dropdown, use it, otherwise use the English dropdown
+    if arabic_color:
+        color_in_english = arabic_to_english_colors.get(arabic_color)
+    elif english_color:
+        color_in_english = english_color
+    else:
+        return "Please select a color."
+
+    # Construct the prompt to change the wall color
     prompt = f"paint the walls with {color_in_english} color"
   
     # Text CFG (guidance_scale) controls how strongly the model follows the prompt
@@ -66,7 +82,7 @@ def change_color(image, color):
     # Image CFG: Simulated value for preserving the original image content
     image_cfg = 1.5
   
-    # Apply the edit using InstructPix2Pix, with text CFG and image CFG influencing the guidance scale
+    # Apply the edit using InstructPix2Pix
     edited_image = pipe(
         prompt=prompt,
         image=image,
@@ -78,7 +94,7 @@ def change_color(image, color):
   
     return edited_image
 
-# Gradio interface for image editing in Arabic
+# Gradio interface for image editing with separate Arabic and English color dropdowns
 def image_interface():
     with gr.Blocks(css=".gradio-container {direction: rtl}") as demo_color:
         gr.Markdown("## تطبيق لتغيير لون الجدران")
@@ -86,15 +102,11 @@ def image_interface():
         # Image upload (translated to Arabic)
         image_input = gr.Image(type="pil", label="قم برفع صورة للغرفة")
 
-        # List of common painting colors in Arabic
-        common_colors = [
-            "أبيض", "أسود", "أزرق", "أخضر", "أحمر", "أصفر", 
-            "رمادي", "برتقالي", "بنفسجي", "وردي", "بني", 
-            "كحلي", "زهري", "فيروزي", "بيج"
-        ]
+        # Dropdown for Arabic wall color
+        arabic_color_input = gr.Dropdown(list(arabic_to_english_colors.keys()), label="اختر لون الجدران (بالعربية)")
 
-        # Dropdown for wall color (Arabic)
-        color_input = gr.Dropdown(common_colors, label="اختر لون الجدران")
+        # Dropdown for English wall color
+        english_color_input = gr.Dropdown(english_colors, label="Choose Wall Color (English)")
 
         # Display output image
         result_image = gr.Image(label="الصورة المعدلة")
@@ -103,7 +115,7 @@ def image_interface():
         submit_button = gr.Button("قم بتغيير لون الجدران")
 
         # Define action on button click (directly pass dropdown color input to the function)
-        submit_button.click(fn=change_color, inputs=[image_input, color_input], outputs=result_image)
+        submit_button.click(fn=change_color, inputs=[image_input, arabic_color_input, english_color_input], outputs=result_image)
 
     return demo_color
 
@@ -116,10 +128,15 @@ def translate_prompt(prompt_ar):
     return prompt_en
 
 
-# General image editing function
-def edit_image(image, instruction_ar):
-    # Translate Arabic instruction to English
-    instruction_en = translate_prompt(instruction_ar)
+# General image editing function (supports both Arabic and English)
+def edit_image(image, instruction):
+    # Detect if the instruction is in Arabic
+    if is_arabic(instruction):
+        # Translate Arabic instruction to English
+        instruction_en = translate_prompt(instruction)
+    else:
+        # If it's in English, use it directly
+        instruction_en = instruction
     
     # Text CFG (guidance_scale) controls how strongly the model follows the prompt
     text_cfg = 12.0
@@ -140,16 +157,16 @@ def edit_image(image, instruction_ar):
     return edited_image
 
 
-# Gradio interface for general image editing in Arabic
+# Gradio interface for general image editing in Arabic and English
 def general_editing_interface():
     with gr.Blocks(css=".gradio-container {direction: rtl}") as demo_general:
         gr.Markdown("## تطبيق تحرير الصور العام")
 
-        # Image upload in Arabic
+        # Image upload in Arabic and English
         image_input = gr.Image(type="pil", label="قم بتحميل صورة")
 
-        # Textbox for instruction in Arabic
-        instruction_input = gr.Textbox(label="أدخل التعليمات", placeholder="وصف التعديلات (مثل: 'اجعل الجو مثلج')")
+        # Textbox for instruction in Arabic or English
+        instruction_input = gr.Textbox(label="أدخل التعليمات", placeholder="وصف التعديلات (مثل: 'اجعل الجو مثلج') أو 'Make it snowy'")
 
         # Display output image
         result_image = gr.Image(label="الصورة المعدلة")