Update app.py

app.py

@@ -1,54 +1,33 @@
-import os
 import openai
-import gradio as gr
-
-# Set up OpenAI API credentials
-api_key = os.environ.get("API_KEY", "API_KEY")
-openai.api_key = api_key
-
-# Set up the GPT-3.5-Turbo model
-model_engine = "gpt-3.5-turbo" # engine parameter used to specify the GPT-3.5-Turbo model
-max_tokens = 1000
-temperature = 0.7
-
-# Define the function that will be called when the user interacts with the Gradio interface
-def chatbot(text):
-    # Generate a response from the OpenAI model
-    response = openai.Completion.create(
-        engine=model_engine,
-        prompt=text,
-        max_tokens=max_tokens,
-        temperature=temperature,
-        stop=None,
-        n=None,
-        logprobs=None,
-        echo=False,
-        presence_penalty=0.5,
-        frequency_penalty=None,
-        best_of=None,
-        response_format="text",
-        v="2022-04-28", # specify the OpenAI API version explicitly
-        model=None, # remove the 'model' parameter
-    )
-    # Extract the generated response from the API response
-    generated_text = response.choices[0].text
-    # Return the response, stripping any leading or trailing whitespace
-    return generated_text.strip()
-
-
-# Define the Gradio interface
-interface = gr.Interface(
-    fn=chatbot,
-    inputs="text",
-    outputs="text",
-    title="Math Teacher Chatbot",
-    description="Ask me any math questions you have and I'll do my best to help!",
-    examples=[
-        ["What's the Pythagorean theorem?"],
-        ["What's the area of a circle with radius 5?"],
-        ["What's the derivative of x^2?"],
-    ]
-)
-
-# Launch the interface
-interface.launch()
+import csv
+
+openai.api_key = "YOUR_API_KEY"
+
+prompt_templates = {"Math Teacher": "I want you to act as a math teacher. I will provide some mathematical equations or concepts, and it will be your job to explain them in easy-to-understand terms. This could include providing step-by-step instructions for solving a problem, demonstrating various techniques with visuals or suggesting online resources for further study. My first request is 'I need help understanding how probability works.'"}
+
+def submit_message(prompt):
+    prompt_template = prompt_templates["Math Teacher"]
+
+    system_prompt = []
+    if prompt_template:
+        system_prompt = [{ "role": "system", "content": prompt_template }]
+
+    prompt_msg = { "role": "user", "content": prompt }
+
+    try:
+        completion = openai.Completion.create(
+            engine="text-davinci-002",
+            prompt=system_prompt + [prompt_msg],
+            max_tokens=1024,
+            n=1,
+            stop=None,
+            temperature=0.7,
+        )
+
+        response = completion.choices[0].text
+    except Exception as e:
+        response = f"Sorry, an error occurred: {e}"
+
+    return response
+
+print(submit_message("I need help understanding how probability works."))