feat: 默认打印

app.py

@@ -1,7 +1,8 @@
-import asyncio
-
 import streamlit as st
-
+import asyncio
+import tempfile
+import os
+import shutil
 from codeinterpreterapi import File
 from frontend.utils import get_images
 
@@ -18,7 +19,6 @@ 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(
@@ -35,4 +35,33 @@ 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))
+    # Call the get_images function directly without asyncio.run
+    output = get_images(input_text, files=uploaded_files_list)
+
+    # Display output and images
+    if output:
+        with st.spinner():
+            st.write(output.content)
+
+            for _file in output.files:
+                st.image(_file.get_image(), caption=input_text,
+                         use_column_width=True)
+
+            # 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))
+
+                zip_path = os.path.join(
+                    os.path.dirname(target_path), "archive")
+                shutil.make_archive(zip_path, 'zip', target_path)
+
+                with open(zip_path + ".zip", 'rb') as f:
+                    st.download_button('Download Results', f,
+                                       file_name="archive.zip",
+                                       use_container_width=True)