Update app.py

app.py

@@ -7,21 +7,27 @@ from PIL import Image, ImageDraw, ImageFont
 import gradio as gr
 import base64
 import mimetypes
-
 from google import genai
 from google.genai import types
 
+# Функція для збереження бінарного файлу (залишаємо без змін)
 def save_binary_file(file_name, data):
     with open(file_name, "wb") as f:
         f.write(data)
 
-def generate(text, file_name, api_key, model="gemini-2.0-flash-exp"):
-    # 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")))
-    
+# Модифікована функція generate - прибираємо api_key як параметр
+def generate(text, file_name, model="gemini-2.0-flash-exp"):
+    # Ініціалізуємо клієнта, читаючи ключ зі змінної оточення geminigoogle
+    # Переконайтеся, що змінна geminigoogle встановлена у ваших налаштуваннях Space
+    api_key = os.environ.get("geminigoogle")
+    if not api_key:
+        raise ValueError("GEMINI_API_KEY environment variable (geminigoogle) not set.")
+
+    client = genai.Client(api_key=api_key)
+
+    # Решта функції generate залишається без змін
     files = [ client.files.upload(file=file_name) ]
-    
+
     contents = [
         types.Content(
             role="user",
@@ -34,6 +40,7 @@ def generate(text, file_name, api_key, model="gemini-2.0-flash-exp"):
             ],
         ),
     ]
+
     generate_content_config = types.GenerateContentConfig(
         temperature=1,
         top_p=0.95,
@@ -45,7 +52,7 @@ def generate(text, file_name, api_key, model="gemini-2.0-flash-exp"):
 
     text_response = ""
     image_path = None
-    # Create a temporary file to potentially store image data.
+
     with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
         temp_path = tmp.name
         for chunk in client.models.generate_content_stream(
@@ -56,49 +63,50 @@ def generate(text, file_name, api_key, model="gemini-2.0-flash-exp"):
             if not chunk.candidates or not chunk.candidates[0].content or not chunk.candidates[0].content.parts:
                 continue
             candidate = chunk.candidates[0].content.parts[0]
-            # Check for inline image data
+
             if candidate.inline_data:
                 save_binary_file(temp_path, candidate.inline_data.data)
                 print(f"File of mime type {candidate.inline_data.mime_type} saved to: {temp_path} and prompt input: {text}")
                 image_path = temp_path
-                # If an image is found, we assume that is the desired output.
                 break
             else:
-                # Accumulate text response if no inline_data is present.
                 text_response += chunk.text + "\n"
-    
-    del files
+
+        # Видаляємо завантажені файли після використання
+        del files
+
     return image_path, text_response
 
-def process_image_and_prompt(composite_pil, prompt, gemini_api_key):
+# Модифікована функція process_image_and_prompt - прибираємо gemini_api_key як параметр
+def process_image_and_prompt(composite_pil, prompt):
     try:
-        # Save the composite image to a temporary file.
         with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
             composite_path = tmp.name
             composite_pil.save(composite_path)
-        
-        file_name = composite_path  
-        input_text = prompt 
-        model = "gemini-2.0-flash-exp" 
 
-        image_path, text_response = generate(text=input_text, file_name=file_name, api_key=gemini_api_key, model=model)
-        
+        file_name = composite_path
+        input_text = prompt
+        model = "gemini-2.0-flash-exp" # Модель вказується тут
+
+        # Викликаємо generate без api_key
+        image_path, text_response = generate(text=input_text, file_name=file_name, 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("RGB")
-            return [result_img], ""  # Return image in gallery and empty text output.
+            return [result_img], ""
         else:
-            # Return no image and the text response.
             return None, text_response
+
     except Exception as e:
+        # Важливо видалити тимчасовий файл у разі помилки
+        if 'composite_path' in locals() and os.path.exists(composite_path):
+             os.remove(composite_path)
         raise gr.Error(f"Error Getting {e}", duration=5)
 
-
-# Build a Blocks-based interface with a custom HTML header and CSS
+# Gradio інтерфейс
 with gr.Blocks(css_paths="style.css",) as demo:
-    # Custom HTML header with proper class for styling
     gr.HTML(
     """
     <div class="header-container">
@@ -107,32 +115,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>
     """
     )
-    
+
+    # Прибираємо секцію API Configuration або змінюємо її опис, оскільки ключ більше не вводиться
     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!  
+    - **Ваш Gemini API ключ має бути збережений у змінній оточення `geminigoogle` в налаштуваннях Hugging Face Space.**
+    - ❗ Іноді модель повертає текст замість зображення.
     """)
 
     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!**
+    ### 📌 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"):
@@ -144,12 +149,13 @@ with gr.Blocks(css_paths="style.css",) as demo:
                 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"
-            )
+            # Прибираємо поле введення API ключа з інтерфейсу
+            # 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...",
@@ -157,38 +163,39 @@ 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"):
             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"
             )
 
-    # Set up the interaction with two outputs.
+    # Налаштовуємо взаємодію - прибираємо gemini_api_key з inputs
     submit_btn.click(
         fn=process_image_and_prompt,
-        inputs=[image_input, prompt_input, gemini_api_key],
+        inputs=[image_input, prompt_input], # Передаємо лише image_input та prompt_input
         outputs=[output_gallery, output_text],
     )
-    
+
     gr.Markdown("## Try these examples", elem_classes="gr-examples-header")
-    
+
+    # Приклади залишаємо без змін, API ключ в них не потрібен
     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", ""],
+        ["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,],
+        inputs=[image_input, prompt_input], # Приклади також не потребують API ключа
         elem_id="examples-grid"
     )