Update app.py

app.py

@@ -9,22 +9,28 @@ def compare_csv_files():
     df2 = pd.read_csv(file2)
     
     merged_df = pd.merge(df1, df2, on="SourceText", suffixes=("_1.5", "_1.4"))
+
     merged_df["WordErrorRate_Diff"] = merged_df["WordErrorRate_1.5"] - merged_df["WordErrorRate_1.4"]
-    if merged_df["WordErrorRate_Diff"] > 0: merged_df["WordErrorRate_Diff"] = f"1.5 is stronger than 1.4 ({merged_df['WordErrorRate_Diff']})"
-    if merged_df["WordErrorRate_Diff"] < 0: merged_df["WordErrorRate_Diff"] = f"1.4 is stronger than 1.5 ({0 - merged_df['WordErrorRate_Diff']})"
-    if merged_df["WordErrorRate_Diff"] == 0: merged_df["WordErrorRate_Diff"] = f"1.4 is the same as 1.5 (0)"
-    if merged_df["CharacterErrorRate_Diff"] > 0: merged_df["CharacterErrorRate_Diff"] = f"1.5 is stronger than 1.4 ({merged_df['CharacterErrorRate_Diff']})"
-    if merged_df["CharacterErrorRate_Diff"] < 0: merged_df["CharacterErrorRate_Diff"] = f"1.4 is stronger than 1.5 ({0 - merged_df['CharacterErrorRate_Diff']})"
-    if merged_df["CharacterErrorRate_Diff"] == 0: merged_df["CharacterErrorRate_Diff"] = f"1.4 is the same as 1.5 (0)"
+    merged_df["WordErrorRate_Comparison"] = merged_df["WordErrorRate_Diff"].apply(
+        lambda x: f"1.5 is stronger than 1.4 ({x})" if x > 0 else (
+            f"1.4 is stronger than 1.5 ({-x})" if x < 0 else "1.4 is the same as 1.5 (0)"
+        )
+    )
+    
+    merged_df["CharacterErrorRate_Diff"] = merged_df["CharacterErrorRate_1.5"] - merged_df["CharacterErrorRate_1.4"]
+    merged_df["CharacterErrorRate_Comparison"] = merged_df["CharacterErrorRate_Diff"].apply(
+        lambda x: f"1.5 is stronger than 1.4 ({x})" if x > 0 else (
+            f"1.4 is stronger than 1.5 ({-x})" if x < 0 else "1.4 is the same as 1.5 (0)"
+        )
+    )
     
-    comparison = merged_df[[
-        "SourceText", 
-        "WhisperText_1.5", "WordErrorRate_1.5", "CharacterErrorRate_1.5",
-        "WhisperText_1.4", "WordErrorRate_1.4", "CharacterErrorRate_1.4",
-        "WordErrorRate_Diff", "CharacterErrorRate_Diff"
+    result = merged_df[[
+        "SourceText",
+        "WordErrorRate_1.5", "WordErrorRate_1.4", "WordErrorRate_Comparison",
+        "CharacterErrorRate_1.5", "CharacterErrorRate_1.4", "CharacterErrorRate_Comparison"
     ]]
     
-    return comparison.to_html(index=False)
+    return result.to_html(index=False)
 
 gr.Interface(
     fn=compare_csv_files,