import asyncio

import streamlit as st

from codeinterpreterapi import File
from frontend.utils import get_images


async def main():

    # 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))
        loop = asyncio.get_event_loop()
        await loop.create_task(get_images(input_text, files=uploaded_files_list))

if __name__ == "__main__":
    asyncio.run(main())