Upload 12 files

.gitattributes

@@ -33,3 +33,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+data/1.jpg filter=lfs diff=lfs merge=lfs -text
+data/1777043.jpg filter=lfs diff=lfs merge=lfs -text
+data/2807615.jpg filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,13 +1,12 @@
 ---
-title: Remove Background Web
-emoji: 🖼️
-colorFrom: red
-colorTo: yellow
-sdk: static
-pinned: false
-models:
-- briaai/RMBG-1.4
-short_description: In-browser background removal
+title: Gemini-2.5-Flash-Nano-Banana
+emoji: 📚
+colorFrom: indigo
+colorTo: blue
+sdk: gradio
+sdk_version: 5.34.0
+app_file: app.py
+pinned: true
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,197 @@
+import json
+import os
+import time
+import uuid
+import tempfile
+from PIL import Image, ImageDraw, ImageFont
+import gradio as gr
+import base64
+import mimetypes
+from io import BytesIO
+
+from google import genai
+from google.genai import types
+
+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"
+        elif part.inline_data is not None:
+            # Create a temporary file to store the generated image
+            with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
+                temp_path = tmp.name
+                generated_image = Image.open(BytesIO(part.inline_data.data))
+                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):
+    """Load and display uploaded images immediately"""
+    uploaded_images = []
+    if uploaded_files:
+        for file in uploaded_files:
+            if file.name.lower().endswith(('.png', '.jpg', '.jpeg', '.webp')):
+                img = Image.open(file.name)
+                if img.mode == "RGBA":
+                    img = img.convert("RGBA")
+                uploaded_images.append(img)
+    return uploaded_images
+
+def process_image_and_prompt(uploaded_files, prompt, gemini_api_key):
+    try:
+        input_text = prompt 
+        model = "gemini-2.5-flash-image-preview" 
+
+        # Load images from uploaded files
+        images = []
+        uploaded_images = []
+        if uploaded_files:
+            for file in uploaded_files:
+                if file.name.lower().endswith(('.png', '.jpg', '.jpeg', '.webp')):
+                    img = Image.open(file.name)
+                    if img.mode == "RGBA":
+                        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)
+            if result_img.mode == "RGBA":
+                result_img = result_img.convert("RGBA")
+            return uploaded_images, [result_img], ""  # Return uploaded images, generated image, and empty text output.
+        else:
+            # Return uploaded images, no generated image, and the text response.
+            return uploaded_images, None, text_response
+    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
+    gr.HTML(
+    """
+    <div class="header-container">
+      <div>
+          <img src="https://www.gstatic.com/lamda/images/gemini_favicon_f069958c85030456e93de685481c559f160ea06b.png" alt="Gemini logo">
+      </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>
+      </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!  
+    """)
+
+    with gr.Accordion("📌 Usage Instructions", open=False, elem_classes="instructions-accordion"):
+        gr.Markdown("""
+    ### 📌 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(
+                file_types=["image"],
+                file_count="multiple",
+                label="Upload Images ",
+                elem_id="image-input",
+                elem_classes="upload-box"
+            )
+            gemini_api_key = gr.Textbox(
+                lines=1,
+                placeholder="Enter Gemini API Key (optional)",
+                label="Gemini API Key (optional)",
+                elem_classes="api-key-input"
+            )
+            prompt_input = gr.Textbox(
+                lines=2,
+                placeholder="Enter prompt here...",
+                label="Prompt",
+                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", 
+                placeholder="Text response will appear here if no image is generated.",
+                elem_classes="output-text"
+            )
+
+    # Set up the interaction with three outputs.
+    submit_btn.click(
+        fn=process_image_and_prompt,
+        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"],
+        ["data/3.webp", 'change text to "Make it "'],
+        ["data/1.jpg", "add joker style only on face"],
+        ["data/1777043.jpg", "add joker style only on face"],
+        ["data/2807615.jpg", "add lipstick on lip only"],
+        ["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,],
+        elem_id="examples-grid"
+    )
+
+demo.queue(max_size=50).launch(mcp_server=True, share=True)

requirements.txt

@@ -0,0 +1,3 @@
+google-genai==1.5.0
+gradio
+pydantic==2.10.6

style.css

@@ -0,0 +1,408 @@
+/* Creating a more modern looking UI for the Gemini Image Editing App */
+/* Global Styles */
+:root {
+  --primary-color: #4f46e5;
+  --primary-light: #6366f1;
+  --primary-dark: #3730a3;
+  --secondary-color: #9333ea;
+  --accent-color: #ec4899;
+  --text-color: #0f172a;
+  --text-light: #64748b;
+  --bg-color: #f8fafc;
+  --card-bg: #ffffff;
+  --border-color: #e2e8f0;
+  --shadow: 0 4px 6px -1px rgba(0, 0, 0, 0.1), 0 2px 4px -1px rgba(0, 0, 0, 0.06);
+  --transition: all 0.3s ease;
+  --radius: 12px;
+  --header-gradient: linear-gradient(135deg, var(--primary-color), var(--secondary-color));
+}
+
+/* Dark mode support */
+.dark {
+  --text-color: #e2e8f0;
+  --text-light: #94a3b8;
+  --bg-color: #0f172a;
+  --card-bg: #1e293b;
+  --border-color: #334155;
+}
+
+/* Base container styling */
+.gradio-container {
+  font-family: 'Inter', system-ui, -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, 'Helvetica Neue', Arial, sans-serif;
+  background-color: var(--bg-color);
+  color: var(--text-color);
+  max-width: 1200px;
+  margin: 0 auto;
+  padding: 2rem;
+  box-shadow: var(--shadow);
+  border-radius: var(--radius);
+}
+
+/* Header styling */
+.header-container {
+  background: var(--header-gradient);
+  border-radius: var(--radius);
+  padding: 2rem;
+  margin-bottom: 2rem;
+  box-shadow: var(--shadow);
+  color: white;
+  display: flex;
+  align-items: center;
+  justify-content: center;
+  gap: 20px;
+}
+
+.header-container img {
+  width: 100px;
+  height: 100px;
+  filter: drop-shadow(0 0 8px rgba(255, 255, 255, 0.5));
+  transition: var(--transition);
+}
+
+.header-container img:hover {
+  transform: scale(1.05) rotate(5deg);
+}
+
+.header-container h1 {
+  font-size: 2.5rem;
+  font-weight: 700;
+  margin-bottom: 0.5rem;
+  background: linear-gradient(45deg, #fff, #f0f0ff);
+  -webkit-background-clip: text;
+  background-clip: text;
+  color: transparent;
+  display: inline-block;
+}
+
+.header-container a {
+  color: white;
+  text-decoration: none;
+  font-weight: 600;
+  border-bottom: 2px solid rgba(255, 255, 255, 0.5);
+  transition: var(--transition);
+  padding-bottom: 2px;
+}
+
+.header-container a:hover {
+  border-color: white;
+}
+
+/* Accordion styling */
+.gr-accordion {
+  border: none !important;
+  background: var(--card-bg);
+  border-radius: var(--radius);
+  overflow: hidden;
+  box-shadow: var(--shadow);
+  margin-bottom: 1.5rem;
+  transition: var(--transition);
+}
+
+.gr-accordion:hover {
+  box-shadow: 0 10px 15px -3px rgba(0, 0, 0, 0.1), 0 4px 6px -2px rgba(0, 0, 0, 0.05);
+}
+
+.gr-accordion-title {
+  background-color: var(--primary-color);
+  color: white !important;
+  padding: 1rem 1.5rem;
+  font-weight: 600;
+  cursor: pointer;
+}
+
+
+/* Input fields */
+.gr-input, .gr-textarea {
+  width: 100%;
+  padding: 0.75rem 1rem;
+  border: 1px solid var(--border-color);
+  border-radius: var(--radius);
+  background-color: var(--card-bg);
+  transition: var(--transition);
+  font-size: 1rem;
+  color: var(--text-color);
+}
+
+.gr-input:focus, .gr-textarea:focus {
+  border-color: var(--primary-color);
+  box-shadow: 0 0 0 3px rgba(79, 70, 229, 0.2);
+  outline: none;
+}
+
+.gr-form .gr-form-label {
+  font-weight: 600;
+  color: var(--text-color);
+  margin-bottom: 0.5rem;
+}
+
+/* Adding Dark mode handling */
+/* Generate button - Modified to use header gradient in both light and dark mode */
+.gr-button.gr-button-primary, 
+button.generate-btn,
+#component-54 > div.svelte-1pf7t34 > div > button {
+  background: var(--header-gradient) !important;
+  color: white !important;
+  border: none !important;
+  border-radius: var(--radius) !important;
+  padding: 0.75rem 1.5rem !important;
+  font-size: 1rem !important;
+  font-weight: 600 !important;
+  cursor: pointer !important;
+  transition: var(--transition) !important;
+  width: 100% !important;
+  text-transform: uppercase !important;
+  letter-spacing: 1px !important;
+  box-shadow: 0 4px 6px -1px rgba(79, 70, 229, 0.4) !important;
+}
+
+.gr-button.gr-button-primary:hover,
+button.generate-btn:hover,
+#component-54 > div.svelte-1pf7t34 > div > button:hover {
+  transform: translateY(-2px) !important;
+  box-shadow: 0 10px 15px -3px rgba(79, 70, 229, 0.4) !important;
+}
+
+.gr-button.gr-button-primary:active,
+button.generate-btn:active,
+#component-54 > div.svelte-1pf7t34 > div > button:active {
+  transform: translateY(1px) !important;
+}
+
+/* Gallery output */
+.gr-gallery {
+  background-color: var(--card-bg);
+  border-radius: var(--radius);
+  padding: 1rem;
+  box-shadow: var(--shadow);
+  min-height: 300px;
+}
+
+.gr-gallery img {
+  border-radius: calc(var(--radius) - 4px);
+  transition: var(--transition);
+  object-fit: contain !important;
+  max-height: 100% !important;
+  max-width: 100% !important;
+}
+
+.gr-gallery img:hover {
+  transform: scale(1.02);
+}
+
+/* Output text area */
+.output-text {
+  background-color: var(--card-bg);
+  border-radius: var(--radius);
+  padding: 1.5rem;
+  box-shadow: var(--shadow);
+  line-height: 1.6;
+  min-height: 100px;
+  color: var(--text-color);
+}
+
+/* Examples section - Fixed for dark mode */
+.gr-examples-header {
+  font-weight: 600 !important;
+  margin: 2rem 0 1rem !important;
+  padding-bottom: 0.5rem !important;
+  /* Removed border-bottom to avoid duplicate lines */
+  color: var(--primary-light) !important;
+  font-size: 1.5rem !important;
+  display: block !important;
+  visibility: visible !important;
+}
+
+/* Handling duplicate lines coming up below the examples header */
+.gradio-container hr {
+  display: none !important;
+}
+
+/* Fix horizontal separator styling if needed */
+.gradio-container hr:first-of-type {
+  display: block !important;
+  border-top: 2px solid var(--primary-light) !important;
+  margin: 1rem 0 2rem 0 !important;
+}
+
+/* Fix dark mode examples grid */
+.gr-examples {
+  display: grid;
+  grid-template-columns: repeat(auto-fill, minmax(200px, 1fr));
+  gap: 1rem;
+  margin-top: 1.5rem;
+}
+
+/* Fix dark mode examples background */
+.gr-sample {
+  background-color: var(--card-bg) !important;
+  border-radius: var(--radius);
+  overflow: hidden;
+  box-shadow: var(--shadow);
+  transition: var(--transition);
+  cursor: pointer;
+}
+
+.gr-sample:hover {
+  transform: translateY(-2px);
+  box-shadow: 0 10px 15px -3px rgba(0, 0, 0, 0.1);
+}
+
+.gr-sample img {
+  width: 100%;
+  height: 150px;
+  object-fit: cover;
+}
+
+/* Examples table styling fixes for dark mode */
+table {
+  background-color: transparent !important;
+  color: var(--text-color) !important;
+}
+
+table tr {
+  background-color: var(--card-bg) !important;
+  color: var(--text-color) !important;
+}
+
+table td {
+  background-color: var(--card-bg) !important;
+  color: var(--text-color) !important;
+  border-color: var(--border-color) !important;
+}
+
+/* Responsive adjustments */
+@media (max-width: 768px) {
+  .gradio-container {
+    padding: 1rem;
+  }
+  
+  .header-container {
+    flex-direction: column;
+    text-align: center;
+    padding: 1.5rem;
+  }
+  
+  .header-container h1 {
+    font-size: 2rem;
+  }
+}
+
+/* Custom scrollbar */
+::-webkit-scrollbar {
+  width: 8px;
+  height: 8px;
+}
+
+::-webkit-scrollbar-track {
+  background: var(--bg-color);
+}
+
+::-webkit-scrollbar-thumb {
+  background: var(--primary-light);
+  border-radius: 4px;
+}
+
+::-webkit-scrollbar-thumb:hover {
+  background: var(--primary-color);
+}
+
+/* Ensure proper viewport handling for mobile */
+@media screen and (max-width: 767px) {
+  body, html {
+    overflow-x: hidden !important;
+    width: 100vw !important;
+    max-width: 100% !important;
+  }
+  
+  /* Force central alignment for the entire app container */
+  .gradio-container {
+    padding: 1rem !important;
+    margin: 0 auto !important;
+    max-width: 100% !important;
+    width: 100% !important;
+    left: 0 !important;
+    right: 0 !important;
+    box-sizing: border-box !important;
+    overflow-x: hidden !important;
+  }
+  
+  /* Fix header container on mobile */
+  .header-container {
+    flex-direction: column !important;
+    text-align: center !important;
+    padding: 1rem !important;
+    width: 100% !important;
+    box-sizing: border-box !important;
+    margin-left: auto !important;
+    margin-right: auto !important;
+  }
+  
+  /* Ensure all content blocks are properly centered */
+  .main-content, 
+  .input-column, 
+  .output-column,
+  .gr-form,
+  .gr-panel,
+  .gr-box,
+  .gr-input,
+  .gr-text-input,
+  .gr-gallery,
+  .gr-button,
+  #component-54 > div.svelte-1pf7t34 > div > button {
+    width: 100% !important;
+    max-width: 100% !important;
+    margin-left: auto !important;
+    margin-right: auto !important;
+    box-sizing: border-box !important;
+  }
+  
+ 
+  /* Fix examples area */
+  .gr-examples {
+    grid-template-columns: 1fr !important;
+    width: 100% !important;
+  }
+  
+  /* Fix example tables on mobile */
+  table {
+    table-layout: fixed !important;
+    width: 100% !important;
+  }
+  
+  table td {
+    word-break: break-word !important;
+  }
+}
+
+/* Fix for extreme narrow screens (small Android phones) */
+@media screen and (max-width: 380px) {
+  .header-container h1 {
+    font-size: 1.8rem !important;
+  }
+  
+  .header-container p {
+    font-size: 0.9rem !important;
+  }
+  
+  /* Stack everything vertically */
+  .gr-panel {
+    display: flex !important;
+    flex-direction: column !important;
+  }
+}
+
+/* Custom fix for the container to always stay centered on any device */
+#gradio-app {
+  display: flex !important;
+  justify-content: center !important;
+  width: 100% !important;
+  max-width: 100% !important;
+  margin: 0 auto !important;
+  overflow-x: hidden !important;
+}
+
+/* Ensure the root container doesn't cause horizontal scroll */
+.root {
+  max-width: 100% !important;
+  overflow-x: hidden !important;
+}