Update app.py
Browse files
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)
|