Update main.py

main.py

@@ -47,19 +47,20 @@ try:
     if not gemini_api_key: raise ValueError("The 'Gemini' environment variable for the API key is not set.")
     
     # --- CORRECTED SDK PATTERN, AS PER OFFICIAL DOCUMENTATION ---
-    client = genai.Client(api_key=gemini_api_key)
-    # This is the correct method to get a model object for generation.
-    gemini_model = client.generative_model(model_name='gemini-2.0-flash')
+    # 1. Configure the API key at the module level.
+    genai.configure(api_key=gemini_api_key)
+    # 2. Create the model instance directly from the genai module.
+    gemini_model = genai.GenerativeModel(model_name='gemini-2.0-flash')
     # --- END OF CORRECTION ---
 
-    logger.info("Google GenAI Client initialized successfully using the correct SDK pattern.")
+    logger.info("Google GenAI Client initialized successfully using the correct GenerativeModel pattern.")
 
     ELEVENLABS_API_KEY = os.environ.get("ELEVENLABS_API_KEY")
     if not ELEVENLABS_API_KEY: raise ValueError("The 'ELEVENLABS_API_KEY' environment variable is not set.")
     logger.info("ElevenLabs API Key loaded.")
 except Exception as e:
     logger.critical(f"FATAL: Error initializing AI Clients: {e}")
-    logger.critical(traceback.format_exc()) # Log the full traceback for debugging
+    logger.critical(traceback.format_exc())
     exit(1)