Update app.py

app.py

@@ -1,6 +1,13 @@
 import gradio as gr
 
-from openai import OpenAI
+from datetime import date
+from langchain.agents import tool
+from langchain.agents.agent_toolkits import create_python_agent
+from langchain.agents import load_tools, initialize_agent
+from langchain.agents import AgentType
+from langchain.tools.python.tool import PythonREPLTool
+from langchain.python import PythonREPL
+from langchain.chat_models import ChatOpenAI
 
 config = {
     "max_tokens": 1000,
@@ -8,30 +15,45 @@ config = {
     "temperature": 0,
 }
 
+@tool
+def time(text: str) -> str:
+    """Returns todays date, use this for any \
+    questions related to knowing todays date. \
+    The input should always be an empty string, \
+    and this function will always return todays \
+    date - any date mathmatics should occur \
+    outside this function."""
+    return str(date.today())
+
 def invoke(openai_api_key, prompt):
     if (openai_api_key == ""):
         raise gr.Error("OpenAI API Key is required.")
     if (prompt == ""):
         raise gr.Error("Prompt is required.")
 
-    content = ""
+    result = ""
     
     try:
-        client = OpenAI(api_key = openai_api_key)
-    
-        completion = client.chat.completions.create(
-            max_tokens = config["max_tokens"],
-            messages = [{"role": "user", "content": prompt}],
-            model = config["model"],
-            temperature = config["temperature"],)
+        llm = ChatOpenAI(temperature=0, model="gpt-4")
+
+        tools = load_tools(["llm-math","wikipedia"], llm=llm)
+
+        agent= initialize_agent(
+            tools + [time], 
+            llm, 
+            agent=AgentType.CHAT_ZERO_SHOT_REACT_DESCRIPTION,
+            handle_parsing_errors=True,
+            verbose = True)
+        
+        result = agent(prompt)
     
-        content = completion.choices[0].message.content
+        #content = completion.choices[0].message.content
     except Exception as e:
         err_msg = e
 
         raise gr.Error(e)
 
-    return content
+    return result
 
 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."""