Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import os
 
-from Agent import invoke_agent
+from agent import invoke_agent
 from openai import OpenAI
 
 from dotenv import load_dotenv, find_dotenv
@@ -9,7 +9,7 @@ _ = load_dotenv(find_dotenv())
 
 config = {
     "model": "gpt-4-0613",
-    "temperature": 0,
+    "temperature": 0
 }
 
 AGENT_OFF = False
@@ -32,14 +32,15 @@ def invoke(openai_api_key, prompt, agent_option):
             completion = client.chat.completions.create(
                 messages = [{"role": "user", "content": prompt}],
                 model = config["model"],
-                temperature = config["temperature"],)
+                temperature = config["temperature"])
     
             output = completion.choices[0].message.content
         else:
-            completion = invoke_agent(model, 
-                                      temperature, 
-                                      openai_api_key, 
-                                      prompt)
+            completion = invoke_agent(
+                config["model"], 
+                config["temperature"], 
+                openai_api_key, 
+                prompt)
     
             output = completion["output"]
     except Exception as e: