File size: 9,045 Bytes
52d2ab4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
7d66a52
 
 
 
 
52d2ab4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
cc6a064
52d2ab4
cc6a064
52d2ab4
 
 
 
 
 
 
 
 
 
 
 
 
 
7d66a52
52d2ab4
 
 
7d66a52
1f02b53
52d2ab4
7d66a52
 
 
 
52d2ab4
 
7d66a52
cc6a064
52d2ab4
 
cc6a064
7d66a52
 
52d2ab4
 
 
 
7d66a52
 
 
 
 
 
 
 
 
 
eda032b
52d2ab4
 
eda032b
52d2ab4
eda032b
52d2ab4
 
7d66a52
52d2ab4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1f02b53
52d2ab4
 
 
 
 
 
 
 
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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
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()