Update app.py

app.py

@@ -1,16 +1,12 @@
 import gradio as gr
 import os
 
-from datetime import date
-from langchain.agents import AgentType, initialize_agent, load_tools, tool
-from langchain.chat_models import ChatOpenAI
+from Agent import invoke_agent
 from openai import OpenAI
 
 from dotenv import load_dotenv, find_dotenv
 _ = load_dotenv(find_dotenv())
 
-OPENWEATHERMAP_API_KEY = os.environ["OPENWEATHERMAP_API_KEY"]
-
 config = {
     "model": "gpt-4-0613",
     "temperature": 0,
@@ -19,13 +15,6 @@ config = {
 AGENT_OFF = False
 AGENT_ON  = True
 
-@tool
-def date_tool(text: str) -> str:
-    """Returns today's date. Use this for any questions related to knowing today's date. 
-       The input should always be an empty string, and this function will always return today's date. 
-       Any date mathematics should occur outside this function."""
-    return str(date.today())
-
 def invoke(openai_api_key, prompt, agent_option):
     if (openai_api_key == ""):
         raise gr.Error("OpenAI API Key is required.")
@@ -47,7 +36,10 @@ def invoke(openai_api_key, prompt, agent_option):
     
             output = completion.choices[0].message.content
         else:
-            completion = invoke_agent(openai_api_key, prompt)
+            completion = invoke_agent(model, 
+                                      temperature, 
+                                      openai_api_key, 
+                                      prompt)
     
             output = completion["output"]
     except Exception as e: