Update app.py

app.py

@@ -7,17 +7,15 @@ import openai
 #load_dotenv()
 
 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?"
+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", 
                      "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 test_handler(optionSelection, prompt: str = "Write an introductory paragraph to explain Generative AI to the reader of this content."):        
+def test_handler(optionSelection, prompt: str = TEST_MESSAGE, model):
     match optionSelection:
         case  "OpenAI API":
             try:
-                #model = "gpt-35-turbo"
-                model = "gpt-4"
                 system_prompt: str = "Explain in detail to help student understand the concept.", 
                 assistant_prompt: str = None,
 
@@ -69,7 +67,7 @@ with gr.Blocks() as LLMDemoTabbedScreen:
 
     test_button.click(
             fn=test_handler,
-            inputs=[llm_options, test_string],
+            inputs=[llm_options, test_string, openai_model],
             outputs=[test_string_output_info, test_string_response]
     )