Spaces:
Running
Running
File size: 1,979 Bytes
dfa45b1 5f264bc dfa45b1 e04b550 f4619cd b700bc7 aeb58f7 dfa45b1 dec5177 e04b550 dfa45b1 6a833cd dfa45b1 6a833cd dfa45b1 f8bbfe3 dfa45b1 6a833cd 5bbf5a8 c5b702c e04b550 5bbf5a8 6a833cd e04b550 dfa45b1 8d5ea99 dfa45b1 f8bbfe3 dfa45b1 0540b69 b1aaabe 0540b69 dfa45b1 0cf28f2 0540b69 dfa45b1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
import gradio as gr
import os
from agent import invoke_agent
from openai import OpenAI
from dotenv import load_dotenv, find_dotenv
_ = load_dotenv(find_dotenv())
config = {
"model": "gpt-4-0613",
"temperature": 0
}
AGENT_OFF = False
AGENT_ON = True
def invoke(openai_api_key, prompt, agent_option):
if (openai_api_key == ""):
raise gr.Error("OpenAI API Key is required.")
if (prompt == ""):
raise gr.Error("Prompt is required.")
if (agent_option is None):
raise gr.Error("Use Agent is required.")
output = ""
try:
if (agent_option == AGENT_OFF):
client = OpenAI(api_key = openai_api_key)
completion = client.chat.completions.create(
messages = [{"role": "user", "content": prompt}],
model = config["model"],
temperature = config["temperature"])
output = completion.choices[0].message.content
else:
completion = invoke_agent(
config["model"],
config["temperature"],
openai_api_key,
prompt)
output = completion["output"]
except Exception as e:
err_msg = e
raise gr.Error(e)
return output
gr.close_all()
demo = gr.Interface(fn = invoke,
inputs = [gr.Textbox(label = "OpenAI API Key", type = "password", lines = 1),
gr.Textbox(label = "Prompt", lines = 1, value = "How does current weather in Los Angeles, New York, and Paris compare in metric and imperial system? Answer in JSON format and include today's date."),
gr.Radio([AGENT_OFF, AGENT_ON], label = "Use Agent", value = AGENT_ON)],
outputs = [gr.Textbox(label = "Completion", lines = 1)],
title = "Real-Time Reasoning Application",
description = os.environ["DESCRIPTION"])
demo.launch() |