Update app.py

app.py

@@ -10,15 +10,12 @@ from PIL import Image
 # Configure Gemini API key using an environment variable
 genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 
-# Initialize Gemini client
-client = genai.Client()
-
 # Initialize Flask app
 app = Flask(__name__)
 
 def save_image(image_data):
     """Save the image from a base64 string to a temporary file and return its path."""
-    # image_data is expected to be in the format "data:image/png;base64,...."
+    # Expected format: "data:image/png;base64,...."
     header, encoded = image_data.split(',', 1)
     image_bytes = base64.b64decode(encoded)
     image = Image.open(io.BytesIO(image_bytes))
@@ -28,13 +25,13 @@ def save_image(image_data):
 
 def remove_object_from_image(image_path, object_type):
     """Use Gemini API to remove a specified object from the image."""
-    # Upload the image file to Gemini
-    uploaded_file = client.files.upload(file=image_path)
+    # Upload the image file to Gemini using the module-level function.
+    uploaded_file = genai.files.upload(file=image_path)
     
     # Prepare the input parts:
-    # 1. The image file
+    # 1. The image file.
     parts = [types.Part.from_uri(file_uri=uploaded_file.uri, mime_type="image/png")]
-    # 2. The Gemini magic text instructing removal
+    # 2. The Gemini magic text instructing removal.
     if object_type:
         parts.append(types.Part.from_text(text=f"Remove {object_type} from the image"))
     
@@ -51,16 +48,16 @@ def remove_object_from_image(image_path, object_type):
 
     result_image = None
 
-    # Process the generation stream from Gemini
-    for chunk in client.models.generate_content_stream(
+    # Use the module-level function to generate content.
+    for chunk in genai.models.generate_content_stream(
         model="gemini-2.0-flash-exp-image-generation",
         contents=contents,
         config=generate_content_config,
     ):
-        if chunk.candidates and chunk.candidates[0].content and chunk.candidates[0].content.parts:
+        if (chunk.candidates and chunk.candidates[0].content and
+            chunk.candidates[0].content.parts):
             part = chunk.candidates[0].content.parts[0]
             if part.inline_data:
-                # Save the generated binary image data
                 file_name = "generated_output.png"
                 with open(file_name, "wb") as f:
                     f.write(part.inline_data.data)
@@ -80,11 +77,11 @@ def process_image():
     image_data = data['image']
     object_type = data['objectType']
 
-    # Save the uploaded image locally
+    # Save the uploaded image locally.
     image_path = save_image(image_data)
 
     try:
-        # Use Gemini to remove the object from the image
+        # Use Gemini to remove the object from the image.
         result_image = remove_object_from_image(image_path, object_type)
         return jsonify({'success': True, 'resultPath': result_image})
     except Exception as e:
@@ -92,4 +89,4 @@ def process_image():
 
 if __name__ == '__main__':
     # For local testing; in production, your hosting provider will manage the server.
-    app.run(host="0.0.0.0", port=7860)
+    app.run(debug=True)