sh1gechan commited on
Commit
e647d43
·
verified ·
1 Parent(s): 0f59cf4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -133,15 +133,16 @@ def select_columns(df: pd.DataFrame, columns: list) -> pd.DataFrame:
133
  if c not in seen:
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.split('>')[1].split('</a')[0] if isinstance(x, str) and '<a' in x else x)
140
 
141
  # フィルタリングされたカラムでデータフレームを作成
142
  filtered_df = df[unique_columns]
143
  return filtered_df
144
 
 
145
  def filter_queries(query: str, filtered_df: pd.DataFrame):
146
  """Added by Abishek"""
147
  final_df = []
 
133
  if c not in seen:
134
  unique_columns.append(c)
135
  seen.add(c)
136
+
137
+ # 'Model' カラムにリンクを含む形式で再構築
138
  if 'Model' in df.columns:
139
+ df['Model'] = df['Model'].apply(lambda x: f'<a href="{x.split("href=")[1].split(" ")[0].strip()}">{x.split(">")[1].split("</a")[0]}</a>' if isinstance(x, str) and '<a' in x else x)
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 = []