Update app.py

app.py

@@ -26,7 +26,7 @@ def invoke(openai_api_key, prompt):
     if (prompt == ""):
         raise gr.Error("Prompt is required.")
 
-    result = ""
+    output = ""
     
     try:
         llm = ChatOpenAI(model_name = config["model_name"],
@@ -39,15 +39,15 @@ def invoke(openai_api_key, prompt):
                                  handle_parsing_errors = True,
                                  verbose = True)
         
-        result = agent(prompt)
+        completion = agent(prompt)
     
-        #content = completion.choices[0].message.content
+        output = completion.output
     except Exception as e:
         err_msg = e
 
         raise gr.Error(e)
 
-    return result
+    return output
 
 description = """<a href='https://www.gradio.app/'>Gradio</a> UI using the <a href='https://openai.com/'>OpenAI</a> API 
                  with <a href='https://openai.com/research/gpt-4'>gpt-4</a> model."""