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 # CORRECT: Import the genai module
+from google import genai
 
 # --- Basic Configuration ---
 logging.basicConfig(level=logging.INFO)
@@ -46,27 +46,26 @@ 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.")
     
-    # --- NEW & CORRECT SDK PATTERN ---
-    # Instantiate the client object, which is the modern, recommended approach.
+    # --- CORRECTED SDK PATTERN, AS PER OFFICIAL DOCUMENTATION ---
     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 ---
+    # This is the correct method to get a model object for generation.
+    gemini_model = client.generative_model(model_name='gemini-2.0-flash')
+    # --- END OF CORRECTION ---
 
-    logger.info("Google GenAI Client initialized successfully using the modern client pattern.")
+    logger.info("Google GenAI Client initialized successfully using the correct SDK 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
     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]
@@ -102,14 +101,11 @@ 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'. 
@@ -132,7 +128,6 @@ 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':
@@ -143,7 +138,6 @@ 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}')
@@ -199,7 +193,6 @@ 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()
@@ -239,7 +232,6 @@ def generate_agent_briefing(uid, project_id):
 # -----------------------------------------------------------------------------
 # 4. USER & AUTHENTICATION ENDPOINTS
 # -----------------------------------------------------------------------------
-# ... [No changes in this section] ...
 @app.route('/api/auth/signup', methods=['POST'])
 def signup():
     try:
@@ -294,7 +286,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'))
@@ -448,7 +440,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'))
@@ -512,7 +504,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: