Update app.py

app.py

@@ -8,7 +8,6 @@ import openai
 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!")
@@ -34,12 +33,9 @@ def test_handler(optionSelection, prompt: str = "Write an introductory paragraph
                 model = model, 
                 messages = messages,
                 temperature = 0.7
-            )
-            
-            print(completion)
+            )           
             response = completion["choices"][0]["message"].content
-            print(response)    
-            return message, response
+            return completion, response
         case  "Azure OpenAI API":
             return "", ""
         case  "Google PaLM API":
@@ -56,8 +52,12 @@ with gr.Blocks() as LLMDemoTabbedScreen:
         test_string_output_info = gr.Label(value="Output Info", label="Info")
         test_button = gr.Button("Try it")
 
+
+test_button.click(
+        fn=test.test_handler,
+        inputs=[llm_api_options, test_string],
+        outputs=[test_string_output_info, test_string_response]
+)
     
-	
-	
 if __name__ == "__main__":
     LLMDemoTabbedScreen.launch()