Update app.py

app.py

@@ -35,22 +35,26 @@ def invoke(openai_api_key, prompt, agent_option):
     output = ""
     
     try:
+        llm = ChatOpenAI(model_name = config["model_name"],
+                         openai_api_key = openai_api_key, 
+                         temperature = config["temperature"])
+
         if (agent_option == AGENT_OFF):
-            output = "TODO"
+            agent = initialize_agent([],
+                                     llm,
+                                     agent = AgentType.CHAT_ZERO_SHOT_REACT_DESCRIPTION,
+                                     handle_parsing_errors = True,
+                                     verbose = True)
         else:
-            llm = ChatOpenAI(model_name = config["model_name"],
-                             openai_api_key = openai_api_key, 
-                             temperature = config["temperature"])
-
-            agent = initialize_agent([time], 
-                                     llm, 
+            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