naman1102 commited on
Commit
3d185cb
·
1 Parent(s): c7aa48f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -0
app.py CHANGED
@@ -122,6 +122,7 @@ def show_combined_repo_and_llm():
122
  df = read_csv_as_text(csv_filename)
123
  for col in ["strength", "weaknesses", "speciality", "relevance rating"]:
124
  df[col] = df[col].astype(str)
 
125
  for idx, row in df.iterrows():
126
  if row["repo id"] == repo_id:
127
  if isinstance(llm_json, dict) and "error" not in llm_json:
@@ -132,9 +133,23 @@ def show_combined_repo_and_llm():
132
  df.at[idx, "weaknesses"] = weaknesses
133
  df.at[idx, "speciality"] = llm_json.get("speciality", "")
134
  df.at[idx, "relevance rating"] = llm_json.get("relevance rating", "")
 
135
  else:
136
  extraction_status = f"JSON extraction: FAILED\nRaw: {llm_json.get('raw', '') if isinstance(llm_json, dict) else llm_json}"
137
  break
 
 
 
 
 
 
 
 
 
 
 
 
 
138
  df.to_csv(csv_filename, index=False)
139
  except Exception as e:
140
  df = read_csv_as_text(csv_filename)
 
122
  df = read_csv_as_text(csv_filename)
123
  for col in ["strength", "weaknesses", "speciality", "relevance rating"]:
124
  df[col] = df[col].astype(str)
125
+ updated = False
126
  for idx, row in df.iterrows():
127
  if row["repo id"] == repo_id:
128
  if isinstance(llm_json, dict) and "error" not in llm_json:
 
133
  df.at[idx, "weaknesses"] = weaknesses
134
  df.at[idx, "speciality"] = llm_json.get("speciality", "")
135
  df.at[idx, "relevance rating"] = llm_json.get("relevance rating", "")
136
+ updated = True
137
  else:
138
  extraction_status = f"JSON extraction: FAILED\nRaw: {llm_json.get('raw', '') if isinstance(llm_json, dict) else llm_json}"
139
  break
140
+ # If not updated (repo_id not found), append a new row
141
+ if not updated and isinstance(llm_json, dict) and "error" not in llm_json:
142
+ extraction_status = "JSON extraction: SUCCESS (new row)"
143
+ strengths = llm_json.get("strength", "")
144
+ weaknesses = llm_json.get("weaknesses", "")
145
+ new_row = {
146
+ "repo id": repo_id,
147
+ "strength": strengths,
148
+ "weaknesses": weaknesses,
149
+ "speciality": llm_json.get("speciality", ""),
150
+ "relevance rating": llm_json.get("relevance rating", "")
151
+ }
152
+ df = pd.concat([df, pd.DataFrame([new_row])], ignore_index=True)
153
  df.to_csv(csv_filename, index=False)
154
  except Exception as e:
155
  df = read_csv_as_text(csv_filename)