feat:Add application file

README.md

@@ -4,7 +4,7 @@ emoji: 🦀
 colorFrom: red
 colorTo: red
 sdk: streamlit
-sdk_version: 1.21.0
+sdk_version: 1.24.1
 app_file: app.py
 pinned: false
 ---

app.py

@@ -1,38 +1,35 @@
+import gradio as gr
 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 🚀')
+def code_interpreter_app(input_text, uploaded_file=None):
+    if uploaded_file is not None:
+        bytes_data = uploaded_file.read()
+        uploaded_file_data = File(name=uploaded_file.name, content=bytes_data)
+        output = asyncio.run(get_images(
+            input_text, files=[uploaded_file_data]))
+    else:
+        # Call get_images without files
+        output = asyncio.run(get_images(input_text))
+    return output
 
-# This will create a sidebar
-st.sidebar.title("Code Interpreter API 🚀")
 
-st.sidebar.markdown("### 给勇哥哥打call 🚀")
+# This will create a textbox where you can input text
+input_text = gr.Textbox("Write your prompt")
 
-st.sidebar.markdown(
-    "![Code Interpreter](https://vercel.brzhang.club/_next/image?url=%2Fcoffer.jpg&w=640&q=75)")
+# This will create a file uploader
+file_uploader = gr.File(label="Upload your file", optional=True)
 
 
-# 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))
+# Interface for the Gradio app
+interface = gr.Interface(
+    fn=code_interpreter_app,
+    inputs=[input_text, file_uploader],
+    outputs=gr.Textbox("Output will be shown here"),  # Dummy output
+    title="Code Interpreter API 🚀",
+    description="给勇哥哥打call 🚀",
+)
+
+interface.launch()

flagged/log.csv

@@ -0,0 +1,7 @@
+prompt,file_path,output,flag,username,timestamp
+范围范文芳,,,,,2023-07-23 13:29:43.611460
+范围范文芳,,,,,2023-07-23 13:29:45.483099
+nihao ,,,,,,Output will be shown here,,,2023-07-23 13:32:27.341759
+,,,,,2023-07-23 13:34:12.231896
+1+1的结果,,Output will be shown here,,,2023-07-23 13:35:04.771405
+hahah,,Please upload a file.,,,2023-07-23 13:36:13.694983

frontend/utils.py

@@ -1,51 +1,38 @@
-from codeinterpreterapi import CodeInterpreterSession
+import asyncio
+
 import streamlit as st
-import tempfile
-import os
-import shutil
-
-
-def create_temp_folder() -> str:
-    """
-    Creates a temp folder
-    """
-    temp_folder = tempfile.mkdtemp()
-    return temp_folder
-
-
-async def get_images(prompt: str, files: list = None):
-    if files is None:
-        files = []
-    with st.chat_message("user"):
-        st.write(prompt)
-    with st.spinner():
-        async with CodeInterpreterSession(model='gpt-3.5-turbo') as session:
-            response = await session.generate_response(
-                prompt,
-                files=files
-            )
-
-            with st.chat_message("assistant"):
-                st.write(response.content)
-
-                # Showing Results
-                for _file in response.files:
-                    st.image(_file.get_image(), caption=prompt, use_column_width=True)
-
-                # Allowing the download of the results
-                if len(response.files) == 1:
-                    st.download_button('Download Results', response.files[0].content,
-                                       file_name=response.files[0].name,
-                                       use_container_width=True)
-                else:
-                    target_path = tempfile.mkdtemp()
-                    for _file in response.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)
+
+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))