'added ChatGPT section and openai package requirement'

app.py

@@ -12,6 +12,8 @@ import pandas as pd
 
 from model import BertLightningModel
 
+import openai
+
 #######################################################################################################################
 current_path = os.path.abspath(os.path.dirname(__file__))
 
@@ -109,6 +111,18 @@ def read_pdf(file):
 
     return all_page_text
 
+def openai_chat(prompt, model, max_tokens):
+    response = openai.Completion.create(
+        engine=model,
+        prompt=prompt,
+        max_tokens=max_tokens,
+        n=1,
+        stop=None,
+        temperature=0.7,
+    )
+    message = response.choices[0].text.strip()
+    return message
+
 def main():
     head_col = st.columns([1,8])
     with head_col[0]:
@@ -307,6 +321,29 @@ def main():
             final_df_ex = grades_df_ex.merge(evals_df_ex, on='Filename')
             st.dataframe(final_df_ex)
 
+            # ChatGPT Evaluation Section
+            st.subheader("")
+            st.markdown("***")
+            st.subheader("ChatGPT Evaluation")
+            openai.api_key = "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"
+
+            chatgpt_prompts = []
+            chatgpt_responses = []
+            for answer in ex_essays:
+                prompt = f"Evaluate the following essay using the Criterion: [cohesion, syntax, vocabulary, phraseology, grammar, conventions]. " \
+                         f"Use a {min_score_ex} to {max_score_ex} score range for each, and provide one final score using the same score range " \
+                         f"and summarized feedback. Essay: \n"
+                prompt += answer
+                response = openai_chat(prompt=prompt, model="text-davinci-002", max_tokens=1024)
+                chatgpt_prompts.append(prompt)
+                chatgpt_responses.append(response)
+
+            chatgpt_ta_val = ""
+            for i, val in enumerate(chatgpt_prompts):
+                chatgpt_ta_val = chatgpt_ta_val + val + "\n" + chatgpt_responses[i] + "\n"
+            chatgpt_ta = st.text_area("ChatGPT", placeholder="ChatGPT's evaluations will display here.",
+                                                  value=chatgpt_ta_val, height=500, disabled=True)
+
 
 if __name__ == '__main__':
     main()

requirements.txt

@@ -9,4 +9,5 @@ seaborn==0.12.1
 pytorch-lightning==1.8.3
 sentencepiece==0.1.97
 docx2txt==0.8
-pypdf2==3.0.1
+pypdf2==3.0.1
+openai