This commit is contained in:
ahmed531998 2023-04-17 08:55:23 +02:00
parent 9412c10460
commit 38a1d614cf
1 changed files with 10 additions and 10 deletions

20
main.py
View File

@ -29,7 +29,7 @@ cors = CORS(app, resources={r"/api/predict": {"origins": url},
r"/api/dm": {"origins": url}, r"/api/dm": {"origins": url},
r"/health": {"origins": "*"} r"/health": {"origins": "*"}
}) })
"""
conn = psycopg2.connect( conn = psycopg2.connect(
host="janet-pg", host="janet-pg",
database=os.getenv("POSTGRES_DB"), database=os.getenv("POSTGRES_DB"),
@ -37,7 +37,7 @@ conn = psycopg2.connect(
password=os.getenv("POSTGRES_PASSWORD")) password=os.getenv("POSTGRES_PASSWORD"))
cur = conn.cursor() cur = conn.cursor()
"""
users = {} users = {}
def vre_fetch(): def vre_fetch():
@ -58,10 +58,10 @@ def user_interest_decay(user):
def clear_inactive(): def clear_inactive():
while True: while True:
time.sleep(1) time.sleep(1)
for user in users: for username in users:
if user['activity'] > 3600: if users[username]['activity'] > 3600:
del users[user['username']] del users[username]
user['activity'] += 1 users[username]['activity'] += 1
@app.route("/health", methods=['GET']) @app.route("/health", methods=['GET'])
def health(): def health():
@ -133,7 +133,7 @@ def predict():
def feedback(): 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, %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['janetResponse'], data['preferredResponse'], data['length'], data['queryModCorrect'], data['correctQuery'], data['evidence'], data['janetResponse'], data['preferredResponse'], data['length'],
@ -141,7 +141,7 @@ def feedback():
data['speed'], data['intent']) data['speed'], data['intent'])
) )
conn.commit() conn.commit()
"""
reply = jsonify({"status": "done"}) reply = jsonify({"status": "done"})
return reply return reply
@ -185,7 +185,7 @@ if __name__ == "__main__":
rg = ResponseGenerator(index,db, rec, generators, retriever) rg = ResponseGenerator(index,db, rec, generators, retriever)
"""
cur.execute('CREATE TABLE IF NOT EXISTS feedback_experimental (id serial PRIMARY KEY,' cur.execute('CREATE TABLE IF NOT EXISTS feedback_experimental (id serial PRIMARY KEY,'
'query text NOT NULL,' 'query text NOT NULL,'
'history text NOT NULL,' 'history text NOT NULL,'
@ -202,5 +202,5 @@ if __name__ == "__main__":
'response_intent text NOT NULL);' 'response_intent text NOT NULL);'
) )
conn.commit() conn.commit()
"""
app.run(host='0.0.0.0') app.run(host='0.0.0.0')