Update app.py

app.py

@@ -8,31 +8,36 @@ 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?"
+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."):        
     match optionSelection:
         case  "OpenAI API":
-            #model = "gpt-35-turbo"
-            model = "gpt-4"
-            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 = os.getenv("OPENAI_API_KEY")
-            openai.api_version = '2020-11-07'
-            
-            completion = openai.ChatCompletion.create(
-                model = model, 
-                messages = messages,
-                temperature = 0.7
-            )           
-            response = completion["choices"][0]["message"].content
-            return "", response
+            try:
+                #model = "gpt-35-turbo"
+                model = "gpt-4"
+                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 = os.getenv("OPENAI_API_KEY")
+                openai.api_version = '2020-11-07'
+                
+                completion = openai.ChatCompletion.create(
+                    model = model, 
+                    messages = messages,
+                    temperature = 0.7
+                )           
+                response = completion["choices"][0]["message"].content
+                return "", response
+            except openai.error.ServiceUnavailableError:
         case  "Azure OpenAI API":
             return "", ""
         case  "Google PaLM API":
@@ -48,10 +53,15 @@ def test_handler(optionSelection, prompt: str = "Write an introductory paragraph
 with gr.Blocks() as LLMDemoTabbedScreen:
     with gr.Tab("Text-to-Text (Text Completion)"):
         llm_options = gr.Radio(llm_api_options, label="Select one", info="Which service do you want to use?", value="OpenAI API")
-        test_string = gr.Textbox(label="Try String", value=TEST_MESSAGE, lines=2)                   
-        test_string_response = gr.Textbox(label="Response")
-        test_string_output_info = gr.Label(value="Output Info", label="Info")
-        test_button = gr.Button("Try it")
+        with gr.Tab("Open AI"):
+            openai_model = gr.Dropdown(openai_models, value="gpt-4", label="Model", info="Select one, for Natural language")
+
+        with gr.Row():
+            with gr.Column(): 
+                test_string = gr.Textbox(label="Try String", value=TEST_MESSAGE, lines=2)
+                test_string_response = gr.Textbox(label="Response")
+                test_string_output_info = gr.Label(value="Output Info", label="Info")
+                test_button = gr.Button("Try it")
 
 
     test_button.click(