Spaces:
Running
Running
import gradio as gr | |
import time | |
from arxiv_agent import * | |
agent = ArxivAgent() | |
def set_profile(name): | |
# Simulate setting the profile based on the name | |
# Replace with actual logic to fetch and set profile | |
names = name.split(" ") | |
for n in names: | |
if len(n) == 0: | |
gr.Info("Please input standard name format.") | |
return None | |
elif n[0].islower(): | |
gr.Info("Please input standard name format.") | |
return None | |
profile = agent.get_profile(name) | |
# import pdb | |
# pdb.set_trace() | |
return profile | |
def edit_profile(profile, author_name): | |
msg = agent.edit_profile(profile, author_name) | |
gr.Info("Edit profile successfully!") | |
return profile | |
def sign_email(profile, email): | |
msg = agent.sign_email(profile, email) | |
gr.Info(email + " signs up successfully!") | |
# return profile | |
def confirm_date(date, profile_input): | |
# Simulate fetching data based on the selected date | |
# data = request.get_json() | |
if len(profile_input) == 0: | |
topic, papers, idea = agent.select_date(date, None) | |
else: | |
topic, papers, idea = agent.select_date(date, profile_input) | |
return topic[0], papers, idea[0] | |
def send_text(query, profile_input): | |
# Simulate sending a query and receiving a response | |
if len(profile_input) <= 1: | |
Ans1, Ans2 = agent.response(query, None) | |
else: | |
Ans1, Ans2 = agent.response(query, profile_input) | |
return Ans1[0], Ans2[0] | |
def send_comment(comment): | |
# Simulate sending a comment | |
message = agent.update_comment(comment) | |
gr.Info("Thank you for your comment!") | |
return message[0] | |
def respond(message, chat_history, profile): | |
bot_message1, bot_message2 = send_text(message, profile) | |
# bot_message1, bot_message2 = "a", "b" | |
chat_history.append((message, None)) | |
chat_history.append((bot_message1, bot_message2)) | |
time.sleep(2) | |
return "", chat_history | |
with gr.Blocks(css="""#chat_container {height: 820px; width: 1000px; margin-left: auto; margin-right: auto;} | |
#chatbot {height: 600px; overflow: auto;} | |
#create_container {height: 750px; margin-left: 0px; margin-right: 0px;} | |
#tokenizer_renderer span {white-space: pre-wrap} | |
""", | |
theme="bethecloud/storj_theme",title="Arxiv Copilot") as app: | |
with gr.Row(): | |
with gr.Column(scale=2): | |
gr.Image( | |
"images/arxiv_copilot.PNG", elem_id="banner-image", show_label=False | |
) | |
with gr.Column(scale=5): | |
gr.Markdown( | |
"""# Arxiv Copilot | |
β‘οΈοΈ **Goals**: Arxiv Copilot aims to provide personalized academic service! | |
β¨ **Guidance**: | |
Step (1) Enter researcher name and generate research profile in "Set your profile!"π§βπΌ | |
Step (2) Select time range and get relevant trending research topic and ideas in "Get trending topics and ideas!"; Here you also can sign up with email to get weekly research news π‘ | |
Step (3) Chat with Arxiv Copilot and choose the better response from two answers in "Chat with Arxiv Copilot!"; Here we appreciate any further feedback π | |
β οΈ **Limitations**: We mainly provide research service related to machine learning field now, other fields will be added in the future. | |
ποΈ **Disclaimer**: User behavior data will be collected for the pure research purpose. If you use this demo, you may implicitly agree to these terms. | |
""" | |
) | |
# gr.Markdown("Provide research service using this demo.") | |
with gr.Accordion("Set your profile!", open=True): | |
gr.Markdown( | |
""" | |
Input your name: You can input your name in standard format to get your profile from arxiv here. Standard examples: Yoshua Bengio. Wrong examples: yoshua bengio, Yoshua bengio, yoshua Bengio. | |
""" | |
) | |
with gr.Row(): | |
with gr.Column(scale=2, min_width=300, variant = 'compact'): | |
name_input = gr.Textbox(label="Input your name:") | |
set_button = gr.Button("Set Profile") | |
with gr.Column(scale=8, min_width=300): | |
with gr.Row(variant = 'compact'): | |
profile_text = gr.Textbox(label="Generated profile (can be edited):", interactive=True, scale=7, lines=5, max_lines=5) | |
edit_button = gr.Button("Edit Profile", scale=1) | |
set_button.click(set_profile, inputs=name_input, outputs=[profile_text]) | |
edit_button.click(edit_profile, inputs=[profile_text, name_input], outputs=[profile_text]) | |
with gr.Accordion("Get trending topics and ideas!", open=True): | |
gr.Markdown( | |
""" | |
(1) Input your email: You can sign up with your email and we will send trending research topics, ideas, and papers related to your profile on Monday of every week. | |
(2) Select time range: We will give you personalized research trend and ideas under selected time range if you have set your profile. Otherwise, general research trend will be provided. | |
""" | |
) | |
with gr.Column(): | |
with gr.Row(): | |
with gr.Column(scale=2.1, min_width=300): | |
with gr.Column(scale=2.1, min_width=300, variant = 'compact'): | |
# gr.Dropdown( | |
# ["day", "week", "bird"], label="Select time range", info="Will add more animals later!" | |
# ), | |
email_input = gr.Textbox(label="Input your email:") | |
sign_button = gr.Button("Sign Up") | |
with gr.Column(scale=2.1, min_width=300, variant = 'compact'): | |
date_choice = gr.Radio(["day", "week", "all"], label="Select time range: ", value="day") | |
date_button = gr.Button("Confirm") | |
papers_text = gr.Textbox(label="Trending Papers", interactive=False, scale=8, lines=12, max_lines=15) | |
with gr.Row(): | |
topic_text = gr.Textbox(label="Trending Topics", interactive=False, scale=5, lines=12, max_lines=15) | |
ideas_text = gr.Textbox(label="Ideas for Trending Topics", interactive=False, scale=5, lines=12, max_lines=15) | |
date_button.click(confirm_date, inputs=[date_choice, profile_text], outputs=[topic_text, papers_text, ideas_text]) | |
sign_button.click(sign_email, inputs=[profile_text, email_input]) | |
with gr.Accordion("Chat with Arxiv Copilot!", open=True): | |
gr.Markdown( | |
""" | |
Each time we will give you two answers. If you prefer the second answer, you can click π below the second answer and the first answer will be removed. If you click π, the second answer will be removed. | |
""" | |
) | |
with gr.Column(): | |
chatbot = gr.Chatbot() | |
with gr.Row(): | |
msg = gr.Textbox(placeholder="Message Arxiv Copilot here...", scale=9, show_label=False) | |
send_button = gr.Button("Send",scale=1) # Adding a Send button | |
clear = gr.ClearButton([msg, chatbot],scale=1) | |
def print_like_dislike(x: gr.LikeData, chat_history): | |
cur_index = x.index[0] | |
if cur_index >= 1 and chat_history[cur_index - 1][1] is None: | |
if x.liked: | |
chat_history[cur_index - 1][1] = chat_history[cur_index][1] | |
agent.update_feedback_thought(chat_history[cur_index - 1][0], chat_history[cur_index][0], chat_history[cur_index][1], 0, 1) | |
# gr.Info("You like the second answer, and the fisrt answer will be removed.") | |
else: | |
agent.update_feedback_thought(chat_history[cur_index - 1][0], chat_history[cur_index][0], chat_history[cur_index][1], 1, 0) | |
chat_history[cur_index - 1][1] = chat_history[cur_index][0] | |
# gr.Info("You dislike the second answer, and the second answer will be removed.") | |
chat_history.remove(chat_history[cur_index]) | |
else: | |
gr.Info("You have gave your feedback. You can ask more questions.") | |
return chat_history | |
msg.submit(respond, [msg, chatbot, profile_text], [msg, chatbot]) # Set up the action for the Send button | |
send_button.click(respond, inputs=[msg, chatbot, profile_text], outputs=[msg, chatbot]) | |
chatbot.like(print_like_dislike, [chatbot], [chatbot]) | |
with gr.Row(): | |
comment_input = gr.Textbox(label="With Arxiv Copilot, how many minutes do you save to obtain the same amount of information?", scale=9, lines=3) | |
comment_button = gr.Button(value="Comment", scale=1) | |
comment_button.click(send_comment, inputs=comment_input, outputs=None) | |
app.launch() | |