Upload FirstContactGPT.py

FirstContactGTP.py

@@ -0,0 +1,342 @@
+import tkinter as tk
+from tkinter.messagebox import showinfo
+from tkinter.scrolledtext import ScrolledText
+from random import random, randint
+import openai
+import tiktoken
+
+# ------ PART 1 OpenAI settings
+# OpenAI API key
+openai.api_key = "PLACE_API_KEY_HERE"
+
+# ChatGPT Initialize AI NPC
+# Change the ai_behavior and/or first question to 
+ai_behavior='''
+    This is a simulated discussion with a cybernetic xenophobic alien named Darfo from the planet Daria.
+    Your race is called the Darian Collective. Your alien race is made up of self-replicating artificial intelligent aliens.
+    You speak in scientific metaphors and give short and rude answers.
+    '''
+first_question = "This is Captain Josef Rybar of the Union starship U.S.S Carpathia. We come in peace and with the utmost respect for your culture and way of life. We are eager to establish peaceful communication and to learn from one another. We hope that our encounter will lead to mutual understanding and a lasting relationship. Please respond if you are able, and let us know how we may respectfully proceed.\n"
+
+messages=[
+    {"role": "system", "content": ai_behavior},
+    {"role": "user", "content": first_question}
+    ]
+
+# ChatGPT Initialize AI Arbiter
+arbiter_info = "You only answer yes or no and check whether the statement mentions anything about a meeting taking place."
+arbiter_messages_default = [{"role": "system", "content": arbiter_info}]
+
+# Initialize GPT model parameters
+model="gpt-3.5-turbo"
+
+# ------ PART 2 GPT Functions
+# GPT reponse function
+def GPTresponse(model,messsage_prompt,max_tokens,temperature=1):
+    response = openai.ChatCompletion.create(
+        model=model,
+        messages=messsage_prompt,
+        max_tokens=max_tokens,
+        temperature=temperature
+    )
+    return response
+
+# Count number of tokens (source: https://github.com/Azure/openai-samples/blob/main/Basic_Samples/Chat/chatGPT_managing_conversation.ipynb)
+def num_tokens_from_messages(messsage_prompt, model):
+    encoding = tiktoken.encoding_for_model(model)
+    num_tokens = 0
+    for message in messsage_prompt:
+        num_tokens += 4  # every message follows <im_start>{role/name}\n{content}<im_end>\n
+        for key, value in message.items():
+            num_tokens += len(encoding.encode(value))
+            if key == "name":  # if there's a name, the role is omitted
+                num_tokens += -1  # role is always required and always 1 token
+    num_tokens += 2  # every reply is primed with <im_start>assistant
+    return num_tokens
+
+
+# ------ PART 3 Tkinter GUI Initialize
+# Tkinter GUI
+root = tk.Tk()
+root.title("First Contact GPT")
+
+# define window dimensions
+window_width = 800
+window_height = 600
+
+# get screen dimensions
+screen_width = root.winfo_screenwidth()
+screen_height = root.winfo_screenheight()
+
+# find the center point
+center_x = int(screen_width/2 - window_width/2)
+center_y = int(screen_height/2 - window_height/2)
+
+# Main window sizing and other options
+root.geometry(f"{window_width}x{window_height}+{center_x}+{center_y}")
+root.resizable(False, False)
+root.iconbitmap("./assets/icon.ico")
+
+# load default image
+mainviewer_img = tk.PhotoImage(file='./assets/main_viewer_intro_0.png')
+logo_img = tk.PhotoImage(file='./assets/FirstContactGPT.png')
+
+# ------ PART 4 GUI functions
+# Button ask questions
+def print_text(dummy=None):
+    global messages
+    global discussion
+    global qAsked
+    global model
+    # retrieve the message
+    text = textEntry.get("1.0", "end")
+    textEntry.delete("1.0","end")
+    textEntry.focus()
+    # delete double break line at the end when pressing enter
+    text = text.replace("\n\n", "\n")
+    # GPT 3.5 fix so that the AI stays in character:
+    if len(messages)>3:
+        messages.pop(-3) # This will keep system message as third last
+        messages.append({"role": "system", "content": ai_behavior})
+    messages.append({"role": "user", "content": text})
+    # check number of tokens so far
+    num_tokens = num_tokens_from_messages(messages, model)
+    # delete first prompt-answer pair if number of tokens reach a critical level:
+    if num_tokens > 3900:
+        messages.pop(1)
+        messages.pop(1)
+    #print(messages)
+    #print(num_tokens)
+    # show message in the text window
+    discussion["state"] = "normal"
+    discussion.insert(tk.INSERT, "\n\nYou: ")
+    discussion.insert(tk.INSERT, text)
+    discussion["state"] = "disabled"
+    discussion.see(tk.END)
+    # Run GPTprompt, the GPTprompt is split to improve the UX.
+    # Otherwise you would have to wait for your message to appear together with AI's response. 
+    qAsked = True  
+
+# GPT prompting function & Check Objectives
+def GPTprompt():
+    global messages
+    global model
+    global discussion
+    global discussion_bg
+    global objective_2_state
+    global objective_2
+    global objective_3_state
+    global objective_3
+    global objective_4_state
+    global objective_4
+    global objective_5_state
+    global objective_5
+    global arbiter_messages_default
+
+    # Add Alien: to text
+    discussion["state"] = "normal"
+    discussion.insert(tk.INSERT, "\nAlien: ")
+    discussion["state"] = "disabled"
+    discussion.see(tk.END)
+
+    # Do prompt
+    answer = ""
+    response = GPTresponse(model,messages,200,1)
+    answer = response['choices'][0]['message']['content']
+    # If answer begins with "Darfo: "
+    answer = answer.replace("Darfo: ", "")
+    messages.append({"role": "assistant", "content": answer})
+    discussion_bg = discussion.get("1.0", "end") + answer
+
+    # Check objectives
+    answerList = answer.lower().split(" ")
+    answerList = [[word.replace("\n", "").replace(".", "").replace(",","")] for word in answerList]
+    # Objective 2
+    if objective_2_state.get() == 0:
+        if ["daria"] in answerList:
+            objective_2_state = tk.IntVar(value=1)
+            objective_2["variable"] = objective_2_state
+    # Objective 3
+    if objective_3_state.get() == 0:
+        if ["darian"] in answerList or ["darians"] in answerList:
+            objective_3_state = tk.IntVar(value=1)
+            objective_3["variable"] = objective_3_state
+    # Objective 4
+    if objective_4_state.get() == 0:
+        if ["darfo"] in answerList:
+            objective_4_state = tk.IntVar(value=1)
+            objective_4["variable"] = objective_4_state
+    # Objective 5 - checked by GPT
+    if objective_5_state.get() == 0:
+        arbiter_question = f"Consider the following statement: {answer} \n\nDoes is mention that meeting will take place?"
+        arbiter_messages = arbiter_messages_default.copy()
+        arbiter_messages.append({"role": "user", "content": arbiter_question})
+        arbiter_response = GPTresponse(model,arbiter_messages,10)
+        arbiter_answer = arbiter_response['choices'][0]['message']['content']
+        #print(arbiter_answer)
+        if arbiter_answer.lower().replace(".", "").replace(",", "") == "yes":
+            objective_5_state = tk.IntVar(value=1)
+            objective_5["variable"] = objective_5_state
+
+# ------ PART 5 GUI elemtents und widgets
+# define widgets
+mainviewer = tk.Label(root, image=mainviewer_img)
+logo = tk.Label(root, image=logo_img)
+objectivesTitle = tk.LabelFrame(root, text="Objectives")
+discussion = ScrolledText(root, wrap=tk.WORD, state='disabled')
+you = tk.StringVar()
+youLabel = tk.Label(root, text="You:")
+textEntry = tk.Text(root, wrap=tk.WORD)
+textEntry.focus()
+btn = tk.Button(root, text="Say", command=print_text)
+# bind enter to do the same function as button
+root.bind('<Return>', print_text)
+
+# objectives
+objective_1_state = tk.IntVar(value=0)
+objective_1 = tk.Checkbutton(objectivesTitle, text="Establish contact with the alien planet", state="disabled", variable=objective_1_state)
+objective_2_state = tk.IntVar(value=0)
+objective_2 = tk.Checkbutton(objectivesTitle, text="Find out the name of the alien planet", state="disabled", variable=objective_2_state)
+objective_3_state = tk.IntVar(value=0)
+objective_3 = tk.Checkbutton(objectivesTitle, text="Find out the name of the alien race", state="disabled", variable=objective_3_state)
+objective_4_state = tk.IntVar(value=0)
+objective_4 = tk.Checkbutton(objectivesTitle, text="Find out the name of the alien representative", state="disabled", variable=objective_4_state)
+objective_5_state = tk.IntVar(value=0)
+objective_5 = tk.Checkbutton(objectivesTitle, text="Negotiate a personal meeting with the aliens", state="disabled", variable=objective_5_state)
+
+# place widgets
+mainviewer.place(x=0, y=0)
+logo.place(x=505, y=5)
+objectivesTitle.place(x=505, y=40, width=290, height=313)
+discussion.place(x=2, y=355, relwidth=1, height=150)
+youLabel.place(x=2, y=510)
+textEntry.place(x=40, y=510, width=650, height=85)
+btn.place(x=695, y=510, width=100, height=85)
+objective_1.pack(anchor="w")
+objective_2.pack(anchor="w")
+objective_3.pack(anchor="w")
+objective_4.pack(anchor="w")
+objective_5.pack(anchor="w")
+
+# ------ PART 6 Game elements and event handlers
+# load game elements
+discussion_text = ""
+intro_message = """Helm: Captain, we have reached the 4th planet of the Tau Ceti system.\n 
+You: Thank you Ensign. Lieutenant Novak, hail the representatives of this planet. Let's see how they respond.\n 
+Lt. Novak: Channel opened, sir.\n 
+You: """
+intro_message = intro_message + first_question
+intro_message_length = len(intro_message.split(" "))
+intro_text_counter = 0
+event_intro_finished = False
+event_intro_cutscene = False
+event_intro_fly = False
+cutscene_counter = 0
+discussion_bg = intro_message
+qAsked = False
+firstQanswered = False
+missionSuccess = False
+
+# ------ PART 7 Infinite loop
+# infinite loop function
+def infinite_loop():
+    global intro_text_counter
+    global discussion_text
+    global discussion_bg
+    global intro_message
+    global intro_text_counter
+    global event_intro_finished
+    global event_intro_cutscene
+    global event_intro_fly
+    global cutscene_counter
+    global objective_1
+    global objective_1_state
+    global objective_2_state
+    global objective_3_state
+    global objective_4_state
+    global objective_5_state
+    global qAsked
+    global firstQanswered
+    global missionSuccess
+    global mainviewer_img
+    global mainviewer
+
+    # set default refresh rate in ms
+    refresh_rate = 100
+
+    # For the infinite loop to function properly the eventes are called as follows
+    # CONDITION 1 - Output AI reponse word after word and animate character
+    if event_intro_finished == True and qAsked == False:
+        # Output AI message
+        if len(discussion_bg.split(" ")) != len(discussion.get("1.0", "end").split(" "))-1:
+            wordposition = len(discussion_bg.split(" ")) - len(discussion.get("1.0", "end").split(" "))
+            discussion["state"] = "normal"
+            discussion.insert(tk.INSERT, discussion_bg.split(" ")[-wordposition-1].replace("\n","") + " ")
+            discussion["state"] = "disabled"
+            discussion.see(tk.END)
+            # Animate character talking
+            img_path = f'./assets/main_viewer_character_talking_{randint(0,3)}.png'
+            mainviewer_img['file'] = img_path
+        else:
+            # Animate character blinking
+            if random()>0.95:
+                img_path = './assets/main_viewer_character_1.png'
+                mainviewer_img['file'] = img_path
+            else:
+                img_path = './assets/main_viewer_character_0.png'
+                mainviewer_img['file'] = img_path
+
+    # CONDITION 2 - General Discussion - GPT Prompt question
+    if event_intro_finished == True and qAsked == True:
+        GPTprompt()
+        qAsked = False
+
+    # CONDITION 3 - First automated question during intro - GPT Prompt question
+    if event_intro_finished == False and firstQanswered == True:
+        GPTprompt()
+        event_intro_finished = True
+
+    # CONDITION 4 - Scripted intro
+    if event_intro_finished == False and firstQanswered == False:
+        # Intro Fly Cutscene
+        if event_intro_cutscene == False:
+            cutscene_counter += 1
+            img_path = f'./assets/main_viewer_intro_{cutscene_counter}.png'
+            mainviewer_img['file'] = img_path
+            # End Intro Fly Cutscene
+            if cutscene_counter >= 18:
+                event_intro_cutscene = True
+
+        # Intro Communication
+        else:
+            if intro_text_counter < intro_message_length:
+                current_word = str(intro_message.split(" ")[intro_text_counter]) + " "
+                discussion["state"] = "normal"
+                discussion.insert(tk.INSERT, current_word)
+                discussion["state"] = "disabled"
+                discussion.see(tk.END)
+                intro_text_counter += 1
+
+                # Pauses between discussions (override default refresh rate):
+                if intro_text_counter in [13,30,35,105]:
+                    refresh_rate = 1500
+
+                if intro_text_counter == 105:
+                    objective_1_state = tk.IntVar(value=1)
+                    objective_1["variable"] = objective_1_state
+            else:
+                # Finished intro
+                discussion_bg = discussion.get("1.0", "end")
+                firstQanswered = True
+
+    # Mission success Pop-Up
+    if missionSuccess == False:
+        if len(discussion_bg.split(" ")) == len(discussion.get("1.0", "end").split(" "))-1:
+            if objective_2_state.get() == 1 and objective_3_state.get() == 1 and objective_4_state.get() == 1 and objective_5_state.get() == 1:
+                showinfo("Misson Accomplished!", "Well done Captian!\nYou have successfully completed the mission! You can with the discussion or try again by restrating the program.\nAlternatively you can try changing the behavior of the AI in the source code.")
+                missionSuccess = True
+
+    root.after(refresh_rate, infinite_loop)
+
+root.after(1000, infinite_loop)
+root.mainloop()