COLLEAGUE-AI / app.py
MrAli813's picture
Create app.py
78e2d06
raw
history blame
1.75 kB
import openai
import gradio as gr
openai.api_key = "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"
def predict(message, history):
history_openai_format = []
for human, assistant in history:
history_openai_format.append({"role": "user", "content": human })
history_openai_format.append({"role": "assistant", "content":assistant})
history_openai_format.append({"role": "user", "content": message})
response = openai.ChatCompletion.create(
model='gpt-3.5-turbo',
messages= history_openai_format,
temperature=1.0,
stream=True
)
partial_message = ""
for chunk in response:
if len(chunk['choices'][0]['delta']) != 0:
partial_message = partial_message + chunk['choices'][0]['delta']['content']
yield partial_message
A1 = gr.ChatInterface(predict,
title="PeachTalk+",
description="An AI Powered Chatbot with Computer Vision and Image Generation Capabilities Currently Under Development By Peach State Innovation and Technology. Ask Me About Question About Anything...From Georgia and Beyond...And I'll Give You An Answer!",
theme= gr.themes.Glass(primary_hue="amber", neutral_hue="lime"),
retry_btn=None,
clear_btn="Clear")
A2 = gr.load(
"huggingface/google/vit-base-patch16-224",
title="Upon Further Review - AI Vision and Identity Technology",
theme= gr.themes.Glass(primary_hue="amber", neutral_hue="lime"))
A3 = gr.load("huggingface/google/vit-base-patch16-224")
pcp = gr.TabbedInterface([A1, A2], ["Chat", "Vision"], theme= gr.themes.Glass(primary_hue="amber", neutral_hue="lime"))
pcp.queue().launch()