import gradio as gr
import openai, config
openai.api_key= os.environ.get("openai.api_key")

messages = [{"role": "system", "content": 'You are Python ML developer. reply only in python code'}]

def transcribe(text):
    
    messages.append({"role": "user", "content": text})

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

    system_message = response["choices"][0]["message"]
    messages.append(system_message)

    

    chat_transcript = ""
    for message in messages:
        if message['role'] != 'system':
            chat_transcript += message['role'] + ": " + message['content'] + "\n\n"

    return chat_transcript

ui = gr.Interface(fn=transcribe, inputs="text", outputs="text").launch()