Spaces:
Running
Running
Edit some comments
Browse files- src/relevancy.py +3 -5
src/relevancy.py
CHANGED
@@ -31,7 +31,7 @@ def encode_prompt(query, prompt_papers):
|
|
31 |
prompt += f"{idx + 1}. Authors: {authors}\n"
|
32 |
prompt += f"{idx + 1}. Abstract: {abstract}\n"
|
33 |
prompt += f"\n Generate response:\n1."
|
34 |
-
|
35 |
return prompt
|
36 |
|
37 |
|
@@ -45,8 +45,6 @@ def is_json(myjson):
|
|
45 |
|
46 |
def post_process_chat_gpt_response(paper_data, response, threshold_score=8):
|
47 |
selected_data = []
|
48 |
-
print("HERE")
|
49 |
-
print(response['message']['content'])
|
50 |
if response is None:
|
51 |
return []
|
52 |
json_items = response['message']['content'].replace("\n\n", "\n").split("\n")
|
@@ -57,14 +55,14 @@ def post_process_chat_gpt_response(paper_data, response, threshold_score=8):
|
|
57 |
json.loads(re.sub(pattern, "", line))
|
58 |
for line in json_items if (is_json(line) and "relevancy score" in line.lower())]
|
59 |
except Exception as e:
|
60 |
-
|
61 |
try:
|
62 |
score_items = score_items[:-1]
|
63 |
except Exception:
|
64 |
score_items = []
|
65 |
print(e)
|
66 |
raise RuntimeError("failed")
|
67 |
-
|
68 |
scores = []
|
69 |
for item in score_items:
|
70 |
temp = item["Relevancy score"]
|
|
|
31 |
prompt += f"{idx + 1}. Authors: {authors}\n"
|
32 |
prompt += f"{idx + 1}. Abstract: {abstract}\n"
|
33 |
prompt += f"\n Generate response:\n1."
|
34 |
+
print(prompt)
|
35 |
return prompt
|
36 |
|
37 |
|
|
|
45 |
|
46 |
def post_process_chat_gpt_response(paper_data, response, threshold_score=8):
|
47 |
selected_data = []
|
|
|
|
|
48 |
if response is None:
|
49 |
return []
|
50 |
json_items = response['message']['content'].replace("\n\n", "\n").split("\n")
|
|
|
55 |
json.loads(re.sub(pattern, "", line))
|
56 |
for line in json_items if (is_json(line) and "relevancy score" in line.lower())]
|
57 |
except Exception as e:
|
58 |
+
pprint.pprint([re.sub(pattern, "", line) for line in json_items if "relevancy score" in line.lower()])
|
59 |
try:
|
60 |
score_items = score_items[:-1]
|
61 |
except Exception:
|
62 |
score_items = []
|
63 |
print(e)
|
64 |
raise RuntimeError("failed")
|
65 |
+
pprint.pprint(score_items)
|
66 |
scores = []
|
67 |
for item in score_items:
|
68 |
temp = item["Relevancy score"]
|