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.Completion.create(engine="davinci-codex", prompt=prompt, max_tokens=2048, n=1,stop=None,temperature=0.5)

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

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

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