davidberenstein1957 HF staff commited on
Commit
85e41fb
·
1 Parent(s): 0cba459

Add rating processing function to app.py for improved data handling

Browse files
Files changed (1) hide show
  1. app/app.py +10 -0
app/app.py CHANGED
@@ -118,6 +118,15 @@ def _process_content(content) -> str | list[str]:
118
  return content
119
 
120
 
 
 
 
 
 
 
 
 
 
121
  def add_fake_like_data(
122
  history: list, session_id: str, language: str, liked: bool = False
123
  ) -> None:
@@ -275,6 +284,7 @@ def submit_conversation(dataframe, session_id, language):
275
  return (gr.Dataframe(value=None, interactive=False), [])
276
 
277
  dataframe["content"] = dataframe["content"].apply(_process_content)
 
278
  conversation = dataframe.to_dict(orient="records")
279
  conversation_data = {
280
  "conversation": conversation,
 
118
  return content
119
 
120
 
121
+ def _process_rating(rating) -> int:
122
+ if isinstance(rating, str):
123
+ return 0
124
+ elif isinstance(rating, int):
125
+ return rating
126
+ else:
127
+ raise ValueError(f"Invalid rating: {rating}")
128
+
129
+
130
  def add_fake_like_data(
131
  history: list, session_id: str, language: str, liked: bool = False
132
  ) -> None:
 
284
  return (gr.Dataframe(value=None, interactive=False), [])
285
 
286
  dataframe["content"] = dataframe["content"].apply(_process_content)
287
+ dataframe["rating"] = dataframe["rating"].apply(_process_rating)
288
  conversation = dataframe.to_dict(orient="records")
289
  conversation_data = {
290
  "conversation": conversation,