Spaces:
Sleeping
Sleeping
File size: 3,687 Bytes
ed2bcfc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 |
import gradio as gr
from akinator.async_aki import Akinator
import akinator
import time
aki = Akinator()
### gradio state variables
count = gr.State(value=1)
to_verify = gr.State(value=False)
game_type = gr.State(value="en")
game_ended = gr.State(value=False)
def clear_state():
print("clear_state")
return 0, False
def change_game_type(drop_down_value):
return drop_down_value
def orchestrate_game_state(message, count, to_verify, game_ended):
time.sleep(1)
if message == "start":
count += 1
game_ended = False
elif (aki.progression <= 80) and (message == "back") and (not game_ended):
count -= 1
elif (aki.progression <= 80) and (not game_ended):
count += 1
elif to_verify and (message in ["yes", "no"]) and (not game_ended):
game_ended = True
elif (not game_ended):
count += 1
to_verify = True
print("orchestrate_game_state", message, count, to_verify, game_ended)
return count, to_verify, game_ended
async def fetch_response(message, history, drop_down_value, count, to_verify, game_type, game_ended):
print("fetch_response", message, drop_down_value, count, to_verify, game_type, game_ended)
if message == "start":
q = await aki.start_game(
language=game_type,
child_mode=True
)
q = f"{count}. " + q
elif (aki.progression <= 80) and (message == "back") and (not game_ended):
try:
q = await aki.back()
q = f"{count}. " + q
except akinator.CantGoBackAnyFurther:
pass
elif (aki.progression <= 80) and (not game_ended):
q = await aki.answer(message)
q = f"{count}. " + q
elif to_verify and message == "yes" and (not game_ended):
q = "Yay!!"
elif to_verify and message == "no" and (not game_ended):
q = "Oof.."
elif (not game_ended):
await aki.win()
q = f"It's {aki.first_guess['name']} ({aki.first_guess['description']})! Was I correct?\n{aki.first_guess['absolute_picture_path']}\n\t"
q = f"{count}. " + q
elif game_ended:
q = "The game has ended. To replay, enter `start` again."
return q
clear_button = gr.Button("Clear")
submit_button = gr.Button("Submit")
input_textbox = gr.Textbox(
placeholder=(
"To play, click [start] and submit. If you answer me genuinely, "
"and I will guess what is in your mind :)"
),
interactive=False,
container=False,
scale=2
)
drop_down_ls = gr.Dropdown(choices=["en", "en_animals", "en_objects"],
value="en",
label="Game Mode")
chat = gr.ChatInterface(
fn=fetch_response,
textbox=input_textbox,
examples=[["start"], ["yes"], ["no"], ["i don't know"], ["probably"], ["probably not"], ["back"]],
clear_btn=clear_button,
submit_btn=submit_button,
additional_inputs=[drop_down_ls, count, to_verify, game_type, game_ended]
)
with gr.Blocks(fill_height=True) as demo:
clear_button.click(fn=clear_state, inputs=None, outputs=[count, to_verify])
drop_down_ls.input(fn=change_game_type, inputs=[drop_down_ls], outputs=[game_type])
submit_button.click(fn=orchestrate_game_state,
inputs=[input_textbox, count, to_verify, game_ended],
outputs=[count, to_verify, game_ended],
trigger_mode="always_last")
chat.render()
if __name__ == "__main__":
demo.launch() |