Update app.py

app.py

@@ -4,6 +4,7 @@ import gradio as gr
 import openai
 
 #from dotenv import load_dotenv
+
 #load_dotenv()
 
 llm_api_options = ["OpenAI API","Azure OpenAI API","Google PaLM API", "Llama 2"]
@@ -41,7 +42,8 @@ def openai_text_completion(prompt: str, model: str):
 def test_handler(optionSelection, prompt: str = TEST_MESSAGE, model: str ="gpt-4"):
     match optionSelection:
         case  "OpenAI API":
-            openai_text_completion(prompt,model)
+            message, response = openai_text_completion(prompt,model)
+            return message, response
         case  "Azure OpenAI API":
             return "", ""
         case  "Google PaLM API":