Update app.py

app.py

@@ -143,7 +143,7 @@ def on_row_select(evt: gr.SelectData, df: pd.DataFrame) -> str:
     row = df.iloc[int(row_idx)]
     row_id = int(row["id"])
     # 透過 df 的 shape 來判斷是哪張表（也可改成傳 flag）
-    if len(df) == len(reading_test_df) and set(df.columns) == set(reading_test_df.columns):
+    if row_id in reading_test_id2data_str.keys():
         # 閱讀測驗
         return reading_test_id2data_str[row_id]
     else: