Spaces:
Sleeping
Sleeping
File size: 3,185 Bytes
11ec56e |
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 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 |
import gradio as gr
import time
import openai
from pathlib import Path
def transcribe(audio, chatbot_history, openai_key):
time.sleep(5)
transcript = openai.Audio.transcribe("whisper-1", open(audio, "rb"), api_key=openai_key)
content = transcript["text"]
if content:
if not chatbot_history:
return [[content, None]]
else:
return chatbot_history + [[content, None]]
else:
return chatbot_history
def openai_stream(history, openai_key, chat_model):
if not history or history[-1][1]:
return history
history[-1][1] = ""
for chunk in openai.ChatCompletion.create(
model=chat_model,
messages=[{
"role": "user",
"content": history[-1][0]
}],
stream=True,
api_key=openai_key,
):
content = chunk["choices"][0].get("delta", {}).get("content")
if content:
history[-1][1] += content
yield history
def show_message(user_message, history):
return "", history + [[user_message, None]]
theme = gr.themes.Soft(
primary_hue="blue",
neutral_hue="slate",
)
parent_path = Path(__file__).parent
with open(parent_path / "header.MD") as fp:
header = fp.read()
available_models = ['gpt-4', 'gpt-4-0314', 'gpt-4-32k', 'gpt-4-32k-0314', 'gpt-3.5-turbo', 'gpt-3.5-turbo-0301']
with gr.Blocks(theme=theme) as demo:
header_component = gr.Markdown(header)
with gr.Row():
with gr.Column(scale=1):
audio = gr.Audio(label="Talk with ChatGPT", source="microphone", type="filepath", streaming=True)
clear = gr.Button("Clear Chat History")
dark_mode_btn = gr.Button("Dark Mode", variant="primary")
with gr.Column(scale=2):
with gr.Row():
chat_model = gr.Dropdown(choices=available_models, value="gpt-3.5-turbo", allow_custom_value=True)
openai_key = gr.Textbox(label="Enter OPENAI API Key", placeholder="Example: sk-AJDKakdAJD...")
chatbot = gr.Chatbot(label="ChatGPT Dialog")
msg = gr.Textbox(label="Chat with ChatGPT", placeholder="Press <Enter> to submit")
streaming_event_kwargs = dict(
fn=openai_stream,
inputs=[chatbot, openai_key, chat_model],
outputs=chatbot,
)
msg.submit(show_message, [msg, chatbot], [msg, chatbot], queue=False).then(
**streaming_event_kwargs
)
audio.stream(transcribe, inputs=[audio, chatbot, openai_key], outputs=[chatbot]).then(
**streaming_event_kwargs
)
clear.click(lambda: None, None, chatbot, queue=False)
# from gradio.themes.builder
toggle_dark_mode_args = dict(
fn=None,
inputs=None,
outputs=None,
_js="""() => {
if (document.querySelectorAll('.dark').length) {
document.querySelectorAll('.dark').forEach(el => el.classList.remove('dark'));
} else {
document.querySelector('body').classList.add('dark');
}
}""",
)
demo.load(**toggle_dark_mode_args)
dark_mode_btn.click(**toggle_dark_mode_args)
demo.queue()
demo.launch()
|