This commit is contained in:
ahmed531998 2023-04-16 19:52:40 +02:00
parent 5f0255f53a
commit 984d51e200
2 changed files with 8 additions and 16 deletions

8
NLU.py
View File

@ -17,15 +17,16 @@ class NLU:
doc = self.coref_resolver(to_resolve)
token_mention_mapper = {}
output_string = ""
cand_clusters = [
clusters = [
val for key, val in doc.spans.items() if key.startswith("coref_cluster")
]
"""
clusters = []
for cluster in cand_clusters:
if cluster[0].text == "I":
continue
clusters.append(cluster)
"""
# Iterate through every found cluster
for cluster in clusters:
first_mention = cluster[0]
@ -88,7 +89,8 @@ class NLU:
self.to_process = self._resolve_coref(history_consec)
intent, score = self._intentpredictor()
#print(score)
if intent == 'CHITCHAT':
self.to_process = utterance
if score > 0.5:
entities = self._entityextractor()
offense = self._offensepredictor()

16
main.py
View File

@ -35,14 +35,6 @@ conn = psycopg2.connect(
user=os.getenv("POSTGRES_USER"),
password=os.getenv("POSTGRES_PASSWORD"))
"""
conn = psycopg2.connect(host="https://janet-app-db.d4science.org",
database="janet",
user="janet_user",
password="2fb5e81fec5a2d906a04")
"""
cur = conn.cursor()
@ -114,10 +106,9 @@ def feedback():
data = request.get_json().get("feedback")
print(data)
cur.execute('INSERT INTO feedback_experimental (query, history, janet_modified_query, is_modified_query_correct, user_modified_query, evidence_useful, response, preferred_response, response_length_feedback, response_fluency_feedback, response_truth_feedback, response_useful_feedback, response_time_feedback, response_intent) VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)',
cur.execute('INSERT INTO feedback_experimental (query, history, janet_modified_query, is_modified_query_correct, user_modified_query, evidence_useful, response, preferred_response, response_length_feedback, response_fluency_feedback, response_truth_feedback, response_useful_feedback, response_time_feedback, response_intent) VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)',
(data['query'], data['history'], data['modQuery'],
data['queryModCorrect'], data['correctQuery'], data['evidence'],
data['janetResponse'], data['preferredResponse'], data['length'],
data['queryModCorrect'], data['correctQuery'], data['evidence'], data['janetResponse'], data['preferredResponse'], data['length'],
data['fluency'], data['truthfulness'], data['usefulness'],
data['speed'], data['intent'])
)
@ -175,8 +166,7 @@ if __name__ == "__main__":
'history text NOT NULL,'
'janet_modified_query text NOT NULL,'
'is_modified_query_correct text NOT NULL,'
'user_modified_query text NOT NULL,'
'evidence_useful text NOT NULL,'
'user_modified_query text NOT NULL, evidence_useful text NOT NULL,'
'response text NOT NULL,'
'preferred_response text,'
'response_length_feedback text NOT NULL,'