frontend/utils.py

import asyncio

import streamlit as st

from codeinterpreterapi import File
from frontend.utils import get_images

# Page configuration
st.set_page_config(layout="wide")

st.title('Code Interpreter API 🚀')

# This will create a sidebar
st.sidebar.title("Code Interpreter API 🚀")

st.sidebar.markdown("### 给勇哥哥打call 🚀")

st.sidebar.markdown(
    "![Code Interpreter](https://vercel.brzhang.club/_next/image?url=%2Fcoffer.jpg&w=640&q=75)")


# This will create a textbox where you can input text
input_text = st.text_area("Write your prompt")
uploaded_files = st.file_uploader(
    "Upload your files", accept_multiple_files=True)

uploaded_files_list = []
for uploaded_file in uploaded_files:
    bytes_data = uploaded_file.read()
    uploaded_files_list.append(File(name=uploaded_file.name,
                                    content=bytes_data))

# This will create a button
button_pressed = st.button('Run code interpreter', use_container_width=True)

# This will display the images only when the button is pressed
if button_pressed and input_text != "":
    asyncio.run(get_images(input_text, files=uploaded_files_list))