Update app.py

app.py

@@ -1,10 +1,8 @@
 import os
-
 import gradio as gr
 import openai
 import google.generativeai as palm
 
-
 llm_api_options = ["OpenAI API","Azure OpenAI API","Google PaLM API", "Llama 2"]
 TEST_MESSAGE = "Write an introductory paragraph to explain Generative AI to the reader of this content."
 openai_models = ["gpt-4", "gpt-4-0613", "gpt-4-32k", "gpt-4-32k-0613", "gpt-3.5-turbo",
@@ -19,69 +17,60 @@ azure_deployment_name = os.getenv("AZURE_OPENAI_DEPLOYMENT_NAME")
 google_palm_key = os.getenv("GOOGLE_PALM_AI_API_KEY")
 openai_api_key = os.getenv("OPENAI_API_KEY")
 azure_openai_api_key = os.getenv("AZURE_OPENAI_KEY")
+temperature = 0.7
 
 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 = openai_api_key
         openai.api_version = '2020-11-07'
-        
         completion = openai.ChatCompletion.create(
             model = model, 
             messages = messages,
-            temperature = 0.7
+            temperature = temperature
         )           
         response = completion["choices"][0]["message"].content
         return "", response
-    except openai.error.ServiceUnavailableError:
+    except Exception as exception:
         print(f"Exception Name: {type(exception).__name__}")
         print(exception)
-        return f" {optionSelection} test_handler Error - {exception}", ""
+        return f" openai_text_completion 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 = azure_openai_api_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
+            temperature = temperature
         )           
         response = completion["choices"][0]["message"].content
         return "", response
-    except openai.error.ServiceUnavailableError:
+    except Exception as exception:
         print(f"Exception Name: {type(exception).__name__}")
         print(exception)
-        return f" {optionSelection} test_handler Error - {exception}", ""
+        return f" azure_openai_text_completion Error - {exception}", ""
 
 
 def palm_text_completion(prompt: str, model: str):
-    try:
-        temperature = 0.7
+    try:        
         candidate_count = 1
         top_k = 40
         top_p = 0.95
@@ -103,10 +92,10 @@ def palm_text_completion(prompt: str, model: str):
           prompt=prompt
         )
         return "", response.result
-    except openai.error.ServiceUnavailableError:
+    except Exception as exception:
         print(f"Exception Name: {type(exception).__name__}")
         print(exception)
-        return f" {optionSelection} test_handler Error - {exception}", ""
+        return f" palm_text_completion Error - {exception}", ""
 
 def test_handler(optionSelection, prompt: str = TEST_MESSAGE, openai_model_name: str ="gpt-4", google_model_name: str ="models/text-bison-001"):
     match optionSelection: