Update main.py

main.py

@@ -893,6 +893,7 @@ def submit_quiz_attempt(quiz_id):
 # Modified speak_notes endpoint with ElevenLabs Studio API and chunking
 @app.route('/api/tutor/notes/<uuid:notes_id>/speak', methods=['POST'])
 def speak_notes(notes_id):
+    """Generate TTS audio for notes and store in Supabase Storage"""
     user, error = verify_token(request.headers.get('Authorization'))
     if error:
         return jsonify({'error': error['error']}), error['status']
@@ -901,7 +902,7 @@ def speak_notes(notes_id):
         return jsonify({'error': 'Backend service unavailable'}), 503
 
     try:
-        # Verify note ownership first
+        # 1. Verify note ownership and get content
         note_res = supabase.table('notes') \
             .select('user_id, content, tts_audio_url') \
             .eq('id', notes_id) \
@@ -912,18 +913,19 @@ def speak_notes(notes_id):
         if not note_res.data:
             return jsonify({'error': 'Note not found or unauthorized'}), 404
 
-        # Check user status and credits
+        # 2. Check user status and credits
         profile_res = supabase.table('profiles') \
             .select('credits, suspended') \
             .eq('id', user.id) \
-            .single().execute()
-            
+            .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
 
-        # Return existing audio if available
+        # 3. Return existing audio if available
         if note_res.data.get('tts_audio_url'):
             return jsonify({
                 'success': True,
@@ -935,7 +937,7 @@ def speak_notes(notes_id):
         if not notes_content:
             return jsonify({'error': 'Notes content is empty'}), 400
 
-        # --- Generate TTS Audio with chunking ---
+        # 4. Generate TTS Audio with chunking
         CHUNK_SIZE = 2000
         chunks = [notes_content[i:i+CHUNK_SIZE] for i in range(0, len(notes_content), CHUNK_SIZE)]
         
@@ -956,30 +958,30 @@ def speak_notes(notes_id):
         if not audio_bytes:
             raise RuntimeError("Generated empty audio file")
 
-        # --- Save to Supabase Storage with RLS compliance ---
+        # 5. Save to Supabase Storage
         bucket_name = 'notes-audio'
         file_path = f'{user.id}/{notes_id}.mp3'
         
         try:
-            # Upload with owner metadata for RLS
+            # Upload audio file
             upload_res = supabase.storage.from_(bucket_name).upload(
                 path=file_path,
                 file=audio_bytes,
                 file_options={
                     'content-type': 'audio/mpeg',
                     'cache-control': '3600',
-                    'upsert': 'true',
-                    'owner': user.id  
+                    'upsert': 'true'
                 }
             )
             
-            if upload_res.error:
-                raise ConnectionError(upload_res.error.message)
+            # Verify upload success (storage upload returns None on success)
+            if upload_res is not None:
+                raise ConnectionError("Audio upload failed")
 
             # Get public URL
             audio_url = supabase.storage.from_(bucket_name).get_public_url(file_path)
 
-            # Update notes table with verification
+            # 6. Update database records
             update_res = supabase.table('notes') \
                 .update({'tts_audio_url': audio_url}) \
                 .eq('id', notes_id) \
@@ -989,7 +991,7 @@ def speak_notes(notes_id):
             if update_res.error:
                 raise ConnectionError(update_res.error.message)
 
-            # Deduct credits with verification
+            # 7. Deduct credits
             new_credits = profile_res.data['credits'] - 5
             credit_res = supabase.table('profiles') \
                 .update({'credits': new_credits}) \
@@ -1007,11 +1009,16 @@ def speak_notes(notes_id):
 
         except Exception as upload_error:
             # Clean up failed upload
-            supabase.storage.from_(bucket_name).remove([file_path])
+            try:
+                supabase.storage.from_(bucket_name).remove([file_path])
+            except Exception as cleanup_error:
+                logging.error(f"Cleanup failed: {cleanup_error}")
+            
+            logging.error(f"Upload failed: {str(upload_error)}")
             raise upload_error
 
     except Exception as e:
-        logging.error(f"Speak error: {traceback.format_exc()}")
+        logging.error(f"Speak endpoint error: {traceback.format_exc()}")
         return jsonify({'error': str(e)}), 500
 
 # New endpoint to view existing audio URL