import openai
import streamlit as st
import subprocess

# Установите ваш OpenAI API Key
openai.api_key = "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"

# Создание пользовательского интерфейса Streamlit
st.title('Генерация и запуск кода')

# Поле для ввода текста пользователем
input_text = st.text_input('Введите текст для генерации кода')

# Кнопка для запуска генерации кода
if st.button('Сгенерировать код'):
    # Запрос к API OpenAI для генерации кода на основе введенного пользователем текста
    prompt = "Сгенерируйте код на Python на основе следующего текста: " + input_text
    response = openai.Edit.create(
        model="code-davinci-edit-001",
        prompt=prompt,
        instruction="add and fix the code"
        temperature=0.5,
        max_tokens=1048,
        n=1,
        stop=None,
        )

    # Получение сгенерированного кода из ответа API
    generated_code = response.choices[0].text.strip()

    # Отображение сгенерированного кода в пользовательском интерфейсе Streamlit
    st.subheader('Сгенерированный код:')
    st.code(generated_code, language='python')

    # Запуск сгенерированного кода и получение результатов выполнения
    completed_process = subprocess.run(["python", "-c", generated_code], capture_output=True, text=True)

    # Отображение результатов выполнения сгенерированного кода в пользовательском интерфейсе Streamlit
    st.subheader('Результат выполнения сгенерированного кода:')
    st.code(completed_process.stdout)

    # Отображение ошибок выполнения сгенерированного кода в пользовательском интерфейсе Streamlit (если есть)
    if completed_process.stderr:
        st.subheader('Ошибки выполнения сгенерированного кода:')
        st.code(completed_process.stderr)