Update image_gen.py

image_gen.py

@@ -31,7 +31,6 @@ import dataframe_image as dfi
 import uuid
 from PIL import ImageFont, ImageDraw, Image
 import seaborn as sns
-import google.generativeai as ggenai
 
 
 logging.basicConfig(level=logging.INFO)
@@ -42,16 +41,7 @@ headers = {"Authorization": f"Bearer {token}"}
 
 GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
 
-def configure_gemini(api_key):
-    try:
-        ggenai.configure(api_key=api_key)
-        return genai.GenerativeModel('gemini-2.0-flash-thinking-exp')
-    except Exception as e:
-        logger.error(f"Error configuring Gemini: {str(e)}")
-        raise
 
-# Initialize Gemini model for story generation
-model_g = configure_gemini(GOOGLE_API_KEY)
 
 def is_valid_png(file_path):
     """Check if the PNG file at `file_path` is valid."""
@@ -187,7 +177,7 @@ def generate_image(prompt_text, style, model="hf"):
             headers_pexels = {
                 "Authorization": pexels_api_key
             }
-            new_prompt = model_g.generate_content(f"Generate keywords or a phrase to search for an appropriate image on the pexels API based on this prompt: {prompt_text} return only the keywords and nothing else")
+            new_prompt = client.models.generate_content(model="models/gemini-2.0-flash-lite", contents=f"Generate keywords or a phrase to search for an appropriate image on the pexels API based on this prompt: {prompt_text} return only the keywords and nothing else")
             pexel_prompt = str(new_prompt.text)
         
             search_url = f"https://api.pexels.com/v1/search?query={pexel_prompt}&per_page=1"