ahmed.ibrahim39699
/
JanetBackEnd
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixes

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
NLU.py
View File

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

16
main.py
View File

@ -35,14 +35,6 @@ conn = psycopg2.connect(
user=os.getenv("POSTGRES_USER"), user=os.getenv("POSTGRES_USER"),
password=os.getenv("POSTGRES_PASSWORD")) 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() cur = conn.cursor()
@ -114,10 +106,9 @@ def feedback():
data = request.get_json().get("feedback") data = request.get_json().get("feedback")
print(data) 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['query'], data['history'], data['modQuery'],
data['queryModCorrect'], data['correctQuery'], data['evidence'], data['queryModCorrect'], data['correctQuery'], data['evidence'], data['janetResponse'], data['preferredResponse'], data['length'],
data['janetResponse'], data['preferredResponse'], data['length'],
data['fluency'], data['truthfulness'], data['usefulness'], data['fluency'], data['truthfulness'], data['usefulness'],
data['speed'], data['intent']) data['speed'], data['intent'])
) )
@ -175,8 +166,7 @@ if __name__ == "__main__":
'history text NOT NULL,' 'history text NOT NULL,'
'janet_modified_query text NOT NULL,' 'janet_modified_query text NOT NULL,'
'is_modified_query_correct text NOT NULL,' 'is_modified_query_correct text NOT NULL,'
'user_modified_query text NOT NULL,' 'user_modified_query text NOT NULL, evidence_useful text NOT NULL,'
'evidence_useful text NOT NULL,'
'response text NOT NULL,' 'response text NOT NULL,'
'preferred_response text,' 'preferred_response text,'
'response_length_feedback text NOT NULL,' 'response_length_feedback text NOT NULL,'