fix some label errors during plotting and close plots

InferenceInterfaces/ToucanTTSInterface.py

@@ -177,49 +177,61 @@ class ToucanTTSInterface(torch.nn.Module):
             pass
 
         if view or return_plot_as_filepath:
-            fig, ax = plt.subplots(nrows=1, ncols=1, figsize=(9, 5))
+            try:
+                fig, ax = plt.subplots(nrows=1, ncols=1, figsize=(9, 5))
 
-            ax.imshow(mel.cpu().numpy(), origin="lower", cmap='GnBu')
-            ax.yaxis.set_visible(False)
-            duration_splits, label_positions = cumsum_durations(durations.cpu().numpy())
-            ax.xaxis.grid(True, which='minor')
-            ax.set_xticks(label_positions, minor=False)
-            if input_is_phones:
-                phones = text.replace(" ", "|")
-            else:
-                phones = self.text2phone.get_phone_string(text, for_plot_labels=True)
-            ax.set_xticklabels(phones)
-            word_boundaries = list()
-            for label_index, phone in enumerate(phones):
-                if phone == "|":
-                    word_boundaries.append(label_positions[label_index])
+                ax.imshow(mel.cpu().numpy(), origin="lower", cmap='GnBu')
+                ax.yaxis.set_visible(False)
+                duration_splits, label_positions = cumsum_durations(durations.cpu().numpy())
+                ax.xaxis.grid(True, which='minor')
+                ax.set_xticks(label_positions, minor=False)
+                if input_is_phones:
+                    phones = text.replace(" ", "|")
+                else:
+                    phones = self.text2phone.get_phone_string(text, for_plot_labels=True)
+                try:
+                    ax.set_xticklabels(phones)
+                except IndexError:
+                    pass
+                word_boundaries = list()
+                for label_index, phone in enumerate(phones):
+                    if phone == "|":
+                        word_boundaries.append(label_positions[label_index])
 
-            try:
-                prev_word_boundary = 0
-                word_label_positions = list()
-                for word_boundary in word_boundaries:
-                    word_label_positions.append((word_boundary + prev_word_boundary) / 2)
-                    prev_word_boundary = word_boundary
-                word_label_positions.append((duration_splits[-1] + prev_word_boundary) / 2)
+                try:
+                    prev_word_boundary = 0
+                    word_label_positions = list()
+                    for word_boundary in word_boundaries:
+                        word_label_positions.append((word_boundary + prev_word_boundary) / 2)
+                        prev_word_boundary = word_boundary
+                    word_label_positions.append((duration_splits[-1] + prev_word_boundary) / 2)
 
-                secondary_ax = ax.secondary_xaxis('bottom')
-                secondary_ax.tick_params(axis="x", direction="out", pad=24)
-                secondary_ax.set_xticks(word_label_positions, minor=False)
-                secondary_ax.set_xticklabels(text.split())
-                secondary_ax.tick_params(axis='x', colors='orange')
-                secondary_ax.xaxis.label.set_color('orange')
-            except ValueError:
-                ax.set_title(text)
-            except IndexError:
-                ax.set_title(text)
+                    secondary_ax = ax.secondary_xaxis('bottom')
+                    secondary_ax.tick_params(axis="x", direction="out", pad=24)
+                    secondary_ax.set_xticks(word_label_positions, minor=False)
+                    secondary_ax.set_xticklabels(text.split())
+                    secondary_ax.tick_params(axis='x', colors='orange')
+                    secondary_ax.xaxis.label.set_color('orange')
+                except ValueError:
+                    ax.set_title(text)
+                except IndexError:
+                    ax.set_title(text)
+                except RuntimeError:
+                    ax.set_title(text)
 
-            ax.vlines(x=duration_splits, colors="green", linestyles="solid", ymin=0, ymax=120, linewidth=0.5)
-            ax.vlines(x=word_boundaries, colors="orange", linestyles="solid", ymin=0, ymax=120, linewidth=1.0)
-            plt.subplots_adjust(left=0.02, bottom=0.2, right=0.98, top=.9, wspace=0.0, hspace=0.0)
-            ax.set_aspect("auto")
+                ax.vlines(x=duration_splits, colors="green", linestyles="solid", ymin=0, ymax=120, linewidth=0.5)
+                ax.vlines(x=word_boundaries, colors="orange", linestyles="solid", ymin=0, ymax=120, linewidth=1.0)
+                plt.subplots_adjust(left=0.02, bottom=0.2, right=0.98, top=.9, wspace=0.0, hspace=0.0)
+                ax.set_aspect("auto")
+            except:
+                pass
 
             if return_plot_as_filepath:
-                plt.savefig("tmp.png")
+                try:
+                    plt.savefig("tmp.png")
+                    plt.close()
+                except:
+                    pass
                 return wave, sr, "tmp.png"
 
         return wave, sr