Update app.py

app.py

@@ -13,9 +13,13 @@ 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"]
 
+azure_endpoint = os.getenv("AZURE_OPENAI_ENDPOINT")
+azure_deployment_name = os.getenv("AZURE_OPENAI_DEPLOYMENT_NAME")
+
 def openai_text_completion(prompt: str, model: str):
     try:
-        system_prompt: str = "Explain in detail to help student understand the concept.", 
+        system_prompt: str = "Explain in detail to help student understand the concept.",
+ 
         assistant_prompt: str = None,
 
         messages = [
@@ -39,6 +43,37 @@ def openai_text_completion(prompt: str, model: str):
         print(exception)
         return f" {optionSelection} test_handler Error - {exception}", ""
 
+def azure_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("AZURE_OPENAI_KEY")
+        openai.api_type = "azure"
+        openai.api_version = "2023-05-15" 
+        openai.api_base = f"https://{azure_endpoint}.openai.azure.com"
+        
+        completion = openai.ChatCompletion.create(
+            model = model, 
+            engine = azure_deployment_name,
+            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":