Update app.py

app.py

@@ -10,6 +10,9 @@ config = {
     "temperature": 0,
 }
 
+AGENT_OFF = False
+AGENT_ON  = True
+
 @tool
 def time(text: str) -> str:
     """Returns todays date, use this for any \
@@ -20,28 +23,33 @@ def time(text: str) -> str:
     outside this function."""
     return str(date.today())
 
-def invoke(openai_api_key, prompt):
+def invoke(openai_api_key, prompt, agent_option):
     if (openai_api_key == ""):
         raise gr.Error("OpenAI API Key is required.")
     if (prompt == ""):
         raise gr.Error("Prompt is required.")
+    if (agent_option is None):
+        raise gr.Error("Use Agent is required.")
 
     output = ""
     
     try:
-        llm = ChatOpenAI(model_name = config["model_name"],
-                         openai_api_key = openai_api_key, 
-                         temperature = config["temperature"])
+        if (agent_option == AGENT_OFF):
+            #
+        else:
+            llm = ChatOpenAI(model_name = config["model_name"],
+                             openai_api_key = openai_api_key, 
+                             temperature = config["temperature"])
 
-        agent = initialize_agent([time], 
-                                 llm, 
-                                 agent = AgentType.CHAT_ZERO_SHOT_REACT_DESCRIPTION,
-                                 handle_parsing_errors = True,
-                                 verbose = True)
+            agent = initialize_agent([time], 
+                                     llm, 
+                                     agent = AgentType.CHAT_ZERO_SHOT_REACT_DESCRIPTION,
+                                     handle_parsing_errors = True,
+                                     verbose = True)
         
-        completion = agent(prompt)
+            completion = agent(prompt)
     
-        output = completion["output"]
+            output = completion["output"]
     except Exception as e:
         err_msg = e
 
@@ -55,8 +63,9 @@ description = """<a href='https://www.gradio.app/'>Gradio</a> UI using the <a hr
 gr.close_all()
 
 demo = gr.Interface(fn = invoke, 
-                    inputs = [gr.Textbox(label = "OpenAI API Key", type = "password", lines = 1),
-                              gr.Textbox(label = "Prompt", lines = 1, value = "What is today's date?")],
+                    inputs = [gr.Textbox(label = "OpenAI API Key", type = "password", lines = 1, value = "sk-"),
+                              gr.Textbox(label = "Prompt", lines = 1, value = "What is today's date?"),
+                              gr.Radio([AGENT_OFF, AGENT_ON], label = "Use Agent", value = AGENT_OFF)],
                     outputs = [gr.Textbox(label = "Completion", lines = 1)],
                     title = "Generative AI - LLM & Agent",
                     description = description,