import gradio as gr
from chatbot_utils import process_user_input, chatbot_response

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() == "종료":
            new_history = chatbot_history + [("종료", "실험에 참가해 주셔서 감사합니다. 후속 지시를 따라주세요")]
            return new_history, gr.update(choices=[], interactive=False)
        
        # Get victim's response first
        victim_response, victim_choices = chatbot_response(input_text, 'victim', n=3)
        
        # Then get offender's response
        offender_response, _ = chatbot_response(input_text, 'offender', n=1)
        
        new_history = chatbot_history + [(input_text, victim_response), (None, offender_response)]
        return new_history, gr.update(choices=victim_choices)

    def handle_case_selection():
        initial_message = "발표가 망한 건 제 잘못도 좀 있지만, 팀장님은 아무것도 안 하면서 이러는 건 선 넘은거죠"
        chatbot_history = [(initial_message, None)]
        
        # Get victim's response first
        victim_response, victim_choices = chatbot_response(initial_message, 'victim', n=3)
        chatbot_history.append((None, victim_response))
        
        # Then get offender's response
        offender_response, _ = chatbot_response(initial_message, 'offender', n=1)
        chatbot_history.append((None, offender_response))
        
        return 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