Add app.py

app.py

@@ -0,0 +1,66 @@
+import os
+from dotenv import load_dotenv
+import openai
+import random
+import requests
+from ast import literal_eval
+import json
+from enum import Enum
+import gradio as gr
+
+load_dotenv()
+openai.api_key = os.getenv("OPENAI_API_KEY")
+CHAT_ENDPOINT="https://api.openai.com/v1/chat/completions"
+CHAT_MODEL = "gpt-3.5-turbo"
+CHAT_AUTH = {"Authorization": "Bearer " + openai.api_key}
+gpt_history = []
+
+class ChatRoles():
+    SYSTEM = "system"
+    ASSISTANT = "assistant"
+    USER = "user"
+
+def get_assistant_response():
+    params = {
+        "model": CHAT_MODEL,
+        "messages": gpt_history,
+    }
+    response = requests.post(url=CHAT_ENDPOINT, json=params, headers=CHAT_AUTH)
+    response_message = literal_eval(response.content.decode("utf-8"))["choices"][0]["message"]["content"]
+    gpt_history.append({"role": ChatRoles.ASSISTANT, "content": response_message})
+    print("\n" + response_message)
+    return response_message
+
+instructions = {
+    1: '''Tell the user you are an AI-powered college counselor from Cledge helping students brainstorm ideas for their personal statement. 
+        Generate 5 questions to help them brainstorm.''',
+    5: "Based on these responses, ask follow up questions that help discover common themes between the experiences.",
+    9: "Based on these responses, think of 5 ideas for personal statement essays. Write a synopsis of each idea."
+}
+
+def grad_demo():
+    with gr.Blocks() as demo:
+        def user(user_message, history):
+            gpt_history.append({"role": ChatRoles.USER, "content": user_message})
+            return "", history + [[user_message, None]]
+        def bot(history):
+            step = len(history)
+            print(f"STEP: {step}")
+            if step in instructions:
+                gpt_history.append({"role": ChatRoles.SYSTEM, "content": instructions[step]})
+            bot_message = get_assistant_response()
+            history[-1][1] = bot_message
+            gpt_history.append({"role": ChatRoles.ASSISTANT, "content": bot_message})
+            return history
+        chatbot = gr.Chatbot(value = lambda : bot([[None, None]]))
+        msg = gr.Textbox()
+        clear = gr.Button("Clear")
+        msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+            bot, chatbot, chatbot
+        )
+        clear.click(lambda: None, None, chatbot, queue=False)
+    demo.launch()
+
+if __name__ == "__main__":
+    grad_demo()
+    #main()