|
|
import gradio as gr |
|
|
import os, threading |
|
|
|
|
|
|
|
|
from crew import get_crew |
|
|
from helper 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() |
