File size: 977 Bytes
c5e39b8
 
 
 
 
 
 
 
 
 
 
 
7d6868c
c5e39b8
40658fc
842c0e1
 
 
40658fc
bcabe95
11f72a1
8f9d8b0
c5e39b8
4cd35b4
 
 
 
 
f905e84
c5e39b8
 
 
 
27c02bd
11f72a1
cc1fae6
c5e39b8
f905e84
c5e39b8
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
# -* coding:UTF-8 -*
# !/usr/bin/env python
import numpy as np
import gradio as gr

import os
from PIL import Image

from codeinterpreterapi import CodeInterpreterSession



def codeinterpreter(openai_key,prompt, files):

    fileList = []
    if files != None:
        for idx, file in enumerate(files):
            fileList.append(file)

    with CodeInterpreterSession(model="gpt-3.5-turbo",openai_api_key=openai_key) as session:
    #async with CodeInterpreterSession(model="gpt-3.5-turbo",openai_api_key="") as session:
        response = session.generate_response_sync(prompt, fileList,True)

        images = []

        for _file in response.files:
            images.append(_file.get_image())

        return [response.content,images]
        


app = gr.Interface(
    fn=codeinterpreter, inputs=[
        gr.Textbox(label="openai_key"),
        gr.Textbox(label="prompt",info="input the prompt"), 
        gr.Files(),
    ], outputs=["text","gallery"]
)
app.launch()