First release of frontend files compatibility

codeinterpreterapi/schema/file.py

@@ -6,6 +6,10 @@ class File(BaseModel):
     name: str
     content: bytes
 
+    @classmethod
+    def from_bytes(cls, name: str, content: bytes):
+        return cls(name=name, content=content)
+
     @classmethod
     def from_path(cls, path: str):
         with open(path, "rb") as f:

frontend/demo.py

@@ -2,6 +2,7 @@ import asyncio
 
 import streamlit as st
 
+from codeinterpreterapi import File
 from frontend.utils import get_images
 
 st.title('Code Interpreter API 🚀')
@@ -13,10 +14,18 @@ st.sidebar.markdown("[Github Repo](https://github.com/shroominic/codeinterpreter
 
 # This will create a textbox where you can input text
 input_text = st.text_input("Write your prompt")
+uploaded_files = st.file_uploader("Choose a CSV file", accept_multiple_files=True,
+                                  type=".csv")
+
+uploaded_files_list = []
+for uploaded_file in uploaded_files:
+    bytes_data = uploaded_file.read()
+    uploaded_files_list.append(File.from_bytes(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 image only when the button is pressed
 if button_pressed and input_text != "":
-    asyncio.run(get_images(input_text))
+    asyncio.run(get_images(input_text, files=uploaded_files_list))

frontend/utils.py

@@ -2,13 +2,16 @@ from codeinterpreterapi import CodeInterpreterSession
 import streamlit as st
 
 
-async def get_images(prompt: str):
+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
+                prompt,
+                files=files
             )
 
             with st.chat_message("assistant"):