Update app.py

app.py

@@ -1,5 +1,4 @@
 import openai
-import csv
 import os
 
 openai.api_key = os.environ["API_KEY"]
@@ -9,16 +8,16 @@ prompt_templates = {"Math Teacher": "I want you to act as a math teacher. I will
 def submit_message(prompt):
     prompt_template = prompt_templates["Math Teacher"]
 
-    system_prompt = []
+    system_prompt = ""
     if prompt_template:
-        system_prompt = [{ "role": "system", "content": prompt_template }]
+        system_prompt = prompt_template + "\n"
 
-    prompt_msg = { "role": "user", "content": prompt }
+    prompt_msg = prompt + "\n"
 
     try:
         completion = openai.Completion.create(
             engine="text-davinci-002",
-            prompt=system_prompt + [prompt_msg],
+            prompt=system_prompt + prompt_msg,
             max_tokens=1024,
             n=1,
             stop=None,