Update app.py

app.py

@@ -32,22 +32,38 @@ def clean_retrieved_context(raw_context):
     return cleaned.strip()
 
 def format_ragas_results(ragas_results):
-    return {metric: f"{score*100:.2f}%" for metric, score in ragas_results.scores.items()}
+    return [
+        {metric: f"{score*100:.2f}%" for metric, score in sample_scores.items()}
+        for sample_scores in ragas_results.scores
+    ]
 
-def plot_radar_chart(ragas_results):
-    labels = list(ragas_results.keys())
-    scores = list(ragas_results.values())
-    
-    scores += scores[:1]
-    angles = np.linspace(0, 2 * np.pi, len(labels) + 1, endpoint=True)
+def plot_ragas_radar_chart(ragas_results):
+    # Extract first sample if multiple scores
+    if isinstance(ragas_results.scores, list):
+        scores_dict = ragas_results.scores[0]  # first sample
+    else:
+        scores_dict = ragas_results.scores
+
+    labels = list(scores_dict.keys())
+    values = list(scores_dict.values())
+
+    # Close the loop for radar chart
+    values += values[:1]
+    angles = np.linspace(0, 2 * np.pi, len(labels), endpoint=False).tolist()
+    angles += angles[:1]
 
     fig, ax = plt.subplots(figsize=(6,6), subplot_kw=dict(polar=True))
-    ax.plot(angles, scores, 'o-', linewidth=2)
-    ax.fill(angles, scores, alpha=0.25)
+
+    ax.fill(angles, values, color='skyblue', alpha=0.4)
+    ax.plot(angles, values, color='blue', linewidth=2)
+
+    ax.set_yticks([0.2, 0.4, 0.6, 0.8, 1.0])
+    ax.set_yticklabels(['0.2', '0.4', '0.6', '0.8', '1.0'])
+
     ax.set_xticks(angles[:-1])
     ax.set_xticklabels(labels)
-    ax.set_yticklabels([])
-    ax.set_title('RAGAS Metrics Radar Chart')
+
+    ax.set_title('RAGAS Metrics Radar Chart', size=14, y=1.05)
     plt.tight_layout()
     return fig
 
@@ -89,7 +105,7 @@ def evaluate_scdd(ai_scdd_file, human_scdd_file, user_input):
     # RAGAS metrics outputs
     
     formatted_scores = format_ragas_results(ragas_result)
-    radar_chart = plot_radar_chart(ragas_result)
+    radar_chart = plot_ragas_radar_chart(ragas_result)
     
     return formatted_scores, radar_chart