Update main.py

main.py

@@ -802,28 +802,37 @@ def submit_quiz_attempt(quiz_id):
 @app.route('/api/tutor/notes/<uuid:notes_id>/speak', methods=['GET'])
 def speak_notes(notes_id):
     user, error = verify_token(request.headers.get('Authorization'))
-    if error: return jsonify({'error': error['error']}), error['status']
-    if not supabase or not elevenlabs_client: return jsonify({'error': 'Backend service unavailable'}), 503
+    if error:
+        return jsonify({'error': error['error']}), error['status']
 
-    profile_res = supabase.table('profiles').select('credits', 'suspended').eq('id', user.id).single().execute()
-    if profile_res.data['suspended']:
-        return jsonify({'error': 'Account suspended'}), 403
-    if profile_res.data['credits'] < 5:
-        return jsonify({'error': 'Insufficient credits (Need 5)'}), 402
+    if not supabase or not elevenlabs_client:
+        return jsonify({'error': 'Backend service unavailable'}), 503
+
+    try:
+        profile_res = supabase.table('profiles').select('credits', 'suspended').eq('id', user.id).single().execute()
+
+        if profile_res.data['suspended']:
+            return jsonify({'error': 'Account suspended'}), 403
+        if profile_res.data['credits'] < 5:
+            return jsonify({'error': 'Insufficient credits (Need 5)'}), 402
+
+        # Fetch notes content
+        notes_res = supabase.table('notes').select('content').eq('id', notes_id).single().execute()
+        if not notes_res.data or 'content' not in notes_res.data:
+            return jsonify({'error': 'Notes not found or missing content'}), 404
 
         notes_content = notes_res.data['content']
         if not notes_content:
-             return jsonify({'error': 'Notes content is empty, cannot generate audio.'}), 400
+            return jsonify({'error': 'Notes content is empty, cannot generate audio.'}), 400
 
         # --- Generate TTS Audio ---
         start_time = time.time()
         logging.info(f"Generating TTS for user {user.id}, notes: {notes_id}")
-        audio_bytes = generate_tts_audio(notes_content) # Add voice selection if needed
+        audio_bytes = generate_tts_audio(notes_content)  # Add voice selection if needed
         logging.info(f"TTS generation took {time.time() - start_time:.2f}s")
 
         # --- Save Audio to Supabase Storage ---
-        # Naming convention: users/{user_id}/notes_audio/{notes_id}.mp3
-        bucket_name = 'notes-audio' # Ensure this bucket exists in Supabase Storage
+        bucket_name = 'notes-audio'  # Ensure this bucket exists in Supabase Storage
         destination_path = f'users/{user.id}/{notes_id}.mp3'
         content_type = 'audio/mpeg'
 
@@ -836,29 +845,29 @@ def speak_notes(notes_id):
             audio_url = upload_to_supabase_storage(bucket_name, tmp_file_path, destination_path, content_type)
             logging.info(f"Uploaded TTS audio to: {audio_url}")
 
-             # --- Update notes table with the URL ---
+            # --- Update notes table with the URL ---
             supabase.table('notes').update({'tts_audio_url': audio_url}).eq('id', notes_id).execute()
+
             # Deduct 5 credits after successful generation
             new_credits = profile_res.data['credits'] - 5
             supabase.table('profiles').update({'credits': new_credits}).eq('id', user.id).execute()
 
-            # OR: Return the URL
             return jsonify({'success': True, 'audio_url': audio_url})
 
         finally:
-             os.remove(tmp_file_path) # Clean up temporary file
-
+            os.remove(tmp_file_path)  # Clean up temporary file
 
     except ConnectionError as e:
         logging.error(f"Connection error during TTS generation: {e}")
         return jsonify({'error': f'A backend service is unavailable: {e}'}), 503
-    except RuntimeError as e: # AI generation errors
-         logging.error(f"RuntimeError during TTS generation for user {user.id}: {e}")
-         return jsonify({'error': str(e)}), 500
+    except RuntimeError as e:  # AI generation errors
+        logging.error(f"RuntimeError during TTS generation for user {user.id}: {e}")
+        return jsonify({'error': str(e)}), 500
     except Exception as e:
         logging.error(f"Unexpected error generating TTS for user {user.id}, notes {notes_id}: {traceback.format_exc()}")
         return jsonify({'error': f'An unexpected error occurred: {e}'}), 500
 
+
 @app.route('/api/user/performance', methods=['GET'])
 def get_user_performance():
     """Retrieves user's quiz performance and provides simple suggestions."""