Upload app.py

app.py

@@ -14,10 +14,6 @@ def pil_to_bytes(img, format="PNG"):
     img.save(img_byte_arr, format=format)
     return img_byte_arr.getvalue()
 
-# Function to save API key in browser local storage
-def save_api_key(api_key):
-    return api_key
-
 # Function to load image as base64
 async def load_image_base64(img):
     if isinstance(img, str):
@@ -28,7 +24,7 @@ async def load_image_base64(img):
         return pil_to_bytes(img)
 
 # Main function to generate edited image using Gemini
-async def generate_image_gemini(prompt, image, api_key):
+async def generate_image_gemini(prompt, image, api_key, temperature=0.4):
     SAFETY_SETTINGS = {
         types.HarmCategory.HARM_CATEGORY_HARASSMENT: types.HarmBlockThreshold.BLOCK_NONE,
         types.HarmCategory.HARM_CATEGORY_HATE_SPEECH: types.HarmBlockThreshold.BLOCK_NONE,
@@ -73,6 +69,7 @@ async def generate_image_gemini(prompt, image, api_key):
             model="gemini-2.0-flash-exp",
             contents=contents,
             config=types.GenerateContentConfig(
+                temperature=temperature,
                 safety_settings=[
                     types.SafetySetting(
                         category=category, threshold=threshold
@@ -100,7 +97,7 @@ async def generate_image_gemini(prompt, image, api_key):
         return None
 
 # Function to process the image edit
-def process_image_edit(image, prompt, api_key, image_history):
+def process_image_edit(image, prompt, api_key, image_history, temperature):
     if not image or not prompt or not api_key:
         return None, image_history, "Please provide an image, prompt, and API key"
     
@@ -112,7 +109,7 @@ def process_image_edit(image, prompt, api_key, image_history):
     
     # Run the async function to edit the image
     try:
-        edited_image = asyncio.run(generate_image_gemini(prompt, image, api_key))
+        edited_image = asyncio.run(generate_image_gemini(prompt, image, api_key, temperature))
         if edited_image:
             return edited_image, image_history, "Image edited successfully"
         else:
@@ -132,46 +129,35 @@ def undo_edit(image_history):
 
 # Create Gradio UI
 def create_ui():
-    with gr.Blocks(title="Gemini Image Editor", js="""
-    function() {
-        // Wait for the DOM to be fully loaded
-        window.addEventListener('DOMContentLoaded', (event) => {
-            setTimeout(function() {
-                // Try to load saved API key from localStorage
-                const savedKey = localStorage.getItem('gemini_api_key');
-                const apiKeyInput = document.querySelector('input[placeholder="Enter your Gemini API key"]');
-                if (savedKey && apiKeyInput) {
-                    apiKeyInput.value = savedKey;
-                    // Dispatch an input event to make sure Gradio recognizes the change
-                    const event = new Event('input', { bubbles: true });
-                    apiKeyInput.dispatchEvent(event);
-                }
-                
-                // Add event listener to save API key when checkbox is checked
-                const saveKeyCheckbox = document.querySelector('input[type="checkbox"]');
-                if (apiKeyInput && saveKeyCheckbox) {
-                    apiKeyInput.addEventListener('change', function(e) {
-                        if (saveKeyCheckbox.checked) {
-                            localStorage.setItem('gemini_api_key', e.target.value);
-                        }
-                    });
-                }
-            }, 1000); // Small delay to ensure elements are loaded
-        });
-    }
-    """) as app:
+    with gr.Blocks(title="Gemini Image Editor") as app:
         gr.Markdown("# Gemini Image Editor")
         gr.Markdown("Upload an image, enter a description of the edit you want, and let Gemini do the rest!")
         
         # Store image history in state
         image_history = gr.State([])
         
+        # Settings visibility state
+        settings_visible = gr.State(False)
+        
         with gr.Row():
             with gr.Column():
                 input_image = gr.Image(type="pil", label="Upload Image")
                 prompt = gr.Textbox(label="Edit Description", placeholder="Describe the edit you want...")
                 api_key = gr.Textbox(label="Gemini API Key", placeholder="Enter your Gemini API key", type="password")
-                save_key = gr.Checkbox(label="Save API key in browser", value=True)
+                
+                # Settings button and panel
+                settings_btn = gr.Button("⚙️ Settings", scale=0.15)
+                
+                with gr.Column(visible=False) as settings_panel:
+                    gr.Markdown("### Advanced Settings")
+                    temperature = gr.Slider(
+                        minimum=0.0, 
+                        maximum=2.0, 
+                        value=0.4, 
+                        step=0.05, 
+                        label="Temperature", 
+                        info="Controls randomness in generation (0 = deterministic, 1 = creative, 2 = extreme)"
+                    )
                 
                 with gr.Row():
                     edit_btn = gr.Button("Edit Image")
@@ -181,10 +167,20 @@ def create_ui():
                 output_image = gr.Image(type="pil", label="Edited Image")
                 status = gr.Textbox(label="Status", interactive=False)
         
+        # Toggle settings visibility
+        def toggle_settings(visible):
+            return not visible, not visible
+        
+        settings_btn.click(
+            fn=toggle_settings,
+            inputs=[settings_visible],
+            outputs=[settings_visible, settings_panel]
+        )
+        
         # Set up event handlers
         edit_btn.click(
             fn=process_image_edit,
-            inputs=[input_image, prompt, api_key, image_history],
+            inputs=[input_image, prompt, api_key, image_history, temperature],
             outputs=[output_image, image_history, status]
         )