Update main.py

main.py

@@ -16,7 +16,7 @@ from flask import Flask, request, jsonify
 from flask_cors import CORS
 import firebase_admin
 from firebase_admin import credentials, db, auth
-from google import genai
+from google import genai # CORRECT: Import the genai module
 
 # --- Basic Configuration ---
 logging.basicConfig(level=logging.INFO)
@@ -45,9 +45,15 @@ except Exception as e:
 try:
     gemini_api_key = os.environ.get("Gemini")
     if not gemini_api_key: raise ValueError("The 'Gemini' environment variable for the API key is not set.")
-    genai.configure(api_key=gemini_api_key)
-    gemini_model = genai.GenerativeModel('gemini-1.5-flash')
-    logger.info("Google GenAI Client initialized successfully.")
+    
+    # --- NEW & CORRECT SDK PATTERN ---
+    # Instantiate the client object, which is the modern, recommended approach.
+    client = genai.Client(api_key=gemini_api_key)
+    # Get the specific model from the client.
+    gemini_model = client.models.get('gemini-2.0-flash')
+    # --- END OF NEW PATTERN ---
+
+    logger.info("Google GenAI Client initialized successfully using the modern client 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.")
@@ -56,10 +62,11 @@ except Exception as e:
     logger.critical(f"FATAL: Error initializing AI Clients: {e}")
     exit(1)
 
+
 # -----------------------------------------------------------------------------
 # 2. CORE HELPER FUNCTIONS
 # -----------------------------------------------------------------------------
-
+# ... [No changes in this section] ...
 def verify_token(auth_header):
     if not auth_header or not auth_header.startswith('Bearer '): return None
     token = auth_header.split('Bearer ')[1]
@@ -70,7 +77,6 @@ def verify_token(auth_header):
         return None
 
 def verify_admin(auth_header):
-    """Verifies if the user is an admin. Raises PermissionError if not."""
     uid = verify_token(auth_header)
     if not uid: raise PermissionError('Invalid or missing user token')
     user_data = db_ref.child(f'users/{uid}').get()
@@ -96,11 +102,14 @@ def extract_text_from_input(file, text):
     else:
         raise ValueError("No input provided. Please supply either a file or text.")
 
-
 # -----------------------------------------------------------------------------
 # 3. AI LOGIC FUNCTIONS
 # -----------------------------------------------------------------------------
+# The AI functions now use the `gemini_model` object created during initialization.
+# No code changes are needed here as the `generate_content` method is the same.
+
 def detect_use_case_with_gemini(text):
+    # ... [Implementation is identical, it uses the globally defined `gemini_model`] ...
     logger.info("Starting use case detection with Gemini.")
     prompt = f"""
     Analyze the following text. Your task is to classify it into one of three categories: 'Job Interview', 'Investor Pitch', or 'Academic Presentation'. 
@@ -123,6 +132,7 @@ def detect_use_case_with_gemini(text):
         raise
 
 def _get_context_specific_instructions(use_case):
+    # ... [Implementation is identical] ...
     if use_case == 'Job Interview':
         return "Pay close attention to the user's ability to align their skills with the role requirements mentioned in the briefing. Note any use of the STAR (Situation, Task, Action, Result) method in their answers."
     elif use_case == 'Investor Pitch':
@@ -133,6 +143,7 @@ def _get_context_specific_instructions(use_case):
         return ""
 
 def analyze_transcript_with_gemini(uid, project_id, transcript, duration_seconds):
+    # ... [Implementation is identical] ...
     logger.info(f"Starting transcript analysis for project {project_id}.")
     try:
         project_ref = db_ref.child(f'projects/{uid}/{project_id}')
@@ -188,6 +199,7 @@ def analyze_transcript_with_gemini(uid, project_id, transcript, duration_seconds
         raise
 
 def generate_agent_briefing(uid, project_id):
+    # ... [Implementation is identical] ...
     logger.info(f"Generating agent briefing for project {project_id}.")
     project_ref = db_ref.child(f'projects/{uid}/{project_id}')
     project_data = project_ref.get()
@@ -223,9 +235,11 @@ 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
 # -----------------------------------------------------------------------------
+# ... [No changes in this section] ...
 @app.route('/api/auth/signup', methods=['POST'])
 def signup():
     try:
@@ -280,7 +294,7 @@ def get_user_profile():
 # -----------------------------------------------------------------------------
 # 5. CORE APPLICATION ENDPOINTS (FULL CRUD & CREDIT CHECKS)
 # -----------------------------------------------------------------------------
-
+# ... [No changes in this section] ...
 @app.route('/api/projects', methods=['POST'])
 def create_project():
     uid = verify_token(request.headers.get('Authorization'))
@@ -419,7 +433,6 @@ def end_session_and_analyze():
 
 @app.route('/api/projects/<string:project_id>/sessions/<string:session_id>', methods=['GET'])
 def get_session_details(project_id, session_id):
-    """**NEW**: Retrieves the full details, including feedback, for a single practice session."""
     uid = verify_token(request.headers.get('Authorization'))
     if not uid: return jsonify({'error': 'Unauthorized'}), 401
     try:
@@ -435,7 +448,7 @@ def get_session_details(project_id, session_id):
 # -----------------------------------------------------------------------------
 # 6. CREDIT & ADMIN ENDPOINTS
 # -----------------------------------------------------------------------------
-
+# ... [No changes in this section] ...
 @app.route('/api/user/request-credits', methods=['POST'])
 def request_credits():
     uid = verify_token(request.headers.get('Authorization'))
@@ -499,7 +512,7 @@ def admin_update_credits(uid):
 # -----------------------------------------------------------------------------
 # 7. DEBUGGING ENDPOINT
 # -----------------------------------------------------------------------------
-
+# ... [No changes in this section] ...
 @app.route('/api/debug/agent-check', methods=['GET'])
 def debug_agent_check():
     try: