-
Notifications
You must be signed in to change notification settings - Fork 0
/
Any_chatbot_Template.py
72 lines (52 loc) · 2.39 KB
/
Any_chatbot_Template.py
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
import tkinter as tk
from tkinter import scrolledtext, Entry, Button
import openai
# Set up OpenAI API key
openai.api_key = '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'
class OrderBotGUI:
def __init__(self, root):
self.root = root
self.root.title("OrderBot Chatbot")
# Improved layout
self.chat_history = scrolledtext.ScrolledText(root, wrap=tk.WORD, width=40, height=10)
self.chat_history.pack(padx=10, pady=(10, 0))
self.user_input = Entry(root, width=40)
self.user_input.pack(padx=10, pady=(0, 10))
self.send_button = Button(root, text="Send", command=self.send_user_input)
self.send_button.pack(padx=10, pady=(0, 10))
# Configure tags for user and assistant messages
self.chat_history.tag_config('user', foreground='blue')
self.chat_history.tag_config('assistant', foreground='green')
# Initialize context
self.context = [{'role': 'system', 'content': """ your_prompt
"""}]
def get_completion_from_messages(self, messages, temperature=0):
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=messages,
temperature=temperature
)
return response.choices[0].message["content"]
def send_user_input(self):
user_prompt = self.user_input.get()
self.user_input.delete(0, tk.END)
# Append user input to context
self.context.append({"role": "user", "content": user_prompt})
# Get assistant's response
assistant_response = self.get_completion_from_messages(messages=self.context)
# Append assistant response to context
self.context.append({"role": "assistant", "content": assistant_response})
# Update chat history with styled messages
self.update_chat_history(user_prompt, assistant_response)
def update_chat_history(self, user_prompt, assistant_response):
# Display user message with 'user' tag
self.chat_history.insert(tk.END, f"You: {user_prompt}\n", 'user')
# Display assistant message with 'assistant' tag
self.chat_history.insert(tk.END, f"OrderBot: {assistant_response}\n", 'assistant')
# Scroll to the bottom
self.chat_history.see(tk.END)
# Main application loop
if __name__ == "__main__":
root = tk.Tk()
app = OrderBotGUI(root)
root.mainloop()