Update app.py

app.py

@@ -6,7 +6,6 @@ import tempfile
 import uuid
 from pathlib import Path
 
-# Initialize the Gemini client globally
 client = None
 if os.environ.get("GOOGLE_API_KEY"):
     client = genai.Client(api_key=os.environ.get("GOOGLE_API_KEY"))
@@ -19,27 +18,21 @@ def save_binary_file(file_name, data):
 
 
 def process_image_with_gemini(image, instruction) -> tuple[str, str, str]:
-    # Create output directory if it doesn't exist
     output_dir = Path("output_gemini")
     output_dir.mkdir(exist_ok=True)
 
-    # Generate a unique ID for this request
     request_id = f"request_{uuid.uuid4().hex[:8]}"
     request_folder = output_dir / request_id
     request_folder.mkdir(exist_ok=True)
 
-    # Save the input image to the request folder
     input_image_path = request_folder / "input.jpg"
     image.save(input_image_path)
 
     try:
-        # Create a temporary directory that will be automatically cleaned up
         with tempfile.TemporaryDirectory() as temp_dir:
-            # Save the image to a temporary file
             temp_image_path = Path(temp_dir) / "temp_input_image.jpg"
             image.save(temp_image_path)
 
-            # Upload the temporary file to Gemini API using global client
             files = [
                 client.files.upload(file=str(temp_image_path)),
             ]
@@ -69,7 +62,7 @@ def process_image_with_gemini(image, instruction) -> tuple[str, str, str]:
                 safety_settings=[
                     types.SafetySetting(
                         category="HARM_CATEGORY_CIVIC_INTEGRITY",
-                        threshold="OFF",  # Off
+                        threshold="OFF",
                     ),
                 ],
                 response_mime_type="text/plain",
@@ -90,19 +83,15 @@ def process_image_with_gemini(image, instruction) -> tuple[str, str, str]:
                 ):
                     continue
 
-                # Handle image response
                 if hasattr(chunk.candidates[0].content.parts[0], "inline_data"):
-                    # Save the generated image
                     edited_image_path = request_folder / "edited.jpg"
                     save_binary_file(
                         str(edited_image_path),
                         chunk.candidates[0].content.parts[0].inline_data.data,
                     )
-                # Handle text response
                 elif hasattr(chunk.candidates[0].content.parts[0], "text"):
                     response_text += chunk.candidates[0].content.parts[0].text
 
-            # Simplify the return statement and ensure consistent types
             if edited_image_path and edited_image_path.exists():
                 return str(edited_image_path), response_text or "", "Success"
             return None, response_text or "", "No image generated"
@@ -119,11 +108,9 @@ def process_image_with_gemini(image, instruction) -> tuple[str, str, str]:
 
 def process_image(image, instruction):
     """Process an image with Gemini based on given instructions.
-
     Args:
         image: Input PIL image
         instruction: Text instructions for editing
-
     Returns:
         Tuple containing (output_image_path, response_text, status_message)
     """
@@ -146,9 +133,9 @@ def process_image(image, instruction):
         return None, "", f"Unexpected error: {str(e)}"
 
 
-with gr.Blocks(title="Gemini Image Editor") as app:
+with gr.Blocks(title="Image Editor", theme='Jonny001/GreenEarth_Theme') as app:
     with gr.Column():
-        gr.Markdown("# 🖼️ Gemini Image Editor")
+        gr.Markdown("# 🖼️ Image Editor")
         gr.Markdown(
             "Upload an image and provide instructions for Gemini to edit it. The AI will generate a new image based on your instructions."
         )
@@ -176,19 +163,6 @@ with gr.Blocks(title="Gemini Image Editor") as app:
             outputs=[output_image, response_text, status],
         )
 
-        # Add sample instructions (without example images)
-        gr.Markdown("### Sample Instructions (upload your own image and try these)")
-        sample_instructions = gr.Examples(
-            examples=[
-                "Make the sky more blue and add birds flying",
-                "Convert this to a watercolor painting style",
-                "Add a sunset effect to this image",
-                "Turn this into a night scene with stars",
-                "Make this look like it was taken in winter with snow",
-            ],
-            inputs=instruction,
-        )
-
         gr.Markdown(
             """
             ### Notes
@@ -199,7 +173,6 @@ with gr.Blocks(title="Gemini Image Editor") as app:
         )
 
 
-# Launch the app
 if __name__ == "__main__":
     print("Starting Gemini Image Editor...")
-    app.launch(ssr_mode=True)
+    app.launch(ssr_mode=True)