app.py

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()