Update app.py

app.py

@@ -81,17 +81,18 @@ def extract_audio_from_video(video_path, output_audio_path):
     except Exception as e:
         raise Exception(f"Failed to extract audio from video: {str(e)}")
 
-def send_fcm_data_message(fcm_token, transcription, file_type, created_date, transcription_name):
+def send_fcm_data_message(fcm_token, transcription, file_type, created_date, transcription_name, message, status):
     """Send an FCM message with transcription details and a notification"""
     try:
         message = messaging.Message(
             notification=messaging.Notification(
                 title=transcription_name,
-                body="Successfully downloaded"
+                body=message
             ),
             data={
-                'transcription': transcription,
-                'file_type': file_type,
+                'status': status,
+                'transcription': transcription if transcription else '',
+                'file_type': file_type if file_type else '',
                 'created_date': created_date,
                 'transcription_name': transcription_name
             },
@@ -175,12 +176,12 @@ def transcribe():
         file_type = 'video' if file_extension in ALLOWED_VIDEO_EXTENSIONS else 'audio'
         
         # Send FCM data message
-    # Send FCM data message
-        send_fcm_data_message(fcm_token, full_text, file_type, created_date, transcription_name)
+        send_fcm_data_message(fcm_token, full_text, file_type, created_date, transcription_name, 'Transcription completed successfully!', 'success)
 
         return jsonify({}), 200
         
     except Exception as e:
+        send_fcm_data_message(fcm_token, full_text, file_type, created_date, transcription_name, 'Transcription failed. Please re-upload the file to try again.','fail')
         return jsonify({'error': str(e)}), 500
         
     finally: