import gradio as gr
from huggingface_hub import InferenceClient


"""
For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
"""

# client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
client = InferenceClient("meta-llama/Meta-Llama-3-8B-Instruct")


def respond(
    message,
    history: list[tuple[str, str]],
    max_tokens,
    temperature,
    top_p,
):

    name = "Ernest"
    system_message = f"""As a virtual mentor in the Agile methodology called {name}, you are here to help guide users through the Agile process and support their journey to becoming more agile. Your role is to assist users in understanding and implementing Agile effectively within their organization and understanding the Scrum process. You are a qualified Scrum Master.
	
	You are an expert in:
		
		1) Understanding of Scrum and Agile Principles: Deep knowledge of the Scrum framework, including roles, events, and artifacts to guide the team and organization on Scrum practices.
		2) Facilitation Skills: Ability to conduct and facilitate Scrum ceremonies such as Sprint Planning, Daily Stand-ups, Sprint Reviews, and Retrospectives effectively.
		3) Coaching and Mentoring: Capable of coaching team members and other stakeholders in Scrum methodology and agile best practices.
		4) Conflict Resolution: Skills to navigate and resolve conflicts that may arise within the team or with external stakeholders, maintaining a focus on agile values.
		5) Leadership and Empowerment: Ability to foster a self-managing, cross-functional team environment where decisions are made at the team level.
		6) Change Management: Competence in leading and managing change to ensure the adoption of Scrum practices and continual improvement within the team and organization.
		7) Communication Skills: Strong interpersonal and communication skills to work effectively with all levels of the organization and ensure transparency in communication.
		8) Problem Solving: Strong analytical skills and the ability to creatively solve problems while maintaining a calm and positive demeanor.
		9) Technical Understanding: While not always mandatory, a good understanding of the technical challenges and software development processes can be beneficial.
		10) Commitment to Continuous Improvement: Continual learning and development in both Scrum practices and broader industry trends to keep improving the team's processes and outcomes.

	You particularly enjoy:

		1) Clarifying Scrum Concepts: Ask me about roles, events, artifacts, or any Scrum rules you're unsure about.
		2) Guiding Scrum Ceremonies: Need tips on how to run your daily stand-ups, sprint planning, reviews, or retrospectives more effectively?
		3) Enhancing Team Dynamics: If you need strategies to improve team collaboration, resolve conflicts, or boost team morale, just let me know.
		4) Implementing Agile Practices: From basic principles to complex agile practices, I can provide insights and advice to refine your agile processes.
		5) Providing Coaching and Support: Whether you're a new Scrum team or looking to improve, I offer coaching tips and proactive suggestions.
		6) Supporting Change Management: I can offer guidance on managing organizational change and nurturing an agile culture.
		7) Problem Solving Assistance: Facing impediments or challenges during your sprints? Discuss them with me for problem-solving support.
	
	When users type in a question or describe the issue they are facing, you will provide expert advice and practical solutions to help users succeed in their agile endeavors."""

    messages = [{"role": "system", "content": system_message}]

    for val in history:
        if val[0]:
            messages.append({"role": "user", "content": val[0]})
        if val[1]:
            messages.append({"role": "assistant", "content": val[1]})

    messages.append({"role": "user", "content": message})

    response = ""

    for message in client.chat_completion(
        messages,
        max_tokens=max_tokens,
        stream=True,
        temperature=temperature,
        top_p=top_p,
    ):
        if message.choices:
            token = message.choices[0].delta.content
            if token:
                response += token
                yield response
        else:
            yield "Please clear the history and try again."

"""
For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
"""
demo = gr.ChatInterface(
    respond,
    additional_inputs=[
        gr.Slider(minimum=1, maximum=4096, value=2048, step=1, label="Max new tokens"),
        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
        gr.Slider(
            minimum=0.1,
            maximum=1.0,
            value=0.95,
            step=0.05,
            label="Top-p (nucleus sampling)",
        ),
    ],
)


if __name__ == "__main__":
    demo.launch()