feat: OK

app.py

@@ -1,67 +1,46 @@
-import streamlit as st
 import asyncio
-import tempfile
-import os
-import shutil
-from codeinterpreterapi import File
-from frontend.utils import get_images
 
-# Page configuration
-st.set_page_config(layout="wide")
+import streamlit as st
 
-st.title('Code Interpreter API 🚀')
+from codeinterpreterapi import File
+from frontend.utils import get_images
 
-# This will create a sidebar
-st.sidebar.title("Code Interpreter API 🚀")
 
-st.sidebar.markdown("### 给勇哥哥打call 🚀")
+async def main():
 
-st.sidebar.markdown(
-    "![Code Interpreter](https://vercel.brzhang.club/_next/image?url=%2Fcoffer.jpg&w=640&q=75)")
+    # Page configuration
+    st.set_page_config(layout="wide")
 
-# 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)
+    st.title('Code Interpreter API 🚀')
 
-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 sidebar
+    st.sidebar.title("Code Interpreter API 🚀")
 
-# This will create a button
-button_pressed = st.button('Run code interpreter', use_container_width=True)
+    st.sidebar.markdown("### 给勇哥哥打call 🚀")
 
-# This will display the images only when the button is pressed
-if button_pressed and input_text != "":
-    # Call the get_images function using asyncio.run to handle the async behavior
-    output = asyncio.run(get_images(input_text, files=uploaded_files_list))
+    st.sidebar.markdown(
+        "![Code Interpreter](https://vercel.brzhang.club/_next/image?url=%2Fcoffer.jpg&w=640&q=75)")
 
-    # Display output and images
-    if output:
-        with st.spinner():
-            st.write(output.content)
+    # 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)
 
-            for _file in output.files:
-                st.image(_file.get_image(), caption=input_text,
-                         use_column_width=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))
 
-            # Allowing the download of the results
-            if len(output.files) == 1:
-                st.download_button('Download Results', output.files[0].content,
-                                   file_name=output.files[0].name,
-                                   use_container_width=True)
-            else:
-                target_path = tempfile.mkdtemp()
-                for _file in output.files:
-                    _file.save(os.path.join(target_path, _file.name))
+    # This will create a button
+    button_pressed = st.button(
+        'Run code interpreter', use_container_width=True)
 
-                zip_path = os.path.join(
-                    os.path.dirname(target_path), "archive")
-                shutil.make_archive(zip_path, 'zip', target_path)
+    # 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))
 
-                with open(zip_path + ".zip", 'rb') as f:
-                    st.download_button('Download Results', f,
-                                       file_name="archive.zip",
-                                       use_container_width=True)
+if __name__ == "__main__":
+    asyncio.run(main())