Update app.py

app.py

@@ -8,7 +8,6 @@ import gradio as gr
 import base64
 import mimetypes
 from io import BytesIO
-
 from google import genai
 from google.genai import types
 
@@ -16,18 +15,17 @@ def generate(text, images, api_key, model="gemini-2.5-flash-image-preview"):
     # Initialize client using provided api_key (or fallback to env variable)
     client = genai.Client(api_key=(api_key.strip() if api_key and api_key.strip() != ""
                                      else os.environ.get("GEMINI_API_KEY")))
-    
+
     # Prepare contents with images first, then text
     contents = images + [text]
-    
+
     response = client.models.generate_content(
         model=model,
         contents=contents,
     )
-
     text_response = ""
     image_path = None
-    
+
     for part in response.candidates[0].content.parts:
         if part.text is not None:
             text_response += part.text + "\n"
@@ -39,7 +37,6 @@ def generate(text, images, api_key, model="gemini-2.5-flash-image-preview"):
                 generated_image.save(temp_path)
                 image_path = temp_path
                 print(f"Generated image saved to: {temp_path} with prompt: {text}")
-    
     return image_path, text_response
 
 def load_uploaded_images(uploaded_files):
@@ -56,9 +53,8 @@ def load_uploaded_images(uploaded_files):
 
 def process_image_and_prompt(uploaded_files, prompt, gemini_api_key):
     try:
-        input_text = prompt 
-        model = "gemini-2.5-flash-image-preview" 
-
+        input_text = prompt
+        model = "gemini-2.5-flash-image-preview"
         # Load images from uploaded files
         images = []
         uploaded_images = []
@@ -70,13 +66,11 @@ def process_image_and_prompt(uploaded_files, prompt, gemini_api_key):
                         img = img.convert("RGBA")
                     images.append(img)
                     uploaded_images.append(img)
-
         if not images:
             raise gr.Error("Please upload at least one image", duration=5)
-
         # Format: [dress_image, model_image, text_input] or [image1, image2, ..., text_input]
         image_path, text_response = generate(text=input_text, images=images, api_key=gemini_api_key, model=model)
-        
+
         if image_path:
             # Load and convert the image if needed.
             result_img = Image.open(image_path)
@@ -89,7 +83,6 @@ def process_image_and_prompt(uploaded_files, prompt, gemini_api_key):
     except Exception as e:
         raise gr.Error(f"Error Getting {e}", duration=5)
 
-
 # Build a Blocks-based interface with a custom HTML header and CSS
 with gr.Blocks(css_paths="style.css",) as demo:
     # Custom HTML header with proper class for styling
@@ -101,34 +94,29 @@ with gr.Blocks(css_paths="style.css",) as demo:
       </div>
       <div>
           <h1>Gemini for Image Editing</h1>
-          <p>Powered by <a href="https://gradio.app/">Gradio</a>⚡️| 
-          <a href="https://huggingface.co/spaces/ameerazam08/Gemini-Image-Edit?duplicate=true">Duplicate</a> this Repo |
-          <a href="https://aistudio.google.com/apikey">Get an API Key</a> | 
-          Follow me on Twitter: <a href="https://x.com/Ameerazam18">Ameerazam18</a></p>
+          <p>Powered by <a href="https://gradio.app/">Gradio</a>⚡️|           <a href="https://huggingface.co/spaces/ameerazam08/Gemini-Image-Edit?duplicate=true">Duplicate</a> this Repo |          <a href="https://aistudio.google.com/apikey">Get an API Key</a> |           Follow me on Twitter: <a href="https://x.com/Ameerazam18">Ameerazam18</a></p>
       </div>
     </div>
     """
     )
-    
+
     with gr.Accordion("⚠️ API Configuration ⚠️", open=False, elem_classes="config-accordion"):
         gr.Markdown("""
-    - **Issue:** ❗ Sometimes the model returns text instead of an image.  
-    ### 🔧 Steps to Address:
-    1. **🛠️ Duplicate the Repository**  
-       - Create a separate copy for modifications.  
-    2. **🔑 Use Your Own Gemini API Key**  
-       - You **must** configure your own Gemini key for generation!  
-    """)
-
+    - **Issue:** ❗ Sometimes the model returns text instead of an image.
+      ### 🔧 Steps to Address:
+    1. **🛠️ Duplicate the Repository**
+         - Create a separate copy for modifications.
+      2. **🔑 Use Your Own Gemini API Key**
+         - You **must** configure your own Gemini key for generation!
+      """)
     with gr.Accordion("📌 Usage Instructions", open=False, elem_classes="instructions-accordion"):
         gr.Markdown("""
-    ### 📌 Usage  
-    - Upload an image and enter a prompt to generate outputs.
+    ### 📌 Usage
+      - Upload an image and enter a prompt to generate outputs.
     - If text is returned instead of an image, it will appear in the text output.
     - Upload Only PNG Image
     - ❌ **Do not use NSFW images!**
     """)
-
     with gr.Row(elem_classes="main-content"):
         with gr.Column(elem_classes="input-column"):
             image_input = gr.File(
@@ -151,12 +139,12 @@ with gr.Blocks(css_paths="style.css",) as demo:
                 elem_classes="prompt-input"
             )
             submit_btn = gr.Button("Generate", elem_classes="generate-btn")
-        
+
         with gr.Column(elem_classes="output-column"):
             uploaded_gallery = gr.Gallery(label="Uploaded Images", elem_classes="uploaded-gallery")
             output_gallery = gr.Gallery(label="Generated Outputs", elem_classes="output-gallery")
             output_text = gr.Textbox(
-                label="Gemini Output", 
+                label="Gemini Output",
                 placeholder="Text response will appear here if no image is generated.",
                 elem_classes="output-text"
             )
@@ -167,16 +155,16 @@ with gr.Blocks(css_paths="style.css",) as demo:
         inputs=[image_input, prompt_input, gemini_api_key],
         outputs=[uploaded_gallery, output_gallery, output_text],
     )
-    
+
     # Update uploaded gallery immediately when files are uploaded
     image_input.upload(
         fn=load_uploaded_images,
         inputs=[image_input],
         outputs=[uploaded_gallery],
     )
-    
+
     gr.Markdown("## Try these examples", elem_classes="gr-examples-header")
-    
+
     examples = [
         ["data/1.webp", 'change text to "AMEER"'],
         ["data/2.webp", "remove the spoon from hand only"],
@@ -187,7 +175,7 @@ with gr.Blocks(css_paths="style.css",) as demo:
         ["data/76860.jpg", "add lipstick on lip only"],
         ["data/2807615.jpg", "make it happy looking face only"],
     ]
-    
+
     gr.Examples(
         examples=examples,
         inputs=[image_input, prompt_input,],