Update app.py

app.py

@@ -12,33 +12,36 @@ openai_models = ["gpt-4", "gpt-4-0613", "gpt-4-32k", "gpt-4-32k-0613", "gpt-3.5-
                      "gpt-3.5-turbo-0613", "gpt-3.5-turbo-16k", "gpt-3.5-turbo-16k-0613", "text-davinci-003", 
                      "text-davinci-002", "text-curie-001", "text-babbage-001", "text-ada-001"]
 
+def openai_text_completion(prompt: str, model: str):
+    try:
+        system_prompt: str = "Explain in detail to help student understand the concept.", 
+        assistant_prompt: str = None,
+
+        messages = [
+            {"role": "user", "content": f"{prompt}"},
+            {"role": "system", "content": f"{system_prompt}"},
+            {"role": "assistant", "content": f"{assistant_prompt}"}
+        ]
+        
+        openai.api_key = os.getenv("OPENAI_API_KEY")
+        openai.api_version = '2020-11-07'
+        
+        completion = openai.ChatCompletion.create(
+            model = model, 
+            messages = messages,
+            temperature = 0.7
+        )           
+        response = completion["choices"][0]["message"].content
+        return "", response
+    except openai.error.ServiceUnavailableError:
+        print(f"Exception Name: {type(exception).__name__}")
+        print(exception)
+        return f" {optionSelection} test_handler Error - {exception}", ""
+
 def test_handler(optionSelection, prompt: str = TEST_MESSAGE, model: str ="gpt-4"):
     match optionSelection:
         case  "OpenAI API":
-            try:
-                system_prompt: str = "Explain in detail to help student understand the concept.", 
-                assistant_prompt: str = None,
-
-                messages = [
-                    {"role": "user", "content": f"{prompt}"},
-                    {"role": "system", "content": f"{system_prompt}"},
-                    {"role": "assistant", "content": f"{assistant_prompt}"}
-                ]
-                
-                openai.api_key = os.getenv("OPENAI_API_KEY")
-                openai.api_version = '2020-11-07'
-                
-                completion = openai.ChatCompletion.create(
-                    model = model, 
-                    messages = messages,
-                    temperature = 0.7
-                )           
-                response = completion["choices"][0]["message"].content
-                return "", response
-            except openai.error.ServiceUnavailableError:
-                print(f"Exception Name: {type(exception).__name__}")
-                print(exception)
-                return f" {optionSelection} test_handler Error - {exception}", ""
+            openai_text_completion(prompt,model)
         case  "Azure OpenAI API":
             return "", ""
         case  "Google PaLM API":