require chrome driver

.history/app_20240731171937.py

@@ -0,0 +1,218 @@
+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
+
+
+	## 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
+	Please provide written feedback, maybe there was an issue with the bot or you really liked some response. You can give written feedback while chatting with the bot; it is recorded in real time with the chat history. Use the second text box for this.
+	Please also provide feedback on the bot's helpfulness. Use the first dropdown (after you completed your chat) to rate how helpful the bot was. Consider how the bot's responses helped your understand faster and more effectively.
+	
+	Thanks for using this bot, and I hope you found it helpful!
+				   
+	This is an open source project maintained by [Rithwik Gupta](github.com/Rithwik-G) (beaboss on codeforces). If you would like to contribute (by adding scrapers for other OJs, features, or prompts), you can make a pull request on HuggingFace community tab. If you want to collaborate, feel free to DM me on discord @Rithwik#1821 or at rithwikca2020@gmail.com.
+	If you found this bot useful, .
+	""")
+
+	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=1000
+
+		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()

.history/app_20240731171947.py

@@ -0,0 +1,218 @@
+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
+
+
+	## 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
+	Please provide written feedback, maybe there was an issue with the bot or you really liked some response. You can give written feedback while chatting with the bot; it is recorded in real time with the chat history. Use the second text box for this.
+	Please also provide feedback on the bot's helpfulness. Use the first dropdown (after you completed your chat) to rate how helpful the bot was. Consider how the bot's responses helped your understand faster and more effectively.
+	
+	Thanks for using this bot, and I hope you found it helpful!
+				   
+	This is an open source project maintained by [Rithwik Gupta](github.com/Rithwik-G) (beaboss on codeforces). If you would like to contribute (by adding scrapers for other OJs, features, or prompts), you can make a pull request on HuggingFace community tab. If you want to collaborate, feel free to DM me on discord @Rithwik#1821 or at rithwikca2020@gmail.com.
+	If you found this bot useful, please star the GitHub repo.
+	""")
+
+	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=1000
+
+		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()

.history/app_20240731171957.py

@@ -0,0 +1,218 @@
+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
+
+
+	## 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
+	Please provide written feedback, maybe there was an issue with the bot or you really liked some response. You can give written feedback while chatting with the bot; it is recorded in real time with the chat history. Use the second text box for this.
+	Please also provide feedback on the bot's helpfulness. Use the first dropdown (after you completed your chat) to rate how helpful the bot was. Consider how the bot's responses helped your understand faster and more effectively.
+	
+	Thanks for using this bot, and I hope you found it helpful!
+				   
+	This is an open source project maintained by [Rithwik Gupta](github.com/Rithwik-G) (beaboss on codeforces). If you would like to contribute (by adding scrapers for other OJs, features, or prompts), you can make a pull request on HuggingFace community tab. If you want to collaborate, feel free to DM me on discord @Rithwik#1821 or at rithwikca2020@gmail.com.
+	If you found this bot useful, please star the GitHub repo.
+	""")
+
+	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=1000
+
+		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()

.history/app_20240731172000.py

@@ -0,0 +1,218 @@
+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
+
+
+	## 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
+	Please provide written feedback, maybe there was an issue with the bot or you really liked some response. You can give written feedback while chatting with the bot; it is recorded in real time with the chat history. Use the second text box for this.
+	Please also provide feedback on the bot's helpfulness. Use the first dropdown (after you completed your chat) to rate how helpful the bot was. Consider how the bot's responses helped your understand faster and more effectively.
+	
+	Thanks for using this bot, and I hope you found it helpful!
+				   
+	This is an open source project maintained by [Rithwik Gupta](github.com/Rithwik-G) (beaboss on codeforces). If you would like to contribute (by adding scrapers for other OJs, features, or prompts), you can make a pull request on HuggingFace community tab. If you want to collaborate, feel free to DM me on discord @Rithwik#1821 or at rithwikca2020@gmail.com.
+	If you found this bot useful, please star the GitHub repo at.
+	""")
+
+	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=1000
+
+		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()

.history/app_20240731172129.py

@@ -0,0 +1,218 @@
+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
+
+
+	## 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
+	Please provide written feedback, maybe there was an issue with the bot or you really liked some response. You can give written feedback while chatting with the bot; it is recorded in real time with the chat history. Use the second text box for this.
+	Please also provide feedback on the bot's helpfulness. Use the first dropdown (after you completed your chat) to rate how helpful the bot was. Consider how the bot's responses helped your understand faster and more effectively.
+	
+	Thanks for using this bot, and I hope you found it helpful!
+				   
+	This is an open source project maintained by [Rithwik Gupta](github.com/Rithwik-G) (beaboss on codeforces). If you would like to contribute (by adding scrapers for other OJs, features, prompts), you can make a pull request on HuggingFace community tab. If you want to collaborate, feel free to DM me on discord @Rithwik#1821 or at rithwikca2020@gmail.com.
+	If you found this bot useful, please star the GitHub repo at.
+	""")
+
+	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=1000
+
+		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()

.history/app_20240731172136.py

@@ -0,0 +1,218 @@
+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
+
+
+	## 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
+	Please provide written feedback, maybe there was an issue with the bot or you really liked some response. You can give written feedback while chatting with the bot; it is recorded in real time with the chat history. Use the second text box for this.
+	Please also provide feedback on the bot's helpfulness. Use the first dropdown (after you completed your chat) to rate how helpful the bot was. Consider how the bot's responses helped your understand faster and more effectively.
+	
+	Thanks for using this bot, and I hope you found it helpful!
+				   
+	This is an open source project maintained by [Rithwik Gupta](github.com/Rithwik-G) (beaboss on codeforces). If you would like to contribute (by adding scrapers for other OJs, features, adjusting prompts), you can make a pull request on HuggingFace community tab. If you want to collaborate, feel free to DM me on discord @Rithwik#1821 or at rithwikca2020@gmail.com.
+	If you found this bot useful, please star the GitHub repo at.
+	""")
+
+	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=1000
+
+		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()

.history/app_20240731172140.py

@@ -0,0 +1,218 @@
+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
+
+
+	## 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
+	Please provide written feedback, maybe there was an issue with the bot or you really liked some response. You can give written feedback while chatting with the bot; it is recorded in real time with the chat history. Use the second text box for this.
+	Please also provide feedback on the bot's helpfulness. Use the first dropdown (after you completed your chat) to rate how helpful the bot was. Consider how the bot's responses helped your understand faster and more effectively.
+	
+	Thanks for using this bot, and I hope you found it helpful!
+				   
+	This is an open source project maintained by [Rithwik Gupta](github.com/Rithwik-G) (beaboss on codeforces). If you would like to contribute (by adding scrapers for other OJs, adding features, adjusting prompts), you can make a pull request on HuggingFace community tab. If you want to collaborate, feel free to DM me on discord @Rithwik#1821 or at rithwikca2020@gmail.com.
+	If you found this bot useful, please star the GitHub repo at.
+	""")
+
+	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=1000
+
+		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()

.history/app_20240731172151.py

@@ -0,0 +1,218 @@
+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
+
+
+	## 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
+	Please provide written feedback, maybe there was an issue with the bot or you really liked some response. You can give written feedback while chatting with the bot; it is recorded in real time with the chat history. Use the second text box for this.
+	Please also provide feedback on the bot's helpfulness. Use the first dropdown (after you completed your chat) to rate how helpful the bot was. Consider how the bot's responses helped your understand faster and more effectively.
+	
+	Thanks for using this bot, and I hope you found it helpful!
+				   
+	This is an open source project maintained by [Rithwik Gupta](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. If you want to collaborate, feel free to DM me on discord @Rithwik#1821 or at rithwikca2020@gmail.com.
+	If you found this bot useful, please star the GitHub repo at.
+	""")
+
+	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=1000
+
+		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()

.history/packages_20240731172954.txt

@@ -0,0 +1 @@
+chromium-driver

.history/utils/chatbot_20240731172749.py

@@ -0,0 +1,44 @@
+import openai
+from openai import OpenAI
+import os
+# openai.api_key = os.environ['OPENAI_KEY']
+
+class Chat:
+
+	def __init__(self, system_message, lim=0.005):
+		self.messages = [
+			{"role": "system", "content" : system_message}
+		]
+		self.left = lim
+		self.client = OpenAI(api_key=API_KEY)
+
+	def chat(self, query):
+		if (self.left <= 0):
+			return "You have reached the chat limit."
+		if (len(query) > 20000):
+			return "Chat input is too long"
+
+		self.messages.append({"role" : "user", "content" : query})
+		
+		# print(self.messages)
+
+		response = self.client.chat.completions.create(
+		  model="gpt-4o-mini",
+		  messages=self.messages,
+		  temperature=0.3
+		)
+		# del self.messages[-1]
+		result = response.choices[0].message.content
+
+		self.messages.append(response.choices[0].message)
+		# print(response)
+		# print([result])
+
+		result = result.replace('\\(', '$')
+		result = result.replace('\\)', '$')
+		# print([result])
+
+		self.left -= 0.6 * response.usage.completion_tokens / (10**6) + 0.15 * response.usage.prompt_tokens /(10**6)
+		print(self.left)
+
+		return (result)

.history/utils/chatbot_20240731173008.py

@@ -0,0 +1,44 @@
+import openai
+from openai import OpenAI
+import os
+openai.api_key = os.environ['OPENAI_KEY']
+
+class Chat:
+
+	def __init__(self, system_message, lim=0.005):
+		self.messages = [
+			{"role": "system", "content" : system_message}
+		]
+		self.left = lim
+		self.client = OpenAI(api_key=API_KEY)
+
+	def chat(self, query):
+		if (self.left <= 0):
+			return "You have reached the chat limit."
+		if (len(query) > 20000):
+			return "Chat input is too long"
+
+		self.messages.append({"role" : "user", "content" : query})
+		
+		# print(self.messages)
+
+		response = self.client.chat.completions.create(
+		  model="gpt-4o-mini",
+		  messages=self.messages,
+		  temperature=0.3
+		)
+		# del self.messages[-1]
+		result = response.choices[0].message.content
+
+		self.messages.append(response.choices[0].message)
+		# print(response)
+		# print([result])
+
+		result = result.replace('\\(', '$')
+		result = result.replace('\\)', '$')
+		# print([result])
+
+		self.left -= 0.6 * response.usage.completion_tokens / (10**6) + 0.15 * response.usage.prompt_tokens /(10**6)
+		print(self.left)
+
+		return (result)

app.py

@@ -43,8 +43,8 @@ with gr.Blocks(title = "CP Helper Bot") as demo:
 	
 	Thanks for using this bot, and I hope you found it helpful!
 				   
-	This is an open source project maintained by [Rithwik Gupta](github.com/Rithwik-G) (beaboss on codeforces). If you would like to contribute (by adding scrapers for other OJs, features, or prompts), you can make a pull request on HuggingFace community tab. If you want to collaborate, feel free to DM me on discord @Rithwik#1821 or at rithwikca2020@gmail.com.
-	If you found this bot useful, please consider  .
+	This is an open source project maintained by [Rithwik Gupta](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. If you want to collaborate, feel free to DM me on discord @Rithwik#1821 or at rithwikca2020@gmail.com.
+	If you found this bot useful, please star the GitHub repo at.
 	""")
 
 	interface = gr.Chatbot(height='300px', placeholder="Provide a link for the problem you're solving.", 

packages.txt

@@ -0,0 +1 @@
+chromium-driver

saved_chats/#29622

@@ -0,0 +1,6 @@
+USER: https://codeforces.com/problemset/problem/1997/D
+
+BOT: Thanks! Now could you provide a link to the editorial.
+
+
+

utils/scrapers/prescraped/codeforces/Problems/1997D

@@ -0,0 +1 @@
+You are given a rooted tree, consisting of n vertices. The vertices in the tree are numbered from 1 to n, and the root is the vertex 1. The value ai is written at the i-th vertex.You can perform the following operation any number of times (possibly zero): choose a vertex v which has at least one child; increase av by 1; and decrease au by 1 for all vertices u that are in the subtree of v (except v itself). However, after each operation, the values on all vertices should be non-negative.Your task is to calculate the maximum possible value written at the root using the aforementioned operation.