Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+
 import gradio as gr
 import openai
 
@@ -9,9 +10,6 @@ llm_api_options = ["OpenAI API","Azure OpenAI API","Google PaLM API", "Llama 2"]
 TEST_MESSAGE = "My favorite TV shows are The Mentalist, The Blacklist, Designated Survivor, and Unforgettable. What are ten series that I should watch next?"
 
 def test_handler(optionSelection, prompt: str = "Write an introductory paragraph to explain Generative AI to the reader of this content."):        
-    if optionSelection not in llm_api_options:
-        raise ValueError("Invalid choice!")
-
     match optionSelection:
         case  "OpenAI API":
             #model = "gpt-35-turbo"
@@ -21,7 +19,6 @@ def test_handler(optionSelection, prompt: str = "Write an introductory paragraph
             
             messages = [
                 {"role": "user", "content": f"{prompt}"},
-
                 {"role": "system", "content": f"{system_prompt}"},
                 {"role": "assistant", "content": f"{assistant_prompt}"}
             ]
@@ -30,20 +27,23 @@ def test_handler(optionSelection, prompt: str = "Write an introductory paragraph
             openai.api_version = '2020-11-07'
             
             completion = openai.ChatCompletion.create(
-                model = model
-, 
+                model = model, 
                 messages = messages,
                 temperature = 0.7
             )           
             response = completion["choices"][0]["message"].content
-            return completion, response
+            return "", response
         case  "Azure OpenAI API":
             return "", ""
         case  "Google PaLM API":
             return "", ""
         case  "Llama 2":
             return "", ""
+        else:
+            if optionSelection not in llm_api_options:
+                return ValueError("Invalid choice!"), ""
 
+        
 
 with gr.Blocks() as LLMDemoTabbedScreen:
     with gr.Tab("Text-to-Text (Text Completion)"):