| import gradio as gr |
| from chat import get_response |
|
|
| def clear_session(): |
| return "", [] |
|
|
| def add_query(chat_history, input): |
| if not input: |
| raise gr.Error("Please enter a question.") |
| chat_history.append((input, None)) |
| return chat_history |
|
|
| def response(history, query, model): |
| chat_history = [] |
| for i in history: |
| if i[0]: |
| chat_history.append(i[0]) |
| if i[1]: |
| chat_history.append(i[1]) |
|
|
| messages = [{"role": "user", "content": chat_history[0]}] |
| for i in range(1, len(chat_history), 2): |
| messages.append({"role": "assistant", "content": chat_history[i]}) |
| messages.append({"role": "user", "content": chat_history[i + 1]}) |
|
|
| res_msg = get_response(model, messages) |
| history[-1] = (query, res_msg) |
| return "", history |
|
|
| demo = gr.Blocks(title= "Chatbot", theme="Soft") |
| with demo: |
| with gr.Column("Chatbot - Family Relationships"): |
| model = gr.Radio(["gpt-3.5-turbo", "gpt-4-turbo", "ft:gpt-3.5-turbo-0125:personal::9hiINdK8"], |
| label="model", |
| info="Kindly choose a model before initiating the chat and clear the chat history before switching models. The last one is fine-tuned models.") |
| chatbot = gr.Chatbot(value=[], elem_id='chatbot') |
|
|
| text_input = gr.Textbox( |
| show_label=False, |
| placeholder="Ask me anything!", |
| container=False) |
|
|
| clear_btn = gr.Button("🧹 Clear") |
| text_input.submit(add_query, inputs=[chatbot, text_input], outputs=[chatbot], concurrency_limit=1).\ |
| success(response, inputs=[chatbot, text_input, model], outputs=[text_input, chatbot]) |
| clear_btn.click(clear_session, inputs=[], outputs=[text_input, chatbot]) |
|
|
| demo.launch() |
