| import gradio as gr | |
| import os, threading | |
| from crew import get_crew | |
| from util import get_questions | |
| lock = threading.Lock() | |
| def invoke(level, question, file_name, ground_truth, openai_api_key): | |
| if not question: | |
| raise gr.Error("Question is required.") | |
| if not openai_api_key: | |
| raise gr.Error("OpenAI API Key is required.") | |
| with lock: | |
| answer = "" | |
| try: | |
| os.environ["OPENAI_API_KEY"] = openai_api_key | |
| answer = get_crew().kickoff(inputs={"topic": question}) | |
| except Exception as e: | |
| raise gr.Error(e) | |
| finally: | |
| del os.environ["OPENAI_API_KEY"] | |
| return answer | |
| gr.close_all() | |
| demo = gr.Interface(fn=invoke, | |
| inputs=[gr.Radio([1, 2, 3], label="Level"), | |
| gr.Textbox(label="Question"), | |
| gr.Textbox(label="File Name"), | |
| gr.Textbox(label="Ground Truth"), | |
| gr.Textbox(label="OpenAI API Key", type="password")], | |
| outputs=[gr.Textbox(label="Answer", lines=1, interactive=False)], | |
| title="General AI Assistant (GAIA) 🤖🤝🤖", | |
| description=os.environ["DESCRIPTION"], | |
| examples=get_questions(), | |
| cache_examples=False | |
| ) | |
| demo.launch() |