ChatbotAI / app.py
TuanScientist's picture
Update app.py
29b56a4
raw
history blame
1.27 kB
import gradio as gr
import openai
DESCRIPTION = """
Hãy để trợ lý của AI Consultant hỗ trợ bạn !
"""
# Define inputs and outputs
inputs = [
gr.inputs.Textbox(label="Câu hỏi:"),
]
outputs = [
gr.outputs.Textbox(label="Câu trả lời:")
]
# Initialize conversation history
conversation_history = []
def chatbot(input):
global conversation_history
openai.api_key = "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"
# Add user input to conversation history
conversation_history.append(f"You: {input}")
# Generate response using conversation history
response = openai.Completion.create(
engine="text-davinci-003",
prompt='\n'.join(conversation_history),
max_tokens=600
)
# Add AI response to conversation history
ai_response = response.choices[0].text.strip()
conversation_history.append(f": {ai_response}")
return ai_response
# Create a Gradio interface
interface = gr.Interface(
fn=chatbot,
inputs=inputs,
outputs=outputs,
title="AI Consultant",
theme="compact",
layout="vertical",
allow_flagging="never",
live=False,
description=DESCRIPTION,
css='style.css'
)
# Launch the interface
interface.launch()