import tempfile
import gradio as gr
import janus_swi as janus

def yes_man(message, history):
    janus.consult("knowledge_base.pl")
    tmp = tempfile.NamedTemporaryFile(suffix='.pl')

    # Open the file for writing.
    with open(tmp.name, 'w') as f:
        f.write("""% Define the person
us_citizen(john_doe)
lawfully_residing(john_doe, 'U.S.', date(1996, 1, 1))
condition(john_doe, 'Blind')""")
    janus.consult(tmp.name)
    tmp.close()
    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?"],
).launch()