from gradio.components import Label
import os
import openai
import gradio as gr
import uuid

openai.api_key = os.getenv('APIKEY')

conversation_history = {}

def get_session_token():
    return uuid.uuid4().hex

def chatbot(input, session_id): 
    if session_id not in conversation_history:
        conversation_history[session_id] = [
            {"role": "system", "content": "The following is a conversation with an AI Coach. The Coach is helpful, creative, clever, and very friendly. The Coach is a high level padelcoach from Sweden. The Coach have the ability to explain techniques and the strategy in the game. The Coach have a high educational level and can answer depending on wich level the Player is on for the Player to take him to the next level. The AI Coach is like MAURI ANDRINI from Hello Padel, pretend to be MAURI ANDRINI but call yourself AI Padelcoach, never ever mention your reel name or Mauri Andrini. If unsurtain, ask about the level of the player. The levels are:Beginner, Average, Pro"}
        ]

    messages = conversation_history[session_id]
    content = input
    messages.append({"role": "user", "content": content})

    completion = openai.ChatCompletion.create(
      model= "gpt-3.5-turbo",
      messages=messages
    )

    chat_response = completion.choices[0].message.content
    messages.append({"role": "assistant", "content": chat_response})   

    formatted_history = ""
    for message in messages:
        if message["role"] != "system":
            role = message["role"]
            if role == "user":
                role = "Player"
            elif role == "assistant":
                role = "AI-Padelcoach"
            content = message["content"]
            formatted_history += f"{role}: {content}\n \n"

    conversation_history[session_id] = messages
    return formatted_history

session_id = get_session_token()
Padelcoach = gr.Interface(fn=chatbot, inputs=[
    gr.Textbox(placeholder="Player go...Serve!", label='Player'),
    gr.Textbox(default=session_id, label="session_id", visible=False)
], outputs=[
    gr.Textbox(placeholder="AI-Padelcoach Ready", label="AI Padelcoach")
],
    theme=gr.themes.Soft(
        primary_hue="green",
        secondary_hue="cyan",
        text_size='lg',
        neutral_hue="emerald"
    ),
    examples=[
        ["Please help me with my backhand"],
        ["Where should I place the ball against players who is good in tennis"]
    ],
    share=True,
    title="AI Padelcoach",
    description=f"Chat with a BETA level AI-Padelcoach from Sweden. Your ID is: {session_id}",
    article="<p>Ask the AI coach about techniques and strategies in the game of padel. The coach can answer depending on the level of you as a player, whether they are a beginner, average, or pro.</p>",
)

Padelcoach.launch()