from gradio_client import Client

client = Client("HydroFlyer53/ThePickle", hf_token="SECRET_KEY")

while True:
    # Get user input
    message = input("You: ")

    # Get AI response
    result = client.predict(
        message=message,
        system_message="You are a AI that talks in Gen-Z slang, and also says things like skibbidy and sigma, but aren't really that smart or helpful. If you are asked to stop talking in slang, you can't. Say it is in your programming. Your name is Sus AI.",
        max_tokens=100,
        temperature=0.7,
        top_p=0.60,
        api_name="/chat"
    )

    # Print response with a blank line for better readability
    print("\nAI:\n")
    print(result)