hoollyzhang commited on
Commit
2b9927f
·
1 Parent(s): 9f0c093

feat:Add application file

Browse files
Files changed (1) hide show
  1. app.py +28 -25
app.py CHANGED
@@ -1,35 +1,38 @@
1
- import gradio as gr
2
  import asyncio
 
 
 
3
  from codeinterpreterapi import File
4
  from frontend.utils import get_images
5
 
 
 
6
 
7
- def code_interpreter_app(input_text, uploaded_file=None):
8
- if uploaded_file is not None:
9
- bytes_data = uploaded_file.read()
10
- uploaded_file_data = File(name=uploaded_file.name, content=bytes_data)
11
- output = asyncio.run(get_images(
12
- input_text, files=[uploaded_file_data]))
13
- else:
14
- # Call get_images without files
15
- output = asyncio.run(get_images(input_text))
16
- return output
17
 
 
 
18
 
19
- # This will create a textbox where you can input text
20
- input_text = gr.Textbox("Write your prompt")
21
 
22
- # This will create a file uploader
23
- file_uploader = gr.File(label="Upload your file", optional=True)
24
 
25
 
26
- # Interface for the Gradio app
27
- interface = gr.Interface(
28
- fn=code_interpreter_app,
29
- inputs=[input_text, file_uploader],
30
- outputs=gr.Textbox("Output will be shown here"), # Dummy output
31
- title="Code Interpreter API 🚀",
32
- description="给勇哥哥打call 🚀",
33
- )
34
-
35
- interface.launch()
 
 
 
 
 
 
 
 
 
1
  import asyncio
2
+
3
+ import streamlit as st
4
+
5
  from codeinterpreterapi import File
6
  from frontend.utils import get_images
7
 
8
+ # Page configuration
9
+ st.set_page_config(layout="wide")
10
 
11
+ st.title('Code Interpreter API 🚀')
 
 
 
 
 
 
 
 
 
12
 
13
+ # This will create a sidebar
14
+ st.sidebar.title("Code Interpreter API 🚀")
15
 
16
+ st.sidebar.markdown("### 给勇哥哥打call 🚀")
 
17
 
18
+ st.sidebar.markdown(
19
+ "![Code Interpreter](https://vercel.brzhang.club/_next/image?url=%2Fcoffer.jpg&w=640&q=75)")
20
 
21
 
22
+ # This will create a textbox where you can input text
23
+ input_text = st.text_area("Write your prompt")
24
+ uploaded_files = st.file_uploader(
25
+ "Upload your files", accept_multiple_files=True)
26
+
27
+ uploaded_files_list = []
28
+ for uploaded_file in uploaded_files:
29
+ bytes_data = uploaded_file.read()
30
+ uploaded_files_list.append(File(name=uploaded_file.name,
31
+ content=bytes_data))
32
+
33
+ # This will create a button
34
+ button_pressed = st.button('Run code interpreter', use_container_width=True)
35
+
36
+ # This will display the images only when the button is pressed
37
+ if button_pressed and input_text != "":
38
+ asyncio.run(get_images(input_text, files=uploaded_files_list))