Update app.py

app.py

@@ -24,7 +24,7 @@ def translate_prompt_to_english(text, api_key, model="gemini-2.0-flash-exp"):
 
     client = genai.Client(api_key=api_key.strip())
 
-    pre_prompt = "Translate this to English accurately:"
+    pre_prompt = "Translate this to English:"
     full_text = pre_prompt + "\n" + text
 
     contents = [
@@ -59,7 +59,8 @@ def generate_with_api(api_key, text, file_name, model="gemini-2.0-flash-exp"):
     client = genai.Client(api_key=api_key.strip())
     files = [client.files.upload(file=file_name)]
 
-    full_text = "Apply this edit to the image: " + text
+    pre_prompt = "Apply these changes to the image:"
+    full_text = pre_prompt + "\n" + text
 
     contents = [
         types.Content(
@@ -78,9 +79,11 @@ def generate_with_api(api_key, text, file_name, model="gemini-2.0-flash-exp"):
         top_p=0.95,
         top_k=40,
         max_output_tokens=8192,
-        response_modalities=["image"],
+        response_modalities=["image", "text"],
+        response_mime_type="text/plain",
     )
 
+    text_response = ""
     image_path = None
     with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
         temp_path = tmp.name
@@ -96,9 +99,11 @@ def generate_with_api(api_key, text, file_name, model="gemini-2.0-flash-exp"):
                 save_binary_file(temp_path, candidate.inline_data.data)
                 image_path = temp_path
                 break
+            else:
+                text_response += chunk.text + "\n"
 
     del files
-    return image_path, ""
+    return image_path, text_response
 
 def process_single_api(api_key, prompt, file_name, model):
     if not api_key:
@@ -106,14 +111,14 @@ def process_single_api(api_key, prompt, file_name, model):
     
     try:
         translated_prompt = translate_prompt_to_english(prompt, api_key, model)
-        image_path, _ = generate_with_api(api_key, translated_prompt, file_name, model)
+        image_path, text_response = generate_with_api(api_key, translated_prompt, file_name, model)
         
         if image_path:
             result_img = Image.open(image_path)
             if result_img.mode == "RGBA":
                 result_img = result_img.convert("RGB")
             return result_img, ""
-        return None, "No image generated"
+        return None, text_response if text_response else "No image generated"
     
     except Exception as e:
         return None, f"Error with API {api_key[-4:]}: {str(e)}"
@@ -153,12 +158,12 @@ def process_image_and_prompt(composite_pil, prompt):
         os.unlink(composite_path)
 
         if not result_images:
-            return None, "\n".join(error_messages)
+            return None, "\n".join(error_messages) + "\n\n**توجه**: اگر تصویر تولید نشد، لطفاً دستور خود را واضح‌تر بنویسید یا دوباره امتحان کنید."
         
         return result_images, ""
 
     except Exception as e:
-        raise gr.Error(f"خطا در پردازش: {e}")
+        raise gr.Error(f"خطا در پردازش: {e}", duration=5)
 
 css = """
 footer { visibility: hidden; }
@@ -168,7 +173,7 @@ display: none !important;
 }
 """
 
-with gr.Blocks(css=css) as demo:
+with gr.Blocks(css_paths="style.css", css=css) as demo:
     gr.HTML(
     """
     <div class="header-container">
@@ -182,30 +187,48 @@ with gr.Blocks(css=css) as demo:
     """
     )
 
-    with gr.Accordion("⚠️ راهنمای استفاده", open=False):
+    with gr.Accordion("⚠️ راهنمای استفاده", open=False, elem_classes="config-accordion"):
+        gr.Markdown("""
+        ### راهنمای استفاده
+        - تصویر خود را آپلود کرده و دستور ویرایش را وارد کنید
+        - در صورت بروز خطا، پیام مربوطه نمایش داده خواهد شد
+        - فقط تصاویر با فرمت PNG آپلود کنید
+        - از آپلود تصاویر نامناسب خودداری کنید
+        """)
+
+    with gr.Accordion("📌 دستورالعمل‌های ویرایش", open=False, elem_classes="instructions-accordion"):
         gr.Markdown("""
-        - تصویر خود را آپلود کنید
-        - دستور ویرایش را وارد کنید
-        - نتیجه را دریافت نمایید
+        ### نمونه دستورات ویرایش
+        - متن تصویر را به \"متن جدید\" تغییر بده
+        - شیء خاصی را از تصویر حذف کن
+        - استایل خاصی به بخشی از تصویر اضافه کن
+        - تغییرات رنگی روی تصویر اعمال کن
         """)
 
-    with gr.Row():
-        with gr.Column():
+    with gr.Row(elem_classes="main-content"):
+        with gr.Column(elem_classes="input-column"):
             image_input = gr.Image(
                 type="pil",
                 label="تصویر را آپلود کنید",
-                image_mode="RGBA"
+                image_mode="RGBA",
+                elem_id="image-input",
+                elem_classes="upload-box"
             )
             prompt_input = gr.Textbox(
                 lines=2,
-                placeholder="دستور ویرایش خود را وارد کنید...",
-                label="دستور ویرایش"
+                placeholder="تصویر چیکار بشه؟ اینجا بنویسید...",
+                label="دستور ویرایش",
+                elem_classes="prompt-input"
             )
-            submit_btn = gr.Button("اعمال تغییرات")
+            submit_btn = gr.Button("اعمال تغییرات", elem_classes="generate-btn")
         
-        with gr.Column():
-            output_gallery = gr.Gallery(label="نتایج")
-            output_text = gr.Textbox(label="پیام سیستم")
+        with gr.Column(elem_classes="output-column"):
+            output_gallery = gr.Gallery(label="تصاویر ویرایش شده", elem_classes="output-gallery")
+            output_text = gr.Textbox(
+                label="پیام سیستم", 
+                placeholder="در صورت بروز خطا، پیام مربوطه اینجا نمایش داده می‌شود.",
+                elem_classes="output-text"
+            )
 
     submit_btn.click(
         fn=process_image_and_prompt,
@@ -213,15 +236,23 @@ with gr.Blocks(css=css) as demo:
         outputs=[output_gallery, output_text],
     )
 
+    gr.Markdown("## نمونه‌های آماده", elem_classes="gr-examples-header")
+
     examples = [
-        ["data/1.webp", 'متن را به "امیر" تغییر بده'],
-        ["data/2.webp", "قاشق را از دست حذف کن"],
-        ["data/3.webp", 'متن را به "بساز" تغییر بده'],
+        ["data/1.webp", 'متن را به "امیر" تغییر بده', ""],
+        ["data/2.webp", "قاشق را از دست حذف کن", ""],
+        ["data/3.webp", 'متن را به "بساز" تغییر بده', ""],
+        ["data/1.jpg", "فقط روی صورت استایل جوکر اضافه کن", ""],
+        ["data/1777043.jpg", "فقط روی صورت استایل جوکر اضافه کن", ""],
+        ["data/2807615.jpg", "فقط روی لب‌ها رژ لب اضافه کن", ""],
+        ["data/76860.jpg", "فقط روی لب‌ها رژ لب اضافه کن", ""],
+        ["data/2807615.jpg", "فقط صورت را شادتر کن", ""],
     ]
 
     gr.Examples(
         examples=examples,
         inputs=[image_input, prompt_input],
+        elem_id="examples-grid"
     )
 
 demo.queue(max_size=50).launch()