anl139 commited on
Commit
45cee9d
·
verified ·
1 Parent(s): c7446b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -49,7 +49,7 @@ def extract_metadata(text: str) -> dict:
49
  ranking_value = ranking_match.group(1).strip()
50
  metadata["raw_ranking"] = ranking_value # Store full ranking information
51
  if ranking_value.lower() == "winner":
52
- metadata["ranking"] = ranking_value
53
 
54
  # Extract the Year field (assuming a four-digit year)
55
  year_match = re.search(r"Year:\s*(\d{4})", text, re.IGNORECASE)
@@ -106,7 +106,7 @@ def load_and_process_data(file_path: str):
106
  continue
107
  metadata = extract_metadata(org_text)
108
  # Insert winners at the beginning of the list
109
- if metadata.get("ranking", "").lower() == "winner":
110
  docs.insert(0, Document(page_content=org_text, metadata=metadata))
111
  else:
112
  docs.append(Document(page_content=org_text, metadata=metadata))
 
49
  ranking_value = ranking_match.group(1).strip()
50
  metadata["raw_ranking"] = ranking_value # Store full ranking information
51
  if ranking_value.lower() == "winner":
52
+ metadata["LA2050 Grant Winner"] = ranking_value
53
 
54
  # Extract the Year field (assuming a four-digit year)
55
  year_match = re.search(r"Year:\s*(\d{4})", text, re.IGNORECASE)
 
106
  continue
107
  metadata = extract_metadata(org_text)
108
  # Insert winners at the beginning of the list
109
+ if metadata.get("LA2050 Grant Winner", "").lower() == "winner":
110
  docs.insert(0, Document(page_content=org_text, metadata=metadata))
111
  else:
112
  docs.append(Document(page_content=org_text, metadata=metadata))