sh1gechan commited on
Commit
0e1e133
·
verified ·
1 Parent(s): d597add

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -134,15 +134,16 @@ def select_columns(df: pd.DataFrame, columns: list) -> pd.DataFrame:
134
  unique_columns.append(c)
135
  seen.add(c)
136
 
137
- # 'Model' カラムがHTMLリンクでない場合を選択
138
  if 'Model' in df.columns:
139
- df['Model'] = df['Model'].apply(lambda x: x if isinstance(x, str) and not x.startswith('<a') else '')
140
 
141
  # フィルタリングされたカラムでデータフレームを作成
142
  filtered_df = df[unique_columns]
143
  return filtered_df
144
 
145
 
 
146
  def filter_queries(query: str, filtered_df: pd.DataFrame):
147
  """Added by Abishek"""
148
  final_df = []
 
134
  unique_columns.append(c)
135
  seen.add(c)
136
 
137
+ # 'Model' カラムがHTMLリンクでなく、プレーンテキストの場合を保持する
138
  if 'Model' in df.columns:
139
+ df['Model'] = df['Model'].apply(lambda x: x if isinstance(x, str) and '<a' not in x else '')
140
 
141
  # フィルタリングされたカラムでデータフレームを作成
142
  filtered_df = df[unique_columns]
143
  return filtered_df
144
 
145
 
146
+
147
  def filter_queries(query: str, filtered_df: pd.DataFrame):
148
  """Added by Abishek"""
149
  final_df = []