Update image_generation.py

image_generation.py

@@ -15,13 +15,11 @@ try:
     api_key = st.secrets.get("GEMINI_API_KEY") or os.getenv("GEMINI_API_KEY")
     if api_key:
         genai.configure(api_key=api_key)
-        # The documentation uses a client instance.
-        client = genai.Client()
-        print("✅ Google AI client for Gemini initialized successfully.")
+        print("✅ Google AI client for Gemini configured successfully.")
     else:
         print("⚠️ Warning: GEMINI_API_KEY not found.")
 except Exception as e:
-    print(f"❌ Error initializing Google AI client: {e}")
+    print(f"❌ Error configuring Google AI client: {e}")
 
 # --- Helper Function ---
 def save_binary_file(file_name: str, data: bytes):
@@ -33,35 +31,30 @@ def save_binary_file(file_name: str, data: bytes):
         print(f"❌ Error saving file {file_name}: {e}")
 
 # --- IMAGE GENERATION FUNCTION ---
+# --- CORRECTED IMAGE GENERATION FUNCTION ---
 def generate_image_with_gemini(
     prompt: str,
     output_file_base: str,
     context_image: Optional[Image.Image] = None
 ) -> Optional[str]:
     """
-    Generates an image using the exact syntax from the official documentation.
+    Generates an image using the correct syntax for your library version.
     """
-    if not client:
-        print("❌ Gemini client not initialized. Cannot generate image.")
-        return None
-
     print(f"--- 🎨 Generating image for prompt: '{prompt[:70]}...' ---")
     try:
-        model = "gemini-2.5-flash-image-preview"
+        # 1. Create the model instance directly
+        model = genai.GenerativeModel(model_name="gemini-2.5-flash-image-preview")
 
+        # 2. Create the content list
         content_parts = []
-
         if context_image:
-            # For image-to-image, the list contains the prompt and the image.
             print(" -> Using previous image as context for consistent styling.")
             content_parts.extend([prompt, context_image])
         else:
-            # For text-to-image, the list just contains the prompt.
             content_parts.append(prompt)
 
-        # --- API CALL ---
-        response = client.models.generate_content(
-            model=model,
+        # 3. Make the call on the model object
+        response = model.generate_content(
             contents=content_parts,
             stream=True
         )
@@ -69,12 +62,11 @@ def generate_image_with_gemini(
         saved_file_path = None
         text_responses = []
 
-        # --- RESPONSE HANDLING---
+        # 4. Process the response
         for chunk in response:
             if chunk.parts:
                 for part in chunk.parts:
                     if part.inline_data:
-                        # This is an image part
                         data = part.inline_data.data
                         mime_type = part.inline_data.mime_type
                         file_extension = mimetypes.guess_extension(mime_type) or ".jpg"
@@ -83,9 +75,8 @@ def generate_image_with_gemini(
                         save_binary_file(full_file_name, data)
                         saved_file_path = full_file_name
                         print(f"✅ Successfully generated and saved image: {full_file_name}")
-                        break # Image found, exit inner loop
+                        break 
                     elif part.text:
-                        # This is a text part
                         text_responses.append(part.text)
         
         if not saved_file_path and text_responses: