- [MINOR] [SOURCE] [UPDATE] 1. update app.py

app.py

@@ -74,32 +74,47 @@ def create_bar_chart(scores, comparisons):
     labels = ['Semantic', 'Aesthetic', 'Technical', 'Overall']
     base_colors = ['#d62728', '#ff7f0e', '#1f77b4', '#bcbd22']
     
-    fig, ax = plt.subplots(figsize=(10, 5))
+    fig, ax = plt.subplots(figsize=(8, 6))
+
+    # Create a bar chart with vertical bars
+    bars = ax.bar(labels, scores, color=base_colors, edgecolor='black', width=0.6)
+
+    # Add gradient effect
+    for bar, color in zip(bars, base_colors):
+        bar.set_facecolor(plt.cm.get_cmap('autumn')(bar.get_height() / 5))
     
-    for i, (label, score, comparison, base_color) in enumerate(zip(labels, scores, comparisons, base_colors)):
-        gradient = patches.Rectangle((0, i), 5, 1, color=base_color, alpha=0.5)
-        ax.add_patch(gradient)
-        
-        # Add the actual score line
-        ax.plot([score, score], [i, i+0.9], color='black', linewidth=2)
-        
-        ax.text(score + 0.1, i + 0.5, f'{score:.1f}', va='center', ha='left', color=base_color)
-        ax.text(5.1, i + 0.5, comparison, va='center', ha='left', color=base_color)
+    # Adding the text labels for scores
+    for bar, score in zip(bars, scores):
+        height = bar.get_height()
+        ax.annotate(f'{score}',
+                    xy=(bar.get_x() + bar.get_width() / 2, height),
+                    xytext=(0, 3),  # 3 points vertical offset
+                    textcoords="offset points",
+                    ha='center', va='bottom',
+                    color='black')
     
-    ax.set_yticks(range(len(labels)))
-    ax.set_yticklabels(labels)
-    for tick, color in zip(ax.get_yticklabels(), base_colors):
-        tick.set_color(color)
-    ax.set_xticks([0, 1, 2, 3, 4, 5])
-    ax.set_xticklabels([0, 1, 2, 3, 4, 5])
-    ax.set_xlim(0, 5)
-    ax.set_xlabel('Score')
+    # Add comparison text
+    comparison_texts = []
+    for i, score in enumerate(scores):
+        percentage = (np.sum(comparisons[i] < score) / len(comparisons[i])) * 100
+        comparison_text = f'Better than {percentage:.0f}% videos in YT-UGC' if percentage > 50 else f'Worse than {100 - percentage:.0f}% videos in YT-UGC'
+        comparison_texts.append(comparison_text)
+        ax.annotate(comparison_text,
+                    xy=(bar.get_x() + bar.get_width(), height),
+                    xytext=(5, 0),  # 5 points horizontal offset
+                    textcoords="offset points",
+                    ha='left', va='center',
+                    color=color)
+    
+    ax.set_xlabel('Categories')
+    ax.set_ylabel('Scores')
+    ax.set_ylim(0, 5)
+    ax.set_title('Video Quality Scores')
     
     plt.tight_layout()
-    image_path = "./bar_chart.png"
+    image_path = "./scores_bar_chart.png"
     plt.savefig(image_path)
-    plt.close()
-    
+    plt.close(fig)
     return image_path
 
 def inference_one_video(input_video):