This commit is contained in:
ahmed531998 2023-04-20 13:34:19 +02:00
parent f2bb7e77df
commit 88e11dcc7e
1 changed files with 55 additions and 50 deletions

13
main.py
View File

@ -99,6 +99,7 @@ def predict():
#rg = users[token]['args']['rg']
#vre = users[token]['args']['vre']
message = {}
try:
if text == "<HELP_ON_START>":
state = {'help': True, 'inactive': False, 'modified_query':"", 'intent':""}
dm.update(state)
@ -145,6 +146,11 @@ def predict():
#users[token]['args']['vre'] = vre
#users[token]['args']['rg'] = rg
return reply
except Exception as e:
message = {"answer": str(e), "query": "", "cand": "candidate", "history": "", "modQuery": ""}
return jsonify(message)
@app.route('/api/feedback', methods = ['POST'])
def feedback():
@ -152,7 +158,6 @@ def feedback():
print(data)
try:
"""
conn = psycopg2.connect(host="janet-pg", database=os.getenv("POSTGRES_DB"), user=os.getenv("POSTGRES_USER"), password=os.getenv("POSTGRES_PASSWORD"))
cur = conn.cursor()
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)',
@ -162,7 +167,7 @@ def feedback():
data['speed'], data['intent']))
conn.commit()
cur.close()
"""
reply = jsonify({"status": "done"})
return reply
@ -202,7 +207,7 @@ if __name__ == "__main__":
threading.Thread(target=vre_fetch, name='updatevre').start()
threading.Thread(target=clear_inactive, name='clear').start()
"""
conn = psycopg2.connect(host="janet-pg", database=os.getenv("POSTGRES_DB"), user=os.getenv("POSTGRES_USER"), password=os.getenv("POSTGRES_PASSWORD"))
cur = conn.cursor()
@ -224,5 +229,5 @@ if __name__ == "__main__":
)
conn.commit()
cur.close()
"""
app.run(host='0.0.0.0')