UI2Code / chatAPI.py
3v324v23's picture
Initial commit with latest files
d5d3bd0
raw
history blame contribute delete
809 Bytes
import httpx
from openai import OpenAI
proxy_url='http://127.0.0.1:10809'
models={'gpt4':'gpt-4-turbo','gpt4o-mini':'gpt-4o-mini'}
def chat(prompt):
client = OpenAI(http_client=httpx.Client(proxy=proxy_url))
completion = client.chat.completions.create(
model=models['gpt4'],
messages=prompt
)
return completion.choices[0].message.content
def process_input(message:str,base64_image=None,history=[]):
if base64_image:
new_conversation= history+[{
'role':'user',
'content':[
{'type':'text','text':message},
{'type':'image_url',"image_url": {
"url": f"data:image/jpeg;base64,{base64_image}",
"detail": "high"
}}
]
}]
return new_conversation