Update app.py

app.py

@@ -55,8 +55,28 @@ def bodyparts_plot(bodyparts, speed_csv, tmpdir):
 
       plt.savefig(f'{tmpdir}/{i}.png')
 
-
 def likelihood_plot(bodyparts, likelihood_csv, tmpdir):
+   for i in bodyparts:
+      plt.figure()
+      plt.plot(likelihood_csv[i], label=i, color=cm.rainbow(int(255 * bodyparts.index(i) / len(bodyparts))))
+      plt.legend()
+      plt.xlabel('frame')
+      plt.ylabel('likeloohood')
+  
+      # 横軸の最大値を取得して設定
+      xmax = likelihood_csv.index.max()
+      plt.xlim(0, xmax)
+      plt.ylim(0, 1)
+  
+      # グラフの凡例をグラフ外に表示。
+      plt.legend(bbox_to_anchor=(1.05, 1), loc='upper left', borderaxespad=0, fontsize=10)
+  
+      # 凡例をすべて画像内に収める
+      plt.tight_layout()
+  
+      plt.savefig(f'{tmpdir}/{i}.png')
+
+def all_likelihood_plot(bodyparts, likelihood_csv, tmpdir):
    plt.figure()
    for i in bodyparts:
       plt.plot(likelihood_csv[i], label=i, color=cm.rainbow(int(255 * bodyparts.index(i) / len(bodyparts))))
@@ -100,6 +120,7 @@ def main(h5_file):
    with tempfile.TemporaryDirectory(dir=".") as tmpdir:
       bodyparts_plot(bodyparts, speed_csv, tmpdir)
       all_plot(bodyparts, speed_csv, tmpdir)
+      all_likelihood_plot(bodyparts, likelihood_csv, tmpdir)
       likelihood_plot(bodyparts, likelihood_csv, tmpdir)
       shutil.make_archive(f"{h5_file_name}", 'zip', root_dir=tmpdir)
       return f"{h5_file_name}.zip"