server.py

from crypt import methods
from flask import Flask
from flask import render_template
from flask import Response, request, jsonify
import textwrap as tw
import openai
import json
import time
from dotenv import load_dotenv
import os

app = Flask(__name__)
load_dotenv()

OPENAI_KEY = os.environ.get("API_KEY")
openai.api_key = OPENAI_KEY

topics = {
    "0" : "Behavioral",
    "1" : "Brain-teaser",
    "2" : "Communication",
    "3" : "Opinion",
    "4" : "Situational",
    "5" : "Technical",
}

fields = {
    "acc" : "Accounting",
    "bio" : "Biology",
    "bsn" : "Business",
    "cs" : "Computer Science",
    "eng" : "Engineering",
    "fin" : "Finance",
    "he" : "Health",
}

chosen_topic = None
chosen_field = None
prompt = None
questions = None
chosen_question = None
user_answer = None

# Helper function to ensure output in a particular format
def convertResponseToList(response):
    whitespaced_list = list(map(lambda option: option.lstrip('0123456789.-) '), response.split("\n")))
    result = list(filter(None, whitespaced_list))
    return result

# Initial route called when page loaded via GET request
# Send the data for topics and fields to display on home page
@app.route('/')
def index():
    return render_template("home.html", topics= topics,fields = fields)

# Ajax call which gets the chosen topic and field, and returns all the questions
@app.route('/generateQuestions', methods = ['GET','POST'])
def generateQuestions():
    global chosen_topic
    global chosen_field
    global prompt
    global questions
    json_data = request.get_json()
    chosen_topic = topics[json_data['topic_index']] 
    chosen_field = fields[json_data['field_index']]
    prompt = f"List 5 {chosen_topic.lower()} interview questions for a role in {chosen_field.lower()} numbered 1 to 5 "
    completion = openai.Completion.create(engine="text-davinci-002",
                                          max_tokens=256,
                                          prompt=prompt)
    result = completion.choices[0].text.strip()
    questions = convertResponseToList(result)
    returnData = {'prompt': prompt, 'questions': questions}
    return jsonify("Success"), 200

# Evaluates the user's answer to the selected question and sends the feedback and keywords
# Called again when user re-enters a new answer to incorporate the feedback 
@app.route('/evaluateAnswer', methods = ['GET','POST'])
def evaluateAnswer():
    grading = 'give a grade out of 5 with reasoning and useful feedback'
    prompt = f"Evaluate the answer '{user_answer}' to the interview question '{chosen_question}' and {grading}."
    completion = openai.Completion.create(engine="text-davinci-002",
                                          max_tokens=256,
                                          prompt=prompt)
    result = completion.choices[0].text.strip()
    chained_prompt =  f"Find keywords from the interview answer '{prompt}'"
    completion = openai.Completion.create(engine="text-davinci-002",
                                          max_tokens=256,
                                          prompt=chained_prompt)
    keyword_result = completion.choices[0].text.strip()
    returnData = {'prompt': prompt, 'feedback': result, 'keywords': keyword_result}
    return render_template("display_feedback.html", topics = topics, fields = fields, return_data = returnData, chosen_question = chosen_question, user_answer = user_answer)

@app.route('/get-started')
def get_started():
    return render_template("get_started.html", topics = topics, fields = fields)

@app.route('/generated-questions')
def generated_questions():
    global prompt
    global questions
    return render_template("generated_questions.html", topics = topics, fields = fields, prompt = prompt, questions = questions, chosen_topic = chosen_topic, chosen_field = chosen_field)

@app.route('/store_user_data', methods = ['GET','POST'])
def store_user_data():
    global chosen_question
    global user_answer
    json_data = request.get_json()
    json_data_keys = json_data.keys()
    if 'chosen_question' in json_data_keys :
        chosen_question = json_data['chosen_question']
    if 'user_answer' in json_data_keys:
        user_answer = json_data['user_answer']
    return jsonify("Success"), 200

@app.route('/answer-question')
def answer_question():
    global chosen_question
    return render_template("answer_question.html", topics = topics, fields = fields, chosen_question = chosen_question)


if __name__ == "__main__":
    app.run(debug=True)