Spaces:
Sleeping
Sleeping
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 |