SFM2001 commited on
Commit
d9014dc
·
1 Parent(s): 6b1b687
Files changed (2) hide show
  1. create_app.py +0 -1
  2. views/infer.py +7 -4
create_app.py CHANGED
@@ -51,7 +51,6 @@ def load_models():
51
  def create_app():
52
  set_seed(42)
53
  load_models()
54
- print("MODEL LOADED", flush=True)
55
  app = Flask(__name__)
56
  app.config.from_pyfile('configs.py')
57
  db.init_app(app)
 
51
  def create_app():
52
  set_seed(42)
53
  load_models()
 
54
  app = Flask(__name__)
55
  app.config.from_pyfile('configs.py')
56
  db.init_app(app)
views/infer.py CHANGED
@@ -22,7 +22,12 @@ def infer():
22
  essay = essay.replace('\n', '<br>')
23
  corrected_essay = corrected_essay.replace('\\n\\n', '\\n').replace('\\n', '<br>')
24
  print("Everything loaded", flush=True)
25
- return render_template('infer.html', result=scores, topic=topic, essay=essay, tr=tr_feedback, cc=cc_feedback, lr=lr_feedback, gra=gra_feedback, corrected_essay=corrected_essay)
 
 
 
 
 
26
 
27
  @infer_bp.route('/', methods=['GET', 'POST'])
28
  def index():
@@ -49,8 +54,6 @@ def index():
49
  session['topic'] = topic
50
  session['essay'] = essay.replace('\\n', '\n')
51
  session['feedback'] = feedback or {} # Ensure feedback is at least an empty dict
52
- token = secrets.token_urlsafe(16)
53
- session['inference_token'] = token
54
  if current_user.is_authenticated:
55
  scores = [int(score) for score in session['scores']]
56
  new_history = History(
@@ -65,7 +68,7 @@ def index():
65
  db.session.add(new_history)
66
  db.session.commit()
67
  print("REDIRECT TO INFER", flush=True)
68
- return redirect(url_for('infer.infer', token=token))
69
  return render_template('index.html', isempty='False', username=current_user.nickname if current_user.is_authenticated else '')
70
 
71
  @infer_bp.route('/rubric_explanation')
 
22
  essay = essay.replace('\n', '<br>')
23
  corrected_essay = corrected_essay.replace('\\n\\n', '\\n').replace('\\n', '<br>')
24
  print("Everything loaded", flush=True)
25
+ try:
26
+ return render_template('infer.html', result=scores, topic=topic, essay=essay, tr=tr_feedback, cc=cc_feedback, lr=lr_feedback, gra=gra_feedback, corrected_essay=corrected_essay)
27
+ except Exception as e:
28
+ print(e, flush=True)
29
+ return render_template('error.html', error=str(e)), 500
30
+
31
 
32
  @infer_bp.route('/', methods=['GET', 'POST'])
33
  def index():
 
54
  session['topic'] = topic
55
  session['essay'] = essay.replace('\\n', '\n')
56
  session['feedback'] = feedback or {} # Ensure feedback is at least an empty dict
 
 
57
  if current_user.is_authenticated:
58
  scores = [int(score) for score in session['scores']]
59
  new_history = History(
 
68
  db.session.add(new_history)
69
  db.session.commit()
70
  print("REDIRECT TO INFER", flush=True)
71
+ return redirect(url_for('infer.infer'))
72
  return render_template('index.html', isempty='False', username=current_user.nickname if current_user.is_authenticated else '')
73
 
74
  @infer_bp.route('/rubric_explanation')