import asyncio

import streamlit as st  # type: ignore

from codeinterpreterapi import File
from 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("[Github Repo](https://github.com/shroominic/codeinterpreter-api)")


input_openai_key = st.text_input("openai_key")
# 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,openai_key=input_openai_key))