pet.py

from util import WeightedRandomMap, ollama_query, speak
from tkinter import simpledialog
import tkinter as tk
from os.path import join


def read_frames(impath):
    output = []
    i = 0
    while True:
        try:
            new_frame = tk.PhotoImage(file=join(impath), format=f'gif -index {i}')
            output.append(new_frame)
        except:
            break
        i += 1
    return output


class PetState:
    def __init__(self, json_obj, impath):
        self.name = json_obj['state_name']
        self.frames = read_frames(join(impath, json_obj['file_name']))
        self.ox, self.oy, self.w, self.h = json_obj['dims']
        if 'move' in json_obj:
            self.dx, self.dy = json_obj['move']
        else:
            self.dx, self.dy = 0, 0
        self.next_states = WeightedRandomMap(json_obj['transitions_to'])


class Pet:
    def __init__(self, states, window):
        self.states = states
        self.window = window
        self.current_state = list(states.values())[0]
        self.__current_frame = 0
        self.x, self.y = 45, 800

    def next_frame(self):
        output = self.current_state.frames[self.__current_frame]
        self.__current_frame += 1
        if self.__current_frame == len(self.current_state.frames):
            self.__state_change()
        self.x, self.y = (
                self.x + self.current_state.dx), (self.y + self.current_state.dy)
        return output

    def __state_change(self):
        self.set_state(self.current_state.next_states.get_rand())

    def set_state(self, name: str):
        self.current_state = self.states[name]
        self.__current_frame = 0

    def start_chat(self, system_message: str, listen_state: str, response_state: str, end_state: str):
        self.set_state(listen_state)
        user_message = simpledialog.askstring("Ollama Input", "What do you want to ask?", parent=self.window)
        response = ollama_query(system_message, user_message)
        self.set_state(response_state)
        speak(response, lambda: self.set_state(end_state))