Spaces:
Running
Running
| from utils.scrapers import codeforces, usaco | |
| from utils.chatbot import Chat | |
| from utils.prompts import * | |
| import os | |
| import gradio as gr | |
| import random | |
| import time | |
| dir_path = os.path.dirname(os.path.realpath(__file__)) | |
| CHAT_PATH = dir_path + '/saved_chats/' | |
| used_ids = os.listdir(CHAT_PATH) | |
| ind = 0 | |
| for i in used_ids: | |
| try: | |
| ind = max(ind, int(i.split('#')[1])) | |
| except: | |
| pass | |
| chats = 500 | |
| last_update = time.time() | |
| with gr.Blocks(title = "CP Helper Bot") as demo: | |
| ins = gr.Markdown(""" | |
| # Usage Instructions | |
| First, paste a valid USACO (https://usaco.org) or codeforces (https://codeforces.com) problem link into the first text box input. | |
| Then, paste the link to an official editorial for the problem. Support for other platforms may (or may not) be coming soon. | |
| Choose between either 'hint' mode or 'editorial' mode. You have to reload the page to start a new conversation, and conversations are not linked with an "account" or "session". | |
| ## Hint Mode | |
| Chat with the bot about your ideas to solve the problem, and it will give you hints to guide your thinking. | |
| ## Editorial Mode | |
| Chat with the bot about the editorial to the problem. | |
| # Feedback | |
| Feedback is not logged or stored in any way by the feedback options below the chatbot (which have been removed). Please contact me directly if the bot is broken or for any feedback. | |
| Thanks for using this bot, and I hope you found it helpful! | |
| This is an open source project maintained by Rithwik Gupta (https://github.com/Rithwik-G, beaboss on codeforces). If you would like to contribute (by adding scrapers for other OJs, adding features, adjusting prompts, making the feedback system better, etc.), you can make a pull request on HuggingFace community tab or GitHub. If you want to collaborate, feel free to DM me on discord @_rithwik or at rithwikca2020@gmail.com. | |
| If you found this bot useful, please star the GitHub repo :) | |
| https://github.com/Rithwik-G/Competitive-Programming-Bot | |
| """) | |
| interface = gr.Chatbot(height='300px', placeholder="Provide a link for the problem you're solving.", | |
| latex_delimiters = [{"left": "\\(", "right" : "\\)", 'display':False}, {"left": "$$$", "right" : "$$$", 'display':False}, {"left": "$", "right" : "$", 'display':False}]) | |
| msg = gr.Textbox(label = "Chat with the bot") | |
| # clear = gr.ClearButton([msg, interface]) | |
| state = gr.State() | |
| def cp_helper(message, history, state): | |
| global chats, last_update | |
| current_time = time.time() | |
| day = 24 * 60 * 60 | |
| if (current_time - last_update >= day): | |
| last_update=current_time | |
| chats=500 | |
| if (chats <= 0): | |
| history.append((message, "Too many chats today, try again later.")) | |
| return "", history, state | |
| chats-=1 | |
| global ind | |
| if (state is None): | |
| # state = {'chatgpt': None, | |
| # 'problem_url': None, | |
| # 'edi_url' : None, | |
| # 'problem_text' : usaco.problem("https://usaco.org/index.php?page=viewproblem2&cpid=1421"), | |
| # 'edi_text' : usaco.editorial("", "https://usaco.org/current/data/sol_prob3_bronze_open24.html"), | |
| # 'first' : True, | |
| # 'style' : 'hint'} | |
| # state['chatgpt'] = Chat(HINT_MESSAGE(state['problem_text'], state['edi_text'])) # HINT_MESSAGE(state['problem_text'], state['edi_text']) | |
| # print('Edi', state['edi_text']) | |
| # assert(False) | |
| # state = {'chatgpt': None, | |
| # 'problem_url': None, | |
| # 'edi_url' : None, | |
| # 'problem_text' : codeforces.problem("https://codeforces.com/contest/1592/problem/E"), | |
| # 'edi_text' : codeforces.editorial("https://codeforces.com/contest/1592/problem/E", "https://codeforces.com/blog/entry/95583"), | |
| # 'first' : True, | |
| # 'style' : 'hint'} | |
| # state['chatgpt'] = Chat(HINT_MESSAGE(state['problem_text'], state['edi_text'])) # | |
| # print(state['problem_text'], state['edi_text']) | |
| ind = random.randint(1, 100000) | |
| state = {'chatgpt': None, | |
| 'problem_url': None, | |
| 'edi_url' : None, | |
| 'problem_text' : '', | |
| 'edi_text' : '', | |
| 'first' : True, | |
| 'style' : None, | |
| 'id': '#' + str(ind)} | |
| # state['chatgpt'] = Chat(HINT_MESSAGE(state['problem_text'], state['edi_text'])) | |
| ind += 1 | |
| bot = "" | |
| if (state['problem_text'] == ''): | |
| try: | |
| state['problem_url'] = message | |
| if ('usaco' in message.lower()): | |
| state['problem_text'] = usaco.problem(state['problem_url'])['statement'] | |
| elif ('codeforces' in message.lower()): | |
| state['problem_text'] = codeforces.problem(state['problem_url'])['statement'] | |
| else: | |
| assert(False) | |
| bot = "Thanks! Now could you provide a link to the editorial." | |
| print(state['problem_text']) | |
| except Exception as e: | |
| print(e) | |
| bot = "There was something wrong with the link. Could you paste it again?" | |
| elif (state['edi_text'] == ''): | |
| try: | |
| state['edi_url'] = message | |
| if ('usaco' in message.lower()): | |
| # cbot = Chat(observation_generation) | |
| state['edi_text'] = usaco.editorial(state['problem_url'], state['edi_url']) | |
| elif ('codeforces' in message.lower()): | |
| # cbot = Chat(observation_generation) | |
| state['edi_text'] = codeforces.editorial(state['problem_url'], state['edi_url']) | |
| else: | |
| assert(False) | |
| assert(len(state['edi_text'])) | |
| print('EDITORIAL\n', state['edi_text']) | |
| state['first']=True | |
| bot = "Would you like hints to help you solve the problem or an interactive explanation of the editorial (hint/editorial)? " | |
| except Exception as e: | |
| print(e) | |
| bot = "There was something wrong with the link. Could you paste it again?" | |
| elif (state['style'] == None): | |
| message = message.lower() | |
| if ('editorial' in message): | |
| state['style'] = 'editorial' | |
| state['chatgpt'] = Chat(EDI_MESSAGE(state['problem_text'], state['edi_text'])) | |
| bot = state['edi_text'] + '\nLet me know if you have any questions.' | |
| elif ('hint' in message): | |
| state['style'] = 'hint' | |
| state['chatgpt'] = Chat(HINT_MESSAGE(state['problem_text'], state['edi_text'])) | |
| bot += "What are you current thoughts on the problem? " | |
| else: | |
| bot += "Would you like hints to help you solve the problem or an interactive explanation of the editorial (hint/editorial)? " | |
| if (len(bot) == 0): | |
| if (state['style'] == 'hint'): | |
| if (state['first']): | |
| state['first']=False | |
| bot = state['chatgpt'].chat(HINT_TEXT(message, state['problem_text'], state['edi_text'])) | |
| else: | |
| bot = state['chatgpt'].chat(HINT_TEXT(message, state['problem_text'], state['edi_text']))# + '\nExplain everything as if I have never seen the problem or editorial before in a maximum of 2 paragraphs.') | |
| else: | |
| state['first']=False | |
| bot = state['chatgpt'].chat(message + '\nExplain everything in a maximum of 2 paragraphs.') | |
| history.append((message, bot)) | |
| chat_history = '' | |
| for i in history: | |
| chat_history += (f"USER: {i[0]}\n\nBOT: {i[1]}\n\n\n\n"); | |
| with open(CHAT_PATH + state['id'], 'w') as f: | |
| f.write(chat_history) | |
| return "", history, state | |
| msg.submit(cp_helper, [msg, interface, state], [msg, interface, state]) | |
| # def sub_rating(state, option): | |
| # if (state['id'] == None): | |
| # return | |
| # with open(CHAT_PATH + 'meta', 'a') as f: | |
| # f.write(f"{state['id']}: {option}\n") | |
| # return state, "Choose an Option" | |
| # rating = gr.Dropdown([ | |
| # "Choose an Option", | |
| # "1 - Very Unhelpful", | |
| # "2 - Unhelpful", | |
| # "3 - Neutral", | |
| # "4 - Helpful", | |
| # "5 - Very Helpful" | |
| # ], value = "Choose an Option", label='Rate the Conversation') | |
| # gr.Interface(sub_rating, [state, rating], [state, rating], allow_flagging=False) | |
| # def sub_feedback(state, message): | |
| # if (state['id'] == None): | |
| # return | |
| # with open(CHAT_PATH + state['id'], 'a') as f: | |
| # f.write(f"FEEDBACK: {message}\n\n\n\n") | |
| # with open(CHAT_PATH + 'feedback', 'a') as f: | |
| # f.write(f"FEEDBACK {state['id']}: {message}\n\n") | |
| # return state, "" | |
| # feedback = gr.Textbox(label = "Submit Feedback") | |
| # feedback.submit(sub_feedback, [state, feedback], [state, feedback]) | |
| demo.launch() |