import gradio as gr
from chatbot_utils import process_user_input, chatbot_response
import time

def create_interface():
    def handle_user_response(user_input, selected_response, chatbot_history):
        input_text = user_input if user_input else selected_response
        
        if input_text.strip().lower() == "종료":
            chatbot_history.append((input_text, "실험에 참가해 주셔서 감사합니다. 후속 지시를 따라주세요"))
            return chatbot_history, gr.update(choices=[], interactive=False)
        
        # Add user's input to history
        chatbot_history.append((input_text, None))
        yield chatbot_history, gr.update(choices=[])  # Immediately show user input
        
        # Get offender's response
        offender_response, _ = chatbot_response(input_text, 'offender', n=1)
        time.sleep(1)  # 2-second delay
        
        # Add offender's response to history
        chatbot_history.append((None, offender_response))
        
        # Generate victim choices for the next turn
        _, victim_choices = chatbot_response(offender_response, 'victim', n=3)
        
        yield chatbot_history, gr.update(choices=victim_choices)

    def handle_case_selection():
        initial_message = "아니, 그럼 어떻게 해야 되는데?"
        chatbot_history = [(initial_message, None)]
        yield chatbot_history, gr.update(choices=[])
        
        offender_response, _ = chatbot_response(initial_message, 'offender', n=1)
        time.sleep(1)  # 2-second delay
        
        chatbot_history.append((None, offender_response))
        _, victim_choices = chatbot_response(offender_response, 'victim', n=3)
        
        yield chatbot_history, gr.update(choices=victim_choices)

    with gr.Blocks() as demo:
        case_selection_button = gr.Button("아니, 그럼 어떻게 해야 되는데?")
        screen = gr.Chatbot()
        user_input = gr.Textbox(label="이곳에 대답을 입력하세요")
        response_choices = gr.Dropdown(label="또는 대답을 선택해 주세요", choices=[], interactive=True)
        submit_button = gr.Button(value="제출")

        case_selection_button.click(handle_case_selection, inputs=[], outputs=[screen, response_choices])
        
        submit_button.click(handle_user_response, inputs=[user_input, response_choices, screen], outputs=[screen, response_choices])

    return demo