app.py

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!"; Here you also can sign up with email to get monthly research news 🧑‍💼
                
                Step (2) Select time range and get relevant topic trend and ideas in "Get research trend and ideas!"💡

                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 research trend and ideas!", open=True):
        gr.Markdown(
            """
            (1) Input your email: You can sign up with your email and we will send research trend, ideas, and top papers related to your profile on 1st of every month.

            (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="Trend Papers", interactive=False, scale=8, lines=12, max_lines=15)

            with gr.Row():
                topic_text = gr.Textbox(label="Topic Trend", interactive=False, scale=5, lines=12, max_lines=15)

                ideas_text = gr.Textbox(label="Ideas Related to Topic Trend", 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()