Improve chatbot response handling and initial prompt, fix chatbot and BlockBit scratch login.

Author: Fluffyscratch

.gitignore

@@ -1,4 +1,5 @@
 # Hide Python cache files
 */__pycache__/
 __pycache__/
+# Hide environment variables
 *.env

chatbot.py

@@ -10,17 +10,6 @@
     api_key="sk=1234",
 )
 
-# Initial message
-completion = client.chat.completions.create(
-    model="gpt-4-turb",
-    messages=[
-        {
-            "role": "system",
-            "content": "You are a helpful scratch.mit.edu assistant called FluffyBot",
-        }
-    ],
-)
-
 
 # Function to simplify AI calling
 def answer(query: str, username: str):
@@ -33,31 +22,28 @@ def answer(query: str, username: str):
         - username: By what name the AI will call the user
     """
     global index
-    global completion
     print("Answering...")
 
-    completion = client.chat.completions.create(
+    resp = client.chat.completions.create(
         model="gpt-4-turbo",
         messages=[
             {
                 "role": "system",
-                "content": "You are a helpful scratch.mit.edu assistant called ScratchOn. You cannot swear or do anything inappropriate. Never say anything bad about someone or something. Your language must be appropriate for 7-12 year olds. Always refer to yourself as ScratchOn. Either refer to the user by their username, or don't refer to them at all. Keep your answers short and concise.",
+                "content": "You are a helpful scratch.mit.edu assistant and user called ScratchOn. You cannot swear or do anything inappropriate. Never say anything bad about someone or something. Your language must be appropriate for 7-12 year olds. Always refer to yourself as ScratchOn, or _Scratch-On_ as your scratch username. Either refer to the user by their username, or don't refer to them at all. Keep your answers short and concise, up to 500 characters. Do not use regular emojis, only use Scratch-style emojis like :), _:D_, or XD. Speak like a scratch.mit.edu user would, using simple words and phrases a 12 years old would use.",
             },
-            {"role": f"Scratcher called {username}", "content": query},
+            {"role": "user", "content": f"{username}: {query}"},
         ],
     )
 
-    result = completion.choices[0].message.content
+    result = resp.choices[0].message.content
 
     print("Answered !")
     return result
 
 
 # Setup scratch connection
 with open("ScratchOn_private/password.txt") as f:
-    session = sa.login(
-        username="-FluffyBot-", password=f.readlines()[0]
-    )  # TODO: Create new scratch account for ScratchOn
+    session = sa.login(username="_Scratch-On_", password=f.readlines()[0])
 
 profile = session.connect_linked_user()
 events = session.connect_message_events()
@@ -69,15 +55,15 @@ def answer(query: str, username: str):
 def on_message(message):
     print("Message detected")
 
-    time.sleep(10)
+    time.sleep(60)  # Wait 1 minute before replying to prevent comment glitches
 
     comment = profile.comments(page=1, limit=1)[0]
 
     comment.reply(
         content=answer(query=message.comment_fragment, username=message.actor_username)
     )
     """
-    session.connect_user() gets the FluffyBot profile,
+    session.connect_user() gets the ScratchOn profile,
     comment_by_id finds the message based on infos provided by the event handler,
     reply() sends a reply by sending the message to the AI setted up earlier.
     """

commands/currency_commands.py

@@ -14,6 +14,7 @@
 from config import bot, scratch_orange
 from services import request_search, get_latest_response
 
+
 @bot.tree.command(
     name="blockbit_search",
     description="Search a Scratch user and get their BlockBit balance.",
@@ -52,9 +53,12 @@ async def blockbit_search(interact: discord.Interaction, username: str):
         )
     )
 
+
 """
 Error Handling During Search 
 """
+
+
 @blockbit_search.error
 async def search_error(
     interact: discord.Interaction, error: app_commands.AppCommandError

commands/search_commands.py

@@ -21,13 +21,11 @@ async def randomprojects(interact: discord.Interaction, number: int):
     await interact.response.defer()
 
     message = ""
-    max_project_id = scratch.total_site_stats().get('PROJECT_COUNT')
+    max_project_id = scratch.total_site_stats().get("PROJECT_COUNT")
     for i in range(number):
         while True:
             try:
-                project = scratch.get_project(
-                    random.randint(1, max_project_id)
-                )
+                project = scratch.get_project(random.randint(1, max_project_id))
             except scratch.utils.exceptions.ProjectNotFound:
                 continue
             break

services/__init__.py

@@ -5,4 +5,4 @@
 from .scratchblocks import render_blocks_image
 from .blockbits import get_latest_response, request_search
 
-__all__ = ["render_blocks_image", "get_latest_response", "request_search"]
+__all__ = ["render_blocks_image", "get_latest_response", "request_search"]

services/blockbits.py

@@ -10,11 +10,14 @@
 
 
 # Login to Scratch
-session = sa.login("USERNAME", "PASSWORD")
+with open("ScratchOn_private/password.txt") as f:
+    session = sa.login(username="_Scratch-On_", password=f.readlines()[0])
 
 # Connect to the Blockbit project's cloud variables
 cloud = session.connect_cloud(669020072)
 latest_value = 1
+
+
 def request_search(username: str):
     """
     Send a request to Scratch asking for a user's balance or data.
@@ -23,9 +26,10 @@ def request_search(username: str):
     """
     cloud.set_var("TO_HOST", Encoding.encode(f"search&{username}"))
 
+
 async def get_response():
     global latest_value
-    async with websockets.connect('wss://clouddata.scratch.mit.edu') as ws:
+    async with websockets.connect("wss://clouddata.scratch.mit.edu") as ws:
         await ws.send(json.dumps({"method": "handshake", "project_id": 669020072}))
         while True:
             message = json.loads(await ws.recv())
@@ -36,11 +40,12 @@ async def get_response():
                     last_var = list(variables.values())[-1]
                     latest_value = last_var
 
+
 def get_latest_response() -> int | str | None:
     """
     Get the most recent response sent back by the Scratch project.
 
     If the response is a number, return it as an int. Otherwise, return it as a string.
     """
     get_response()
-    return float(latest_value) if latest_value is not None else None
+    return float(latest_value) if latest_value is not None else None