fix funscript next action

Author: arch

funscript_editor/data/funscript.py

@@ -175,10 +175,11 @@ def get_next_action(self, current_timestamp: int) -> dict:
         Returns:
             dict: action dictionary with {'pos', 'at'}
         """
-        if len(self.data['actions']) < 2: return {'pos': 0, 'at': 0}
+        if len(self.data['actions']) < 1: return {'pos': 0, 'at': 0}
         idx = (np.abs(np.array(self.get_actions_times()) - current_timestamp)).argmin()
         if self.data['actions'][idx]['at'] > current_timestamp + 1: return self.data['actions'][idx]
-        elif idx < len(self.data['actions']) + 1: return self.data['actions'][idx+1]
+        elif len(self.data['actions']) == 1: return self.data['actions'][0]
+        elif idx+1 < len(self.data['actions']): return self.data['actions'][idx+1]
         else: return self.data['actions'][0]
 
 

funscript_editor/ui/funscript_editor_window.py

@@ -298,6 +298,8 @@ def __generate_funscript(self):
         else:
             end_frame = -1
 
+        self.__logger.info("Stop at Frame {}".format(end_frame))
+
         reply = QtWidgets.QMessageBox.question(None, 'Track Men', 'Do you want to track the Men? ',
                 QtWidgets.QMessageBox.Yes | QtWidgets.QMessageBox.No, QtWidgets.QMessageBox.No)
         trackMen = True if reply == QtWidgets.QMessageBox.Yes else False