Update main.py

main.py

@@ -47,13 +47,13 @@ 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 ---
-    # 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')
+    # Instantiate the client object. This is the single, correct entry point.
+    client = genai.Client(api_key=gemini_api_key)
+    # Define the model name to be used in generation calls.
+    MODEL_NAME = 'models/gemini-2.0-flash' 
     # --- END OF CORRECTION ---
 
-    logger.info("Google GenAI Client initialized successfully using the correct GenerativeModel pattern.")
+    logger.info(f"Google GenAI Client initialized successfully for model {MODEL_NAME}.")
 
     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.")
@@ -115,7 +115,8 @@ def detect_use_case_with_gemini(text):
     Text: "{text[:4000]}"
     """
     try:
-        response = gemini_model.generate_content(prompt)
+        # CORRECTED: Use the client object and specify the model name.
+        response = client.generate_content(model=MODEL_NAME, contents=prompt)
         category = response.text.strip().replace("'", "").replace('"', '')
         valid_categories = ['Job Interview', 'Investor Pitch', 'Academic Presentation']
         if category in valid_categories:
@@ -168,7 +169,8 @@ def analyze_transcript_with_gemini(uid, project_id, transcript, duration_seconds
         }}
         Transcript to analyze: "{transcript}"
         """
-        response = gemini_model.generate_content(prompt)
+        # CORRECTED: Use the client object and specify the model name.
+        response = client.generate_content(model=MODEL_NAME, contents=prompt)
         feedback_json_text = response.text.strip().lstrip("```json").rstrip("```")
         feedback_data = json.loads(feedback_json_text)
         session_id = str(uuid.uuid4())
@@ -221,7 +223,8 @@ def generate_agent_briefing(uid, project_id):
         - "The user struggles with concise communication. Ask multi-part questions to test their ability to stay on track."
         Your directive for the agent:
         """
-        response = gemini_model.generate_content(summary_prompt)
+        # CORRECTED: Use the client object and specify the model name.
+        response = client.generate_content(model=MODEL_NAME, contents=summary_prompt)
         dynamic_directive = response.text.strip()
         logger.info(f"Generated dynamic directive for agent: {dynamic_directive}")
         return f"{base_briefing} {dynamic_directive}"
@@ -229,7 +232,6 @@ def generate_agent_briefing(uid, project_id):
         logger.error(f"Could not generate dynamic briefing for project {project_id}: {e}")
         return base_briefing
 
-
 # -----------------------------------------------------------------------------
 # 4. USER & AUTHENTICATION ENDPOINTS
 # -----------------------------------------------------------------------------