Rulga commited on
Commit
4903398
·
1 Parent(s): da199b7

Refactor evaluation interface in evaluation_interface.py: Update DataFrame columns for clarity in QA pairs display

Browse files
Files changed (2) hide show
  1. app.py +2 -3
  2. web/evaluation_interface.py +4 -4
app.py CHANGED
@@ -1209,9 +1209,8 @@ with gr.Blocks(css="""
1209
  def on_table_select(evt: gr.SelectData) -> str:
1210
  """Handle table row selection"""
1211
  try:
1212
- # evt.value содержит данные выбранной строки
1213
- # Возвращаем conversation_id из первой колонки
1214
- return evt.value[0]
1215
  except Exception as e:
1216
  logger.error(f"Error in table selection: {str(e)}")
1217
  return ""
 
1209
  def on_table_select(evt: gr.SelectData) -> str:
1210
  """Handle table row selection"""
1211
  try:
1212
+ # Возвращаем полный conversation_id из первой колонки
1213
+ return evt.value[0] if evt.value else ""
 
1214
  except Exception as e:
1215
  logger.error(f"Error in table selection: {str(e)}")
1216
  return ""
web/evaluation_interface.py CHANGED
@@ -46,7 +46,7 @@ def get_evaluation_status(evaluator, force_reload=False):
46
 
47
  # Import pandas here to avoid circular imports
48
  import pandas as pd
49
- empty_df = pd.DataFrame(columns=["Conversation ID", "Question", "Timestamp", "Evaluated"])
50
 
51
  return f"Error getting status: {str(e)}", empty_df, f"Error: {str(e)}"
52
 
@@ -82,7 +82,7 @@ def get_qa_pairs_dataframe(evaluator, show_evaluated=False, force_reload=False):
82
  {
83
  "Conversation ID": qa["conversation_id"],
84
  "Question": qa["question"][:50] + "..." if len(qa["question"]) > 50 else qa["question"],
85
- "Timestamp": qa.get("timestamp", ""),
86
  "Evaluated": "Yes" if qa["conversation_id"] in evaluated_ids else "No"
87
  }
88
  for qa in qa_pairs
@@ -90,11 +90,11 @@ def get_qa_pairs_dataframe(evaluator, show_evaluated=False, force_reload=False):
90
  return df
91
  else:
92
  import pandas as pd
93
- return pd.DataFrame(columns=["Conversation ID", "Question", "Timestamp", "Evaluated"])
94
  except Exception as e:
95
  logger.error(f"Error getting QA pairs dataframe: {e}")
96
  import pandas as pd
97
- return pd.DataFrame(columns=["Conversation ID", "Question", "Timestamp", "Evaluated"])
98
 
99
  def load_qa_pair_for_evaluation(conversation_id: str, evaluator: ChatEvaluator) -> Tuple[str, str, str, int, int, int, int, int, str]:
100
  """
 
46
 
47
  # Import pandas here to avoid circular imports
48
  import pandas as pd
49
+ empty_df = pd.DataFrame(columns=["Conversation ID", "Question", "Answer", "Evaluated"])
50
 
51
  return f"Error getting status: {str(e)}", empty_df, f"Error: {str(e)}"
52
 
 
82
  {
83
  "Conversation ID": qa["conversation_id"],
84
  "Question": qa["question"][:50] + "..." if len(qa["question"]) > 50 else qa["question"],
85
+ "Answer": qa["original_answer"][:100] + "..." if len(qa["original_answer"]) > 100 else qa["original_answer"],
86
  "Evaluated": "Yes" if qa["conversation_id"] in evaluated_ids else "No"
87
  }
88
  for qa in qa_pairs
 
90
  return df
91
  else:
92
  import pandas as pd
93
+ return pd.DataFrame(columns=["Conversation ID", "Question", "Answer", "Evaluated"])
94
  except Exception as e:
95
  logger.error(f"Error getting QA pairs dataframe: {e}")
96
  import pandas as pd
97
+ return pd.DataFrame(columns=["Conversation ID", "Question", "Answer", "Evaluated"])
98
 
99
  def load_qa_pair_for_evaluation(conversation_id: str, evaluator: ChatEvaluator) -> Tuple[str, str, str, int, int, int, int, int, str]:
100
  """