Update app.py

app.py

@@ -7,11 +7,7 @@ import base64
 import os
 
 # Initialize the Google Generative AI client with the API key from environment variables
-api_key = os.getenv("GEMINI_API_KEY")
-genai.configure(api_key=api_key)
-
-# Initialize the Gemini model for text generation
-gemini_model = genai.GenerativeModel('gemini-2.5-flash-preview-04-17')
+client = genai.Client(api_key='GEMINI_API_KEY')
 
 def generate_item(tag):
     """
@@ -25,11 +21,11 @@ def generate_item(tag):
     """
     # Generate text using Gemini LLM
     prompt = f"Generate a short, engaging post about {tag} in the style of a TikTok caption."
-    text_response = gemini_model.generate_content(prompt)
+    text_response = client.models.generate_content(model='gemini-2.5-flash-preview-04-17',prompt)
     text = text_response.text.strip()
 
     # Generate an image using Imagen based on the generated text
-    image_response = genai.generate_images(
+    image_response = client.models.generate_images(
         model='imagen-3.0-generate-002',
         prompt=text,
         config=types.GenerateImagesConfig(