|
import gradio as gr |
|
|
|
MODELS = ["Mixtral-8x7B-Instruct-v0.1"] |
|
|
|
def generate(message, chat_history, model, system_prompt): |
|
"""Simulates response generation.""" |
|
response = f"Simulated response for: {message}" |
|
|
|
|
|
chat_history.append((message, response)) |
|
|
|
return chat_history, "" |
|
|
|
DEFAULT_SYSTEM_PROMPT = """ |
|
You are a helpful assistant in normal conversation. |
|
When given a problem to solve, you are an expert problem-solving assistant. |
|
Your task is to provide a detailed, step-by-step solution to a given question. |
|
""" |
|
|
|
with gr.Blocks() as demo: |
|
gr.Markdown("# Custom Chat Interface") |
|
|
|
with gr.Row(): |
|
model = gr.Dropdown(choices=MODELS, label="Select Model", value=MODELS[0]) |
|
|
|
system_prompt = gr.Textbox(value=DEFAULT_SYSTEM_PROMPT, lines=5, label="System Prompt") |
|
chatbot = gr.Chatbot(label="Chat") |
|
msg = gr.Textbox(label="Type your message here...", placeholder="Enter your message...") |
|
|
|
def clear_chat(): |
|
return [], "" |
|
|
|
gr.Button("Clear Chat").click(clear_chat, inputs=None, outputs=[chatbot, msg]) |
|
|
|
msg.submit(generate, inputs=[msg, chatbot, model, system_prompt], outputs=[chatbot, msg]) |
|
|
|
demo.launch() |
