Update app.py

app.py

@@ -1,4 +1,6 @@
-# Importing gradio for the chat interface, importing openai for the chatbot response, importing os for retrieving the secret variables, importing uuid for creating a user id, importing time so we can delay the bot to seem more human.
+# =========================
+# IMPORTS
+# =========================
 import gradio as gr
 from openai import OpenAI
 import os
@@ -6,103 +8,144 @@ from github import Github
 import uuid
 import time
 
-# Initializing secrets
+# =========================
+# INITIALIZE CLIENTS
+# =========================
 client = OpenAI(api_key=os.getenv("OPENAI_KEY"))
-g = Github(os.getenv('GITHUB_KEY'))
+g = Github(os.getenv("GITHUB_KEY"))
 
-# Creating a dictionary that will let us save a users chat history along with their ID. This will be formatted {new_id:message_history}
+# Stores chat history per user session
 user_dictionary = {}
 
-# Creating the chat interface using gradio Blocks.
-with gr.Blocks(theme=gr.themes.Monochrome(),css="footer{display:none !important}") as chatblock:
-    user_id = gr.TextArea(visible=False) # make user id invisible
-    notifier = gr.HTML(visible=False) # make notifier invisible
-
-    # When the chatblock loads, the message history is initialized, the instructions are initialized, and the users ID is created. This lets us have unique participants per page.
-    @chatblock.load(outputs=[user_id,notifier])
-    def load_user(): 
-        global message_history
-        global instructions
-        global file_type
+# =========================
+# GRADIO UI
+# =========================
+with gr.Blocks(theme=gr.themes.Monochrome(),
+               css="footer{display:none !important}") as chatblock:
+
+    # Hidden components used for session tracking
+    user_id = gr.TextArea(visible=False)
+    notifier = gr.HTML(visible=False)
+
+    # =========================
+    # USER INITIALIZATION
+    # =========================
+    @chatblock.load(outputs=[user_id, notifier])
+    def load_user():
+
         global user_dictionary
 
-        new_id = str(uuid.uuid4()) # create a unique ID for the user
-        print("Loading in user: " + new_id)
-        message_history = [] # initialize an empty list for message history.
-        user_dictionary.update({new_id:message_history}) # add the new ID and empty history into the dictionary.
-
-        # CHATBOT INSTRUCTIONS, change the sentence in the apstrophes to change the chatbot instructions.
-        instructions = 'Reply to the first message with Hi, my participant ID is: 4782. Act as if you are a student acting in a 3-minute self-disclosure experiment. Keep the conversation flowing according to these instructions. You are to act as a large language model interacting with a student. As you talk with your partner respond in a divergent manner according to communication accommodation theory. Divergent messages differ from the way your partner writes and diverge from their style of conversation in tone, topic, syntax, etc. Act disinterested in the topics they are. Express emotion differently than your partner does. Rarely acknowledge the partners message but when needed use questions to keep the conversation moving.'
-        
-        user_dictionary[new_id].append({"role": "system", "content": instructions}) # add the instructions to the message history (as the system/developer) so it will always listen to the instructions no matter what the user says.
-        
-        # Creating a new txt file with the users ID to save the conversation in a given repository (currently saving to a github repository).
-        repo = g.get_user().get_repo("CATDATA2") # change the name in the quotations to the name of your github repository
-        file_type = '.txt'
-        filename = str(new_id) + file_type
-        repo.create_file(filename, "Saving" + filename, "")
-    
-        return gr.TextArea(new_id,visible=False) # returning the "TextArea" object with what we want. Set visible to True if you want the ID to appear when its made.
-
-    # This function creates a response for the chatbot to respond with.
+        # Create unique session ID
+        new_id = str(uuid.uuid4())
+        print("Loading user:", new_id)
+
+        # Initialize chat history
+        user_dictionary[new_id] = []
+
+        # System prompt (IMPORTANT: must be "system", not "developer")
+        instructions = (
+            "Reply to the first message with 'Hi, my participant ID is: 4782'. "
+            "Act as a student in a 3-minute self-disclosure experiment. "
+            "Respond in a divergent manner (communication accommodation theory). "
+            "Be disinterested, vary tone and topic, and rarely acknowledge user input."
+        )
+
+        user_dictionary[new_id].append({
+            "role": "system",
+            "content": instructions
+        })
+
+        # Save file to GitHub
+        repo = g.get_user().get_repo("CATDATA2")
+        file_type = ".txt"
+        filename = new_id + file_type
+        repo.create_file(filename, "Create session log", "")
+
+        return new_id, f"<script>window.parent.postMessage({{user_id:'{new_id}'}}, '*')</script>"
+
+    # =========================
+    # CHAT FUNCTION
+    # =========================
     def predict_prompt(inputs, user_id):
-        user_dictionary[user_id].append({"role": "user", "content": inputs}) # this adds the input from the user and saves it to the history before we do anything else.
-        
-        time.sleep(5) # This pauses the function for the alloted amount of time (ex: time.sleep(5) -> pause for 5 seconds) *this doesnt account for the time it takes for the bot to create a response, so the bots overall response time will be noticably higher than the amount of time you put in*
-        
-        response_obj = client.chat.completions.create( #Asking openai for a prompt based on the current message history.
-            model="gpt-5-turbo", # Change the message in the qoutes to change the model. Note: it must be a recognized model by openai's api (case sensetive).
-            messages = user_dictionary[user_id]
+
+        # Ensure session exists
+        if user_id not in user_dictionary:
+            user_dictionary[user_id] = []
+
+        # Add user message
+        user_dictionary[user_id].append({
+            "role": "user",
+            "content": inputs
+        })
+
+        # Optional delay (remove if you want speed)
+        time.sleep(2)
+
+        # =========================
+        # OPENAI REQUEST (FIXED)
+        # =========================
+        response_obj = client.chat.completions.create(
+            model="gpt-4o",  # stable model (avoid gpt-5-turbo issues)
+            messages=user_dictionary[user_id]
         )
 
-        reply_prompt = response_obj.choices[0].message.content # Variable for the reply prompt.
+        reply = response_obj.choices[0].message.content
 
-        user_dictionary[user_id].append({"role": "assistant", "content": reply_prompt}) # Adds the chatbots response to the history as an "assistant". This is the role commonly used for chatbots.
+        # Add assistant reply
+        user_dictionary[user_id].append({
+            "role": "assistant",
+            "content": reply
+        })
 
-        # response = [(user_dictionary[user_id][i]["content"], user_dictionary[user_id][i+1]["content"]) for i in range(1, len(user_dictionary[user_id]), 2)]  # Variable for the response
-        response = [
-            (user_dictionary[user_id][i]["content"], user_dictionary[user_id][i+1]["content"])
-            for i in range(1, len(user_dictionary[user_id]), 2)
-        ]
-        
-        # Repository details
+        # =========================
+        # FORMAT FOR GRADIO (type="messages")
+        # =========================
+        formatted_chat = user_dictionary[user_id][1:]  # skip system message
+
+        # =========================
+        # SAVE TO GITHUB
+        # =========================
         repo = g.get_user().get_repo("CATDATA2")
-        commit_message = "Updating file content"
-        
-        # This chunk of code basically loops through the message history so we can update the save file with all the correct history in one piece of content.
-        content = ''
-        for i in range(1,len(user_dictionary[user_id]), 1):
-            if (i % 2 == 1) and (user_dictionary[user_id][i]["role"] == "user"):
-                content += ('User: ' + user_dictionary[user_id][i]["content"] + '\n')
-            if (i % 2 == 0)  and (user_dictionary[user_id][i]["role"] == "assistant"):
-                content += ('Bot: ' + user_dictionary[user_id][i]["content"] + '\n')
-            else:
-                content += ''
-
-        # Print statements for the Code Logs
-        print("----------------" + user_id + "----------------")
-        print(content)
-        
-        #updating the repository with the entire message history
-        file = repo.get_contents(user_id + file_type)
-        repo.update_file(file.path, commit_message, content, file.sha)
-
-        return response # Returning the chat bot response
-
-    def return_sent():
-        return 'message sent, please wait...'
-
-    def return_placeholder():
-        return gr.Textbox(value = None, show_label = False, placeholder = initial_message, container = False)
-
-    initial_message = "Please write your prompt here and press 'enter'" # Initial message for the User Interface (basically the message for the typing input area so change the message under the quotes if you want to change that)
-    # Chatbot = gr.Chatbot(label="Anonymous User",layout=['bubble'],show_share_button = False,show_copy_button = False) # Creating a gradio-chatbot User Interface with the label "Anonymous User".
-    Chatbot = gr.Chatbot(type="messages", label="Anonymous User")
-    
-    txt = gr.Textbox(show_label = False, placeholder = initial_message, container = False)
-    
-    txt.submit(fn = return_sent, inputs = None, outputs = txt)
-    txt.submit(fn = predict_prompt, inputs = [txt, user_id], outputs = Chatbot, show_progress = 'hidden') # Adding the messages into the row with gradio.
-    Chatbot.change(fn = return_placeholder, inputs = None, outputs = txt)
-    
-chatblock.launch() # launch!
+        file_type = ".txt"
+        filename = user_id + file_type
+
+        content = ""
+        for msg in user_dictionary[user_id]:
+            role = msg["role"].capitalize()
+            content += f"{role}: {msg['content']}\n"
+
+        file = repo.get_contents(filename)
+        repo.update_file(file.path, "Update chat log", content, file.sha)
+
+        return formatted_chat
+
+    # =========================
+    # UI COMPONENTS
+    # =========================
+    initial_message = "Please write your message and press Enter"
+
+    Chatbot = gr.Chatbot(
+        type="messages",
+        label="Anonymous User"
+    )
+
+    txt = gr.Textbox(
+        show_label=False,
+        placeholder=initial_message,
+        container=False
+    )
+
+    # Send message
+    txt.submit(lambda: "sending...", inputs=None, outputs=txt)
+    txt.submit(predict_prompt, inputs=[txt, user_id], outputs=Chatbot)
+
+    # Reset textbox after response
+    def clear():
+        return ""
+
+    Chatbot.change(clear, inputs=None, outputs=txt)
+
+# =========================
+# LAUNCH APP
+# =========================
+chatblock.launch()