Spaces:
Runtime error
Runtime error
File size: 4,997 Bytes
4d7183d b80d291 4d7183d b80d291 4d7183d b80d291 4d7183d b80d291 4d7183d b80d291 4d7183d b80d291 4d7183d b80d291 4d7183d 241d2a5 4d7183d |
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 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
import gradio as gr
from config import check_openai_api_key
from agent.research_agent import ResearchAgent
from agent.toolkits import english_polishing
from statics.style import *
check_openai_api_key()
history = ""
history_num = 0
history_tasks = []
max_histroy_num = 30
def run_agent(task, agent, report_type):
global history, history_num, history_tasks
history_num += 1
history_tasks.append(task)
assistant = ResearchAgent(task, agent)
yield from assistant.write_report(report_type)
with gr.Blocks(theme=gr.themes.Base(),
title="AI Research Assistant",
css=css) as demo:
gr.HTML(top_bar)
with gr.Tab(label="Report"):
with gr.Column():
gr.HTML(research_report_html)
research_report = gr.Markdown(value=" **Research report will appear here...**",
elem_classes="output")
with gr.Row():
agent_type = gr.Dropdown(label="# Agent Type",
value="Default Agent",
interactive=True,
allow_custom_value=False,
choices=["Default Agent",
"Business Analyst Agent",
"Finance Agent",
"Travel Agent",
"Academic Research Agent",
"Computer Security Analyst Agent",
"Clinical Medicine Agent",
"Basic Medicine Agent",
"Social Science Research Agent"])
report_type = gr.Dropdown(label="# Report Type",
value="Research Report",
interactive=True,
allow_custom_value=False,
choices=["Research Report",
"Resource Report",
"Outline Report"])
input_box = gr.Textbox(label="# What would you like to research next?", placeholder="Enter your question here")
submit_btn = gr.Button("Generate Report", elem_id="primary-btn")
gr.Examples(["Should I invest in the Large Language Model industry in 2023?",
"Is it advisable to make investments in the electric car industry during the year 2023?",
"What constitutes the optimal approach for investing in the Bitcoin industry during the year 2023?",
"What are the most recent advancements in the domain of superconductors as of 2023?"],
inputs=input_box)
with gr.Accordion(label="# Research History (Max 30)", elem_id="history", open=False):
research_history = gr.Markdown()
def store_research(content):
global history_num, history_tasks, history
if 0 < history_num <= max_histroy_num:
history += f'<details>\
<summary>Research History {history_num}: <i>{history_tasks[-1]}</i></summary>\
<div id="history_box">{content}</div>\
</details>'
return history
submit_btn.click(run_agent, inputs=[input_box, agent_type, report_type], outputs=research_report)\
.then(store_research, inputs=[research_report], outputs=research_history)
with gr.Tab("English Polishing"):
gr.HTML(english_polishing_html)
polished_result = gr.Markdown(" **Polished result will appear here...**", elem_classes="output")
sentences = gr.Textbox(label="# What would you like to polish?", placeholder="Enter your sentence here")
with gr.Row():
polish_btn = gr.Button("Polish", elem_id="primary-btn")
save_btn = gr.Button("Save", elem_id="primary-btn")
polish_btn.click(english_polishing, inputs=[sentences], outputs=polished_result)
def save_result(history, origin, result):
history += f"\n**Origin** : {origin}\n\n**Polished Result** : {result}"
return history
gr.HTML(history_result_html)
history_result = gr.Markdown(" **History result will appear here...**")
save_btn.click(save_result, inputs=[history_result, sentences, polished_result], outputs=history_result)
with gr.Tab("Literature Review"):
gr.HTML(literature_review_html)
demo.queue().launch() |