import gradio as gr
from pyswip import Prolog
prolog = Prolog()
prolog.consult("knowledge_base.pl")

def yes_man(message, history):
    # % Define the person
    prolog.assertz("us_citizen(john_doe)")
    prolog.assertz("lawfully_residing(john_doe, 'U.S.', date(1996, 1, 1))")
    prolog.assertz("condition(john_doe, 'Blind')")
    if message.endswith("?"):
        return prolog.query_once("eligible_for_ssi(john_doe)")
    else:
        return "Ask me about `eligible_for_ssi`!"

gr.ChatInterface(
    yes_man,
    title="Yes Man",
    description="Ask Yes Man any question",
    examples=["Hello", "Am I cool?", "Are tomatoes vegetables?"],
    cache_examples=True,
    retry_btn=None,
).launch()