import openai
import gradio
import os
import re

openai.api_key = os.environ.get("API_TOKEN")

messages = [{"role": "system", "content": "You are an education expert who can correct essays. You are bilingual in English and Japanese"}]

MAX_TOKENS = 4096
MAX_HISTORY = 1

def CustomChatGPT(user_input):
    global messages
    essay_keywords = ["essay", "エッセイ", "論文"]
    action_keywords = ["write", "make", "create", "生成", "作成", "書く"]

    if any(re.search(f"{action_kw}.*{essay_kw}", user_input.lower()) for action_kw in action_keywords for essay_kw in essay_keywords):
        return "I'm sorry, I cannot write an essay for you."

    # Clear the messages list before adding new messages
    messages = [{"role": "system", "content": "You are an education expert who can correct English writing. You are bilingual in English and Japanese"}]

    user_message = {"role": "user", "content": f"Step 1, find errors in sentences' logic. Focus mainly on how well the sentences are connected to each other and proper use of discourse markers. You do not need to worry about grammatical mistakes. You do not have to correct the sentences either. Step 3, Explain in detail all the errors from what I originally wrote. Put each explanation in a bullet point. You do not need to edit the sentences, just explain everything wrong with it: [{user_input}]"}
    
    messages.append(user_message)
    
    while True:
        response = openai.ChatCompletion.create(
            model="gpt-3.5-turbo",
            messages=messages
        )

        total_tokens = response['usage']['total_tokens']
        if total_tokens < MAX_TOKENS:
            break

    ChatGPT_reply = response["choices"][0]["message"]["content"]
    messages.append({"role": "assistant", "content": ChatGPT_reply})

    return ChatGPT_reply

# Add text instructions on top of the input and output boxes
input_text = "ここに訂正してほしい英語の作文を置いてください。そして「Submit」を押してください:"
output_text = "提案される変更点はこちらです:"
instructions = "このアプリケーションは、あなたのエッセイの論理と推論をチェックするために使用できます。エッセイを修正するのではなく、あなたの論理と推論にある誤りを指摘します。文法的な提案は提供しません。文法的な訂正については、私の他のアプリをご覧ください。アプリは、1つのパラグラフずつ入力する場合に最適に機能します。例えば、3つのパラグラフから成る作文をチェックしたい場合は、それぞれのパラグラフを「Submit」してください。つまり、プログラムを3回実行し、各パラグラフごとに1回ずつ実行してください。"

# Modify the Gradio interface to include the text instructions and image
demo = gradio.Interface(fn=CustomChatGPT, inputs=gradio.inputs.Textbox(lines=5, label=input_text), outputs=gradio.outputs.Textbox(label=output_text), title="Teacher Jihan's Logic Checker", description=instructions)

demo.launch(share=False, debug=True)